Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training with only a few but high resolution image pairs #416

Closed
yingzwang opened this issue Oct 30, 2018 · 2 comments
Closed

Training with only a few but high resolution image pairs #416

yingzwang opened this issue Oct 30, 2018 · 2 comments

Comments

@yingzwang
Copy link

I want to generate watercolor paintings from pencil sketches. I have perfectly paired training data, but only a few pairs.

Q: When we have paired training data, pix2pix always works better than CycleGAN. Is that correct?

I trained pix2pix model on only two training pairs, and the output is amazing. Please see below my training pairs, and the generated new paintings.
training-pair1
training-pari2
output_paintings
The results above are obtained by training on images of size 512 x 512. The original scans of my paintings have much higher resolution, i.e., 4096 x 4096. I feel that the results can get much better if I take advantage of the high resolution images.

Q: Will it help if I apply a random crop on the original image of size 4096 x 4096, to create many 512 x 512 training pairs?

@junyanz
Copy link
Owner

junyanz commented Oct 31, 2018

Your results look really cool. Data augmentation might help! See this paper for training a pix2pix model with only one image. They have some tricks!

@yingzwang
Copy link
Author

Thanks for your reply. I'll check the paper and try their tricks.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants