Support batch_size >= 2 testing, fakeA is also outputted in the result file and index.html

taesungp / contrastive-unpaired-translation

Contrastive unpaired image-to-image translation, faster and lighter training than cyclegan (ECCV 2020, in PyTorch)

Other

2.23k stars 417 forks source link

Open HaiyangPeng opened 2 years ago

HaiyangPeng commented 2 years ago

This commit solves the #126 and the restriction of only support batch_size=1 in test.py. More, details:

A to B and B to A outputs are both available in the result file. Index.html is compatible accordingly.
To accelerate the testing speed and original restriction in test.py, batch_size>=2 is supported (multiple GPU inference is supported). The outputs are verified correctly.
Additionally, the arg of '--phase' should be 'test' in the test command in README. Any questions are appreciated ^-^