Merge pull request #12 from luyuxuan98/main

fix(ssim): correct SSIMLoss input scaling by multiplying with 255

Author: Srameo

basicgs/losses/basic_loss.py

@@ -147,7 +147,7 @@ def forward(self, pred, target, weight=None, **kwargs):
             weight (Tensor, optional): of shape (N, C, H, W). Element-wise weights. Default: None.
         """
         pred = pred.clip(None, 1)
-        l = (1 - _ssim_map_pth(pred, target))
+        l = (1 - _ssim_map_pth(pred * 255., target * 255.))
         l = weight_reduce_loss(l, weight, reduction=self.reduction)
         return self.loss_weight * l
 
@@ -193,4 +193,4 @@ def forward(self, pred, target, weight=None, **kwargs):
             loss = l1_loss(rendered_linear_rgb, target, weight=None, reduction='none')
         scaling_grad = self.get_scaling_grad(rendered_linear_rgb)
         l = loss * scaling_grad
-        return weight_reduce_loss(l, weight, self.reduction)
+        return weight_reduce_loss(l, weight, self.reduction)