Hi, I just want to know where the I-SSIM loss is in your code?

dk-liang / FIDTM

[IEEE TMM] Focal Inverse Distance Transform Maps for Crowd Localization

MIT License

169 stars 41 forks source link

Hi, I just want to know where the I-SSIM loss is in your code? #11

Closed Ssssssyz closed 3 years ago

Ssssssyz commented 3 years ago

I have not found the I-SSIM loss in the code, and I just find the nn.MSE loss in the code. So, I want to know where the I-SSIM loss is. Thanks!!!

dk-liang commented 3 years ago

Thanks for your attention! We now have not released the I-SSIM loss, and this project only contains the baseline code for training. Of course, the baseline code also achieves excellent localization and counting performance. We will release the I-SSIM code when the review completed.