StableVSR icon indicating copy to clipboard operation
StableVSR copied to clipboard

Missing Key Code/Files for Training & Validation on Vimeo-90K and REDS Datasets

Open Walnutes opened this issue 10 months ago • 6 comments

Hi,

Thank you for sharing your excellent work! I encountered a few issues while trying to train on the Vimeo-90K and REDS datasets:

  1. Vimeo-90K Dataset Process and Training Code/Scripts:
    The dataset directory only contains reds_dataset.py, and train.py seems to import this class exclusively. Could you kindly provide the complete data loading code for Vimeo-90K, as well as any relevant configuration files like config_vimeo.yaml and metadata.txt?
  2. Validation Output for REDS Dataset:
    While validation_steps is defined, I couldn’t locate the corresponding log, output images (e.g., super-resolved images), or evaluation metrics mentioned in your paper. Could you clarify this part?
  3. Training Parameters and Key Files:
    In the train.sh script, VALIDATION_IMAGE points to the train/gt directory. Should this be corrected to train/lq? Also, the required metadata.txt file for this appears to be missing. I attempted to create it based on REDS_train_metadata.txt and the training settings mentioned in your paper as follows: 000 100\n 011 100\n 015 100\n 020 100 Could you confirm if this matches your setup? If not, could you kindly upload the correct file? Additionally, if Vimeo-90K also requires such these files/configs, could you provide it as well?

Thank you in advance for your support and for providing additional details. Your assistance would be invaluable in reproducing your results!

Walnutes avatar Dec 03 '24 03:12 Walnutes