search

hzwer / WACV2024-SAFA

WACV2024 - Scale-Adaptive Feature Aggregation for Efficient Space-Time Video Super-Resolution
MIT License
125 stars 12 forks source link

how to train #1

Closed fzuqxt closed 5 months ago

fzuqxt commented 11 months ago

Hello author, I am very interested in your work, may I ask how to start training?

hzwer commented 11 months ago

We are cleaning up the code and the training loop is similar to videoINR and RIFE.

fzuqxt commented 11 months ago

ok,thanks

fzuqxt commented 10 months ago

Hello author, can you provide the training code

hzwer commented 10 months ago

hi @fzuqxt The code is uploaded. https://github.com/megvii-research/WACV2024-SAFA/blob/main/train.py and https://github.com/megvii-research/WACV2024-SAFA/blob/main/dataset.py

Not tested yet, I'll get around to it this week. Sorry for the inconvenience caused to you.

fzuqxt commented 10 months ago

Thank you very much. Can the training code run normally now?

hzwer commented 8 months ago

Did you encounter any problems? The codes of training loop are mainly from https://github.com/megvii-research/ECCV2022-RIFE/tree/main. I think they should be OK.