From 700a17f8d35d9893dd231d0be6281861e9e4ab2f Mon Sep 17 00:00:00 2001
From: Thorsten Hempel <42150441+thohemp@users.noreply.github.com>
Date: Thu, 16 May 2024 11:49:47 +0200
Subject: [PATCH] Update rotate augmentation to prevent mirroring objects

https://github.com/NVlabs/Deep_Object_Pose/issues/352
---
 train2/utils_dope.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train2/utils_dope.py b/train2/utils_dope.py
index 642ac079..6485e3db 100755
--- a/train2/utils_dope.py
+++ b/train2/utils_dope.py
@@ -258,7 +258,7 @@ def __getitem__(self,index):
         transform = A.Compose(
             [
                 A.RandomCrop(width=400, height=400),
-                A.Rotate(limit=180),
+                A.Rotate(limit=180, border_mode=cv2.BORDER_CONSTANT),
                 A.RandomBrightnessContrast(brightness_limit=0.2,contrast_limit=0.15,p=1),
                 A.GaussNoise(p=1),