Downloads · 30 days
0
Uvdr2026/wgan-cifar10-airplanes
wgan-cifar10-airplanes is a machine learning model from Uvdr2026. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for tensorflow.
This repository contains a Wasserstein GAN with Gradient Penalty (WGAN-GP) trained on CIFAR-10 airplane images.
Downloads · 30 days
0
Access
Public
Updated Jan 4, 2026
Repo size
5 MB
Likes
0
Public
Click a slice to open those files.
.keras5 MB · 91%
From the Hugging Face model README
This repository contains a Wasserstein GAN with Gradient Penalty (WGAN-GP)
trained on CIFAR-10 airplane images.
wgan_generator.keras – Trained generator (use for inference)wgan_critic.keras – Trained critictraining_config.json – Training hyperparametersimport tensorflow as tf
from tensorflow import keras
generator = keras.models.load_model("wgan_generator.keras")
LATENT_DIM = 128
noise = tf.random.normal([9, LATENT_DIM])
images = (generator(noise, training=False) + 1) / 2