search

junyanz / interactive-deep-colorization

Deep learning software for colorizing black and white images with a few clicks.
https://richzhang.github.io/ideepcolor/
MIT License
2.69k stars 447 forks source link

Fine-tune model #28

Open zeynelsamak opened 7 years ago

zeynelsamak commented 7 years ago

Hello, I would like to fine-tune the model with new data. It would be great if you could share the train.prototxt. Thanks

junyanz commented 6 years ago

We are looking at it. Will upload one soon.

zeynelsamak commented 6 years ago

thank you @junyanz, I look forward it

j-w-shin commented 6 years ago

Any updates regarding this @junyanz ?

richzhang commented 6 years ago

My apologies for the delay. It is on the list of todo's but I have been caught up with many things. I will get to it as soon as possible.

sushant21 commented 6 years ago

@junyanz @richzhang any updates regarding this?

richzhang commented 6 years ago

I'm working on a pytorch version for public release. Thanks for your patience.

sushant21 commented 6 years ago

Great to hear that! Can you give a time estimate for this release, just to know if I can hold off my work till then?

richzhang commented 6 years ago

Sorry for the delay. I am retraining it in pytorch now. Hopefully release in a few weeks. Thanks for your patience.

richzhang commented 6 years ago

A PyTorch reimplementation is in this repository. We updated this repository as well to support PyTorch models in the backend. Apologies again for the delay.