junyanz / pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to choose which training results to display ?

AnnaBoim opened this issue · comments

Hi,
I'm training Pix2Pix on my own dataset and i want to compare the training results after each epoch. For this i need the same image to be displayed in the web folder after each epoch.
Currently, after each epoch a different image is displayed (i don't know if they're chosen randomly or by the best result or something else?)
Where can i control which images will be saved for each epoch and their amount (if i want more then one)?

Thanks

Currently, images are chosen randomly based on the data loader. To achieve your goal, you can use the test script with different epoch numbers. Or you can try to modify the visualization code here.