ComenzarEmpieza gratis

Generación de imágenes

Ahora que ya has diseñado y entrenado tu GAN, es el momento de evaluar la calidad de las imágenes que puede generar. Para empezar, realizarás una inspección visual para ver si la generación se parece en algo a los Pokémons. Para ello, crearás ruido aleatorio como entrada para el generador, lo pasarás al modelo y graficarás las salidas.

El generador convolucional profundo con pesos entrenados está disponible como gen. torch y matplotlib.pyplot como plt ya están importados.

Este ejercicio forma parte del curso

Aprendizaje profundo para imágenes con PyTorch

Ver curso

Instrucciones del ejercicio

  • Crea un tensor de ruido aleatorio con forma num_images_to_generate 16 , el tamaño del ruido de entrada que utilizaste para entrenar el generador, y asígnalo a noise.
  • Genera imágenes pasando el ruido al generador y asígnalas a fake.
  • Dentro del bucle for, corta fake para extraer la imagen i-th y asígnala a image_tensor.
  • Permuta las dimensiones de image_tensor de (color, altura, anchura) a (altura, anchura, color) y asigna el resultado a image_tensor_permuted.

Ejercicio interactivo práctico

Prueba este ejercicio completando el código de muestra.

num_images_to_generate = 5
# Create random noise tensor
noise = ____

# Generate images
with torch.no_grad():
    fake = ____
print(f"Generated tensor shape: {fake.shape}")
    
for i in range(num_images_to_generate):
    # Slice fake to select i-th image
    image_tensor = ____
    # Permute the image dimensions
    image_tensor_permuted = ____
    plt.imshow(image_tensor_permuted)
    plt.show()
Editar y ejecutar código