YapengTian/TDAN-VSR-CVPR-2020

How to prepare training data?

ShammyFu opened this issue · 0 comments

Hi, I'm gonna train the model from scratch using Vimeo 90K dataset following the provided configurations.
I see that in issue12 you used the script RDN_TrainCode/Prepare_TrainData/Prepare_TrainData_HR_LR_BI.m to process and organize Vimeo dataset and I guess you provided the resulting dataset directory to class Dataset in dataset.py or SR_dataset.py to in further generate the training Dataloader available in Pytorch.

But I have a question that the folder structure of Vimeo 90K is as follows:

image

but the Prepare_TrainData_HR_LR_BI.m script was designed for DIV2K dataset.
May I ask how you modified this script to make it work with Vimeo dataset?
Thanks so much !