search

mahmoudnafifi / Deep_White_Balance

Reference code for the paper: Deep White-Balance Editing (CVPR 2020). Our method is a deep learning multi-task framework for white-balance editing.
Other
536 stars 70 forks source link

Dataset problem #16

Open M11202226HSU opened 3 hours ago

M11202226HSU commented 3 hours ago

Since I couldn't find your [Rendered WB dataset], I decided to download the raw dataset. It contains 62,532 images and 1,991 ground truth images. Could you please help me understand how to set up my dataset? I have a total of 62,532 images and 1,991 ground truth images. What should the data format be? Additionally, how should I divide the dataset into training, validation, and testing sets?

Thank you!

mahmoudnafifi commented 3 hours ago

Hi, the dataset is available here. If links are not working, please check the Google Drive mirror.

Hope that helps.

M11202226HSU commented 3 hours ago

Hello,

I have downloaded the dataset, and now I have input images and ground truth images. I noticed file names like 8D5U5524_C_FF, 8D5U5524_C_A4, 8D5U5524_C_AS, 8D5U5524_D_AS, 8D5U5524_C_L, and 8D5U5524_C_N. What do the suffixes C_FF, C_A4, C_AS, C_N, and C_L mean? Are they necessary for training?

Thank you for your response!

mahmoudnafifi commented 2 hours ago

Yes they are necessary for training. File format is as follows: name_WB_picStyle and the corresponding ground truth image (for auto white balance): name_G_AS. G here refers to ground-truth (manually corrected white balance), C (for example) refers to cloudy white balance, and so on.