Update transforms.py

hustvl · Dec 30, 2024 · e9c60be · e9c60be
1 parent 3909ede
commit e9c60be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gausstr/datasets/transforms.py b/gausstr/datasets/transforms.py
@@ -365,7 +365,7 @@ def transform(self, results):
                              filename.split('/')[-1].split('.')[0] + '.npy'))
             feat = torch.from_numpy(feat)
 
-            if img_aug_mats is not None:
+            if self.apply_aug and img_aug_mats is not None:
                 post_rot = img_aug_mats[i][:3, :3]
                 post_tran = img_aug_mats[i][:3, 3]
                 assert post_rot[0, 1] == post_rot[1, 0] == 0  # noqa