search

Po-Hsun-Su / pytorch-ssim

pytorch structural similarity (SSIM) loss
Other
1.89k stars 367 forks source link

Loss function #38

Open xsk0802 opened 3 years ago

xsk0802 commented 3 years ago

Hello, I want to use SSIM as a loss function in my own project. How can I reference your code?

Aopotai commented 3 years ago

maybe use pip install pytorch_ssim