fix!: fixed flip for normal maps

alexjercan · May 10, 2021 · 55be05e · 55be05e
1 parent 7c31269
commit 55be05e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/my_albumentations.py b/my_albumentations.py
@@ -16,7 +16,7 @@ def targets(self):
 
     def apply_normal(self, img, **params):
         # when flipping horizontally the normal map should be inversed on the x axis
-        img[:, :, 0] = -1 * img[:, :, 0] + 1
+        img[:, :, 0] = -1 * img[:, :, 0]
         return super().apply(img, **params)
 
 
@@ -26,7 +26,7 @@ def targets(self):
         return dict(super().targets, **{'depth': self.apply, 'normal': self.apply_normal})
 
     def apply_normal(self, img, **params):
-        img[:, :, 1] = -1 * img[:, :, 1] + 1  # y axis flip for normal maps
+        img[:, :, 1] = -1 * img[:, :, 1]  # y axis flip for normal maps
         return super().apply(img, **params)