FMA-Net icon indicating copy to clipboard operation
FMA-Net copied to clipboard

training dataset and super resolution capability

Open jimmycv07 opened this issue 1 year ago • 1 comments

Hi, thank you for your great work! I have a couple of questions:

  1. Are the experimental results' pretrained weights trained on the REDS train set (excluding Clips 000, 011, 015, and 020) or on the REDS train, validation, and test sets (excluding Clips 000, 011, 015, and 020)?

  2. Can FMA-Net handle custom time super resolution, such as factors other than 4x?

Thank you in advance for your response.

jimmycv07 avatar May 03 '24 15:05 jimmycv07

Hi, thanks for your interest in FMA-Net!

  1. The provided pretrained weights are trained on the REDS train (excluding 000, 011, 015, 020) and validation sets following previous works. The test set is excluded since ground truth is not provided. Please refer to this code.

  2. FMA-Net can handle scale factors other than 4x. However, we don't provide pretrained models for this, so users need to train the model themselves. Specify the desired scale factor, modify the Pixel_Shuffle_Block accordingly (this is implemented only for 4x upsampling), and proceed with training.

GeunhyukYouk avatar May 03 '24 18:05 GeunhyukYouk

Hi, thank you for your prompt and detailed response! I understand now!

jimmycv07 avatar May 06 '24 07:05 jimmycv07