city-gan-128

Progressive GAN Image Generator for Cities - 128x128 Resolution

This model was trained on a custom city image dataset using a Progressive GAN architecture with techniques to prevent mode collapse. It generates images up to 128x128 resolution.

Training Information

• Epochs: 19/20
• Current Date: 2025-05-10 23:34:14
• Current Resolution: 5x5
• Alpha: 1.0
• Latent Dimension: 100
• Batch Size: 8

Latest Training Metrics

• Generator Loss: 0.6912718415260315
• Discriminator Loss: 0.6670093536376953
• Discriminator Accuracy: 0.625

Sample Images

Training Techniques

• Progressive Growing from 8x8 to 128x128
• Minibatch Discrimination to prevent mode collapse
• Instance Noise for training stability
• Dynamic batch sizing based on resolution

Usage

import torch
from torch import nn

# Define generator architecture
class Generator(nn.Module):
    # ... (architecture code)

# Load the model
generator = Generator(latent_dim=100, channels=3)
generator.load_state_dict(torch.load('model/generator_epoch_19.pt'))
generator.eval()

# Generate images
z = torch.randn(1, 100)
with torch.no_grad():
    fake_image = generator(z)
    # Convert to range [0, 1] for display
    fake_image = (fake_image + 1) / 2.0

Training Progress

Training is currently 95.0% complete. Current resolution phase: 5 of 5