https://github.com/jayzhan211/cyclegan-pytorch
a pytorch code for CycleGAN
https://github.com/jayzhan211/cyclegan-pytorch
cyclegan-pytorch pytorch-implementation
Last synced: 2 months ago
JSON representation
a pytorch code for CycleGAN
- Host: GitHub
- URL: https://github.com/jayzhan211/cyclegan-pytorch
- Owner: jayzhan211
- Created: 2019-10-09T06:06:59.000Z (over 5 years ago)
- Default Branch: master
- Last Pushed: 2020-05-06T02:18:15.000Z (about 5 years ago)
- Last Synced: 2025-01-25T14:22:10.399Z (4 months ago)
- Topics: cyclegan-pytorch, pytorch-implementation
- Language: Python
- Size: 393 MB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# CycleGAN-pytorch
code based on [junyanz/pytorch-CycleGAN-and-pix2pix](https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix)
# TODO
- [ ] efficient-det as discriminator backbone
- [ ] trivial upsample generator
# Command Line
## Installation
`pip install -r requirements.txt`
## CycleGAN train/test
### To view training results and loss plots, run `python -m visdom.server` and click the URL [http://localhost:8097](http://localhost:8097).
*If you run without it is fine. The code will run the command automaitc*
### Train the model
`python train.py --dataroot ./dataset/selfie2anime --name selfie2anime_ustyle --model ustyle`
`python train.py --dataroot ./dataset/selfie2anime --name selfie2anime_ustyle --model ugatit --light`
#### Continue training
`python train.py --dataroot ./dataset/selfie2anime --name selfie2anime_ustyle --model ustyle --continue_train`
`python train.py --data_root ./horse2zebra --name h2z_cyclegan --model cycle_gan`
### Test the model
`python test.py --dataroot ./dataset/selfie2anime --name selfie2anime_ustyle --model ustyle`
`python test.py --data_root ./horse2zebra --name h2z_cyclegan --model cycle_gan`