tianyu0207 / IGD

Official code for 'Deep One-Class Classification via Interpolated Gaussian Descriptor' [AAAI 2022 Oral]
63 stars 8 forks source link

AssertionError: Image size should be larger than 32 due to the 4 downsamplings in ms-ssim #5

Open caiyu6666 opened 2 years ago

caiyu6666 commented 2 years ago

In p32.ssim_main, there isreal_data = data_tmp.reshape(-1, 3, 32, 32). However, when computing the mssim the image size should be larger than 32. How to solve this problem?

zhongjian1999 commented 1 year ago

Hello, have you solved this problem?

caiyu6666 commented 1 year ago

No, do you find the solution?

zhongjian1999 commented 1 year ago

No, do you find the solution?

change 'assert smaller_side > (win_size - 1)' to 'assert smaller_side >= (win_size - 1)' in package pytorch_ssim line 200

caiyu6666 commented 1 year ago

Thank you very much!

caiyu6666 commented 1 year ago

I didn't think about modifying the internal code of the ms-ssim package before.

cjzhzh commented 11 months ago

I change it but it not work

cjzhzh commented 11 months ago

Oh,it work,Thank you