Edited from FUNIT-pytorch
- Change Input/Output Logic
├── dataset
└── YOUR_DATASET_NAME
├── train
├── class1 (class folder)
├── xxx.jpg (class1 image)
├── yyy.png
├── ...
├── class2
├── aaa.jpg (class2 image)
├── bbb.png
├── ...
├── class3
├── ...
├── test
├── class1
├── zzz.jpg (class1 image)
├── www.png
├── ...
├── class2
├── ccc.jpg (class2 image)
├── ddd.jpg
├── ...
> python main.py --name FUNIT --dataset YOUR_DATASET_NAME --phase train --K 1
> python main.py --name FUNIT --dataset YOUR_DATASET_NAME --phase test --K 1