search

dingkeyan93 / DISTS

IQA: Deep Image Structure and Texture Similarity Metric
MIT License
359 stars 42 forks source link

Getting below runtime error when i try to validate the Model #9

Open sunilbage opened 1 year ago

sunilbage commented 1 year ago

Traceback (most recent call last): File "E:\Softwares\AI_Tools\Image_Quality_Assesment\DISTS\DISTS_pytorch\DISTS_pt.py", line 139, in score = model(ref, dist) File "D:\ProgramData\Anaconda3\envs\DISTS\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl return forward_call(*input, *kwargs) File "E:\Softwares\AI_Tools\Image_Quality_Assesment\DISTS\DISTS_pytorch\DISTS_pt.py", line 106, in forward xy_cov = (feats0[k]feats1[k]).mean([2,3],keepdim=True) - x_mean*y_mean RuntimeError: The size of tensor a (455) must match the size of tensor b (451) at non-singleton dimension 3