How to train on custom dataset?

mahmoudnafifi / mixedillWB

Reference code for the paper Auto White-Balance Correction for Mixed-Illuminant Scenes.

Other

88 stars 17 forks source link

How to train on custom dataset? #12

Open luser350 opened 2 months ago

luser350 commented 2 months ago

Hi, thanks for sharing your great work. I have a dataset with ground truth (prepared by myself). Your training script is for the WB dataset. Please guide me, on what changes I can make to start training on my dataset.