colorization-pytorch icon indicating copy to clipboard operation
colorization-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon richzhang

Custom Dataset

Open Mps24-7uk opened this issue 4 years ago • 1 comments

The model is a train on ILSVRC2012. Could someone please help me out in clarifying how to train it on my custom dataset?

Mps24-7uk avatar Feb 15 '21 02:02 Mps24-7uk

Change the data root in train.py. Keep the format the same as ILSVRC2012 so u don't need to do extra coding.

Wanzooo626 avatar May 01 '22 00:05 Wanzooo626