search

junyanz / pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch
Other
23.09k stars 6.32k forks source link

Added pytorch color_jitter augmentation #1494

Closed juroberttyb closed 1 year ago

juroberttyb commented 2 years ago

I find pytorch color_jitter augmentation techniques could help when training with small dataset, so I added it as command line argument for people who would like to try this method.

juroberttyb commented 2 years ago

Please let me know if the branch name need to be changed, if so I will create a new pull request.

wjh1065 commented 1 year ago

Hello, I have reviewed the changes you made to the code and tried applying them to my code.

Although color jitter augmentation is a good technique, when I apply it in the get_transform function, it seems to affect both image A and B, and also impacts the output image. If we are training from B to A, it might be better to apply the color jitter only to B (the input).

My response may not be accurate! Thank you.