Skip to content

Latest commit

 

History

History
31 lines (21 loc) · 887 Bytes

README.md

File metadata and controls

31 lines (21 loc) · 887 Bytes
Raw

Modified PyTorch-CycleGAN

This project is based on the PyTorch-CycleGAN repository.

Modifications

The changes include:

  1. Training:

    • Added the ability to specify the save path.
    • Designated checkpoint folders and more.

  2. Testing:

    • Allows generating images with the original source dimensions.
    • Enables specifying the checkpoint folder to locate models for predictions.

Usage Examples

train

python train.py --dataroot /path/to/dataset --save_dir /path/to/save/models --checkpoint_dir checkpoints/horse2zebra --batchSize size --cuda

test

python test.py --dataroot /path/to/dataset --batchSize size --checkpoint_dir checkpoints/horse2zebra --save_dir results/

All credits go to the original author @aitorzip

I made this minor modifications just to better suit my own use.