To get the dataset for training run the following commands. The data is saved in DiscoGAN/datsets folder. You will need to install the required packages as prompted.
- python preprocessing/scrape_tshirts.py
- pytton preprocessing/scrape_watches.py
- python preprocessing/edgeDetector.py
Alternatively, download the dataset from Edges2Tshirts and Edges2Watches and save it in DiscoGAN/datsets folder.
To train tshirts2watches, run
- python ./discogan/image_translation.py --task_name='TshirtsToWatches' --starting_rate=0.5