Training model on custom dataset

gustavla / autocolorize

Automatic colorizaton of grayscale images using Deep Learning

BSD 3-Clause "New" or "Revised" License

220 stars 65 forks source link

Closed grgvineet closed 7 years ago

grgvineet commented 7 years ago

Can you provide code or network to train model on custom dataset ?

gustavla commented 7 years ago

Hi, I have added instructions of how to train from scratch under the train directory.