Skip to content

2. Generative

2.1 gan

2.1.1 show_generated_samples(generator, noise, device, num_samples=16)

Función auxiliar para mostrar muestras generadas

Source code in src/layers/generative/gan.py 
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
def show_generated_samples(
    generator: nn.Module, noise, device: str, num_samples: int = 16
) -> None:
    """Función auxiliar para mostrar muestras generadas"""
    generator.eval()
    with torch.no_grad():
        samples = generator(noise[:num_samples]).cpu()
        samples = (samples + 1) / 2  # Desnormalizar de [-1,1] a [0,1]

        fig, axes = plt.subplots(4, 4, figsize=(8, 8))
        for i in range(num_samples):
            row, col = i // 4, i % 4
            axes[row, col].imshow(samples[i, 0], cmap="gray")
            axes[row, col].axis("off")
        plt.tight_layout()
        plt.show()