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),