How many epochs did you train your model for to obtain the result?
harrysdc opened this issue · comments
Hi, your work is amazing!
I am wondering how many epochs did you train your model for to obtain your result? I tried to train your model for 100 epochs but the result is not as good as expected.
Thank you for the report!
Before I answer, what the model did you train and what the dataset do you use? And, what result did you got?
In my opinion, regardless of the model, 100 epoch would be enough. I asked them for confirmation.
I trained nohint_pix2pix with a dataset on Kaggle. All of the result I obtained is bluish. I suspect it might be some issue related to pre-processing since I reuse your class IllustDataset
and most of your code in train.py
in my test script. But I have no clue how to Improve it. Any advice would be highly appreciated. Thank you:)
I think that the problem would be related to color space. Judging from the results, they are in BGR space. It depends on tools for saving image files (ex: OpenCV, Pillow). In both cases, you may be able to get the accurate results after you change color space into another color space (RGB->BGR, BGR->RGB).
Have you solved?