Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
cyqqq
f7fe698b8e
|
2 years ago | |
|---|---|---|
| data | 2 years ago | |
| images | 2 years ago | |
| models | 2 years ago | |
| options | 2 years ago | |
| scripts | 2 years ago | |
| util | 2 years ago | |
| LICENSE | 2 years ago | |
| README.md | 2 years ago | |
| requirements.txt | 2 years ago | |
| test.py | 2 years ago | |
| train.py | 2 years ago | |
README.md
ZUNIT
Dependencies
you can install all the dependencies by
pip install -r requirements.txt
Datasets
- Download CUB dataset.
- Unzip the birds.zip at
./dataset.
Training
- To view training results and loss plots, run
python -m visdom.server -p 8080
and click the URL http://localhost:8080.
- Run
bash ./scripts/train_bird.sh
Testing
- Run
bash ./scripts/test_bird.sh
- The testing results will be saved in
checkpoints/{exp_name}/resultsdirectory.
Results
Acknowledgement
The code used in this research is inspired by DMIT and FUNIT.
Contact
Feel free to contact me if there is any questions (cyq373@pku.edu.cn).