Final project for CSE455 Computer Vision. Forked from Pytorch pix2pix.
I built a small dataset of Shiba Inu that contains paired RGB and edge images.
Simply download this dataset and upzip it to the datasets/ folder.
To train the model with dog dataset, run
python train.py --dataroot ./datasets/dog --name dog_pix2pix --model pix2pix --direction AtoB