Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

GAN-Based Image Generation with CIFAR-10 Dataset #516

Merged
merged 2 commits into from
Oct 22, 2024

Conversation

Kaibalya27
Copy link
Contributor

Related Issues or bug

  • This pull request adds the GAN-based image generation model using the CIFAR-10 dataset. It includes the architecture for the generator, discriminator, and the GAN model itself. The GAN model is trained on CIFAR-10 to generate images that resemble real images from the dataset.

Fixes: #505

Proposed Changes

  • Added a GAN model that uses the CIFAR-10 dataset to generate images.
  • Integrated functionality for saving intermediate model checkpoints every 10 epochs.
  • Implemented image saving functionality for generated images during training.

Additional Info

  • The model is trained with a batch size of 128 and a latent dimension of 100.
  • Models (generator, discriminator, and combined GAN) are saved as .h5 files every 10 epochs for future use.
  • Generated images are saved in the JPG format.

Results

0bc0cf2c-3d1e-4514-aeed-bd161e094fb0

Copy link

👋 Thank you for opening this pull request! We appreciate your contribution to improving this project. Your PR is under review, and we'll get back to you shortly.
Don't forget to mention the issue you solved!.

To help move the process along, please tag @UppuluriKalyani, @Neilblaze, and @SaiNivedh26 for a faster review!

@SaiNivedh26 SaiNivedh26 merged commit 4302b9b into UppuluriKalyani:main Oct 22, 2024
3 checks passed
Copy link

🎉🎉 Thank you for your contribution! Your PR #516 has been merged! 🎉🎉

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Add GAN Model for Image Generation using CIFAR-10 Dataset
2 participants