YapengTian / TDAN-VSR-CVPR-2020

TDAN: Temporally-Deformable Alignment Network for Video Super-Resolution, CVPR 2020
MIT License
400 stars 62 forks source link

How to prepare training data? #45

Open ShammyFu opened 3 years ago

ShammyFu commented 3 years ago

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 !