Hello! This is how we reused the original implementation of WGAN from this repo for generating custom pictures of houses.
To get our pictures, we used selenium for python and Google Image Search. Our crawler is here. The dataset of 6.5 thousand pictures can be downloaded from here. Initially, our crawled pictures look like this (resized in 64x64 format):
Hyper-parameters and algorithms do not differ much from the one presented in original paper, which is implemented in this notebook (attempt 1) or this notebook (attempt 2), or original tutorial which is implemented in this notebook.
-
First things first, make sure you have last version of our generator and discriminator modules on your Google Drive
-
Then clone this repo:
git clone https://github.com/Dmmc123/wgan_houses.git
- Use notebooks accordingle to crawl data, or re-train the models, or run the GUI