IQA-PyTorch icon indicating copy to clipboard operation
IQA-PyTorch copied to clipboard

Run NR tests on I420 pixel format frames / images

Open pentanol2 opened this issue 2 years ago • 1 comments

I am trying to run some NR metrics (musiq as an example) on images of pixel format I420. The frame size 1920x1080. I get this error shape '[-1, 3, 32, 32]' is invalid for input of size 3331072.

def proecss_frame(frame): 
   frame = TF.to_tensor(frame)
   frame = bt_frame.unsqueeze(0)
   return frame

What is the reason for this error? How can this be solved ?

pentanol2 avatar Mar 02 '23 16:03 pentanol2

Would you please provide example images and the metrics which give errors ? Thanks.

chaofengc avatar Mar 03 '23 10:03 chaofengc