Training Models

For re-training, you need

• line image folder
• color image folder
• image name list file

image name in 2 folder should be same. and set it in line 67 in train_128.py ( or train_512.py dataset = Image2ImageDataset("dat/images_color_train.dat",root+"line/",root+"color/", train=True)

"dat/images_color_train.dat" is image name list file. root+"line/",root+"color/" are path to line/color image folder