Modify how the mask is handled to reduce its negative effect on SSIM loss

Author: CedricThebault

gsplatInterface/trainer.py

@@ -709,9 +709,7 @@ def train(self):
                 )
 
                 if masks is not None:
-                    colors[~masks] = 0
-                    pixels[~masks] = 0
-
+                    colors[~masks] = pixels[~masks]
 
                 # loss
                 l1loss = F.l1_loss(colors, pixels)