fixed the random rotation augmentation inconsistency (#155)

PatBall1 · Nov 22, 2024 · c983eda · c983eda
1 parent b322c7b
commit c983eda
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/detectree2/models/train.py b/detectree2/models/train.py
@@ -463,9 +463,8 @@ def build_train_loader(cls, cfg):
 
         # Define basic augmentations including rotation and flipping
         augmentations = [
-            T.RandomRotation(angle=[90, 90], expand=False),
-            T.RandomFlip(prob=0.4, horizontal=True, vertical=False),
-            T.RandomFlip(prob=0.4, horizontal=False, vertical=True),
+            T.RandomRotation(angle=[0, 360], expand=False),
+            T.RandomFlip(prob=0.5, horizontal=True, vertical=False),
         ]
 
         # Additional augmentations for RGB images

diff --git a/docs/source/tutorial.rst b/docs/source/tutorial.rst
@@ -474,9 +474,8 @@ By default, random rotations and flips will be performed on input images.
 .. code-block:: python
 
    augmentations = [
-      T.RandomRotation(angle=[90, 90], expand=False),
-      T.RandomFlip(prob=0.4, horizontal=True, vertical=False),
-      T.RandomFlip(prob=0.4, horizontal=False, vertical=True),
+      T.RandomRotation(angle=[0, 360], expand=False),
+      T.RandomFlip(prob=0.5, horizontal=True, vertical=False),
    ]
 
 If the input data is RGB, additional augmentations will be applied to adjust the brightness, contrast, saturation, and