Po-Hsun-Su / pytorch-ssim

pytorch structural similarity (SSIM) loss
Other
1.87k stars 364 forks source link

Support for images in range 0 to 1 #41

Open svshivapuja-intel opened 2 years ago

svshivapuja-intel commented 2 years ago

My model consumes data in the range of 0 to1 instead of 0 to 255. Can the same function be used in that case or should I convert it ?