How to train on my own dataset?

[Badw0lf613]

Sorry to bother. I'm trying to train DDColor on my own dataset. Is there any training scripts? By the way, how to prepare and organize my dataset?

[Badw0lf613]

@piddnad

[Badw0lf613]

I see the training scripts and files, that is basicsr/train.py.

[piddnad]

Sure! It's easy to train DDColor on your own dataset.

First, you need to organize your dataset by placing the images in a folder, and then use data_list/get_meta_file.py to generate a datalist file.

Next, you need to modify the path to the training dataset's metadata file in the training config.

Finally, as you mentioned, you can use basicsr/train.py for training. The command format can be found in scripts/train.sh.

Hope this answer has addressed your question.