L_ms_ssim

RongLiu-Leo · May 24, 2024 · 02a7dbd · 02a7dbd
1 parent b1ae8fb
commit 02a7dbd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/arguments/__init__.py b/arguments/__init__.py
@@ -81,7 +81,7 @@ def __init__(self, parser):
         self.opacity_lr = 0.05
         self.scaling_lr = 0.005
         self.rotation_lr = 0.001
-        self.lambda_ssim = 0.1
+        self.lambda_ms_ssim = 0.1
         self.lambda_normal = 0.1
 
         self.densification_interval = 100

diff --git a/train.py b/train.py
@@ -73,9 +73,9 @@ def training(dataset, opt, pipe, testing_iterations, saving_iterations, checkpoi
         # Loss
         gt_image = viewpoint_cam.original_image.cuda()
         Ll1 = l1_loss(image, gt_image)
-        Lssim = (1.0 - ms_ssim(image, gt_image))
+        L_ms_ssim = (1.0 - ms_ssim(image, gt_image))
 
-        Lrgb =  (1.0 - opt.lambda_ssim) * Ll1 + opt.lambda_ssim * Lssim 
+        Lrgb =  (1.0 - opt.lambda_ms_ssim) * Ll1 + opt.lambda_ms_ssim * L_ms_ssim 
         loss = Lrgb
 
         if iteration < opt.atom_proliferation_until: