richzhang / colorization

Automatic colorization using deep neural networks. "Colorful Image Colorization." In ECCV, 2016.
http://richzhang.github.io/colorization/
BSD 2-Clause "Simplified" License
3.34k stars 921 forks source link

tensorflow #72

Closed Flyzzz closed 4 years ago

Flyzzz commented 4 years ago

hi, I wonder is there any implement base on tensorflow?

richzhang commented 4 years ago

Hi, I do not have a tensorflow implementation. But I did convert the weights to pytorch; see the updated repo