Sid1279 / LEGO-GAN Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

Generative Adversarial Networks (GANs) to trained to generate images of LEGO minifigures. Techniques adapted from the following research paper: https://arxiv.org/pdf/1511.06434.pdf

0 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 15 Commits
LEGO_DCGAN		LEGO_DCGAN
.gitignore		.gitignore
README.md		README.md

Repository files navigation

LEGO-GAN

A deep convolutional generative adversarial neural network to generate realistic images of LEGO minifigures

LEGO_DCGAN/minifig_images contains images of LEGO minifigures to train the model. Obtained via some web-scraping on Rebrickable.
LEGO_DCGAN/minifig_output contains generated images for each trial during training.
LEGO_DCGAN/generated: has several generated images by running LEGO_DCGAN/generated/inference.py. Give it a try yourself ;)

Below are some images generated by the DCGAN generator:

1. 64x64 (took about 7-8 hours of training on a CPU)

2. 128x128 (took about 10-12 hours of training on a CPU)

Learn more about implementing your own DCGAN in PyTorch here

About

Generative Adversarial Networks (GANs) to trained to generate images of LEGO minifigures. Techniques adapted from the following research paper: https://arxiv.org/pdf/1511.06434.pdf

python pytorch generative-adversarial-network dcgan

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%