search

junyanz / pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch
Other
22.8k stars 6.29k forks source link

Pretrained model for colorizing #1554

Open viethoang303 opened 1 year ago

viethoang303 commented 1 year ago

Hi, author

I am a newbie with GAN model. I see this repository and read README but there are no checkpoint for image colorization task. Can you release it, please?

taesungp commented 1 year ago

We don't have such a model. Actually, the setting of CycleGAN is likely not the best for tackling image colorization task, because image colorization can be trained in a paired image-to-image setting, rather than unpaired.

junyanz commented 1 year ago

Richard's repo has some pre-trained colorization checkpoints: https://github.com/richzhang/colorization-pytorch