How can I test for Vimeo-90K and Vid4, and output PSNR/SSIM by using this model?

ckkelvinchan / BasicVSR_PlusPlus

Official repository of "BasicVSR++: Improving Video Super-Resolution with Enhanced Propagation and Alignment"

Apache License 2.0

589 stars 64 forks source link

How can I test for Vimeo-90K and Vid4, and output PSNR/SSIM by using this model? #34

Open CuteYuzu opened 1 year ago

CuteYuzu commented 1 year ago

I have no idea about how to use this model to test the pre-trained model of BasicVSR++, It doesn't mentioned in "readme.md" and I don't know how to do that... VRT and RVRT has clear steps of how to test, but step of this model is not very plain. Could someone please give me some suggestions? I'd really appreciate your reply!

CuteYuzu commented 1 year ago

I can't find how to calculate PSNR and SSIM in person using the dataset Vimeo-90K and Vid4... There's only demo .py file and I can't find this entrance of calculate and output PSNR and SSIM. Could someone please give me some suggestions? I'd really appreciate your reply!