taesungp / contrastive-unpaired-translation

Contrastive unpaired image-to-image translation, faster and lighter training than cyclegan (ECCV 2020, in PyTorch)
https://taesung.me/ContrastiveUnpairedTranslation/
Other
2.2k stars 417 forks source link

How to use "detect_cat_face.py"? #48

Open dlshu opened 3 years ago

dlshu commented 3 years ago

Hi, I'm wondering how the script file CUT/datasets/detect_cat_face.py is used in the training or the test phase of the neural network model? From the repository, I couldn't find a part where detect_cat_face.py is called. Thanks!

JunlinHan commented 3 years ago

I think it's for aligning the dataset Cat <-> Dog( AFHQ(Animal faces dataset)). If you are interested in this dataset, you may go to StarGANV2 repo and download it. It's already aligned.

ranch-hands commented 3 years ago

I think it's for aligning the dataset Cat <-> Dog( AFHQ(Animal faces dataset)). If you are interested in this dataset, you may go to StarGANV2 repo and download it. It's already aligned.

hi do u know how we can test the network on other datasets?