richzhang / colorization-pytorch

PyTorch reimplementation of Interactive Deep Colorization
https://richzhang.github.io/ideepcolor/
MIT License
597 stars 113 forks source link

Interactive colorization does not work for retrained model. #10

Closed bl0 closed 5 years ago

bl0 commented 5 years ago

Hi, thanks for your great work. I tried the interactive colorization GUI from https://github.com/junyanz/interactive-deep-colorization with your provided retrained model, but it doesn't work. The screen shot is as following: Screenshot from 2019-05-26 15-26-23

I also tried the caffe model, it works well. Screenshot from 2019-05-26 15-32-39