search

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

PyTorch Example #89

Open luke-777 opened 3 years ago

luke-777 commented 3 years ago

Hi is it possible or is there a example to train the model with PyTorch with my own dataset?