Added additional augmentations

ultralytics/data/augment.py

@@ -1153,18 +1153,25 @@ def classify_augmentations(
 
     # Transforms
     albumentations_transform = A.Compose([
-        A.ISONoise(p=0.25, intensity=(0.02, 0.1)),
-        A.ToGray(p=0.12),
-        A.CLAHE(p=0.2, clip_limit=0.5),
-        A.MotionBlur(p=0.2, blur_limit=3),
-        A.RandomToneCurve(p=0.25, scale=0.05),
+        A.ISONoise(p=0.1, intensity=(0.02, 0.1)),
+        A.ToGray(p=0.02),
+        A.CLAHE(p=0.1, clip_limit=0.5),
+        A.MotionBlur(p=0.1, blur_limit=3),
+        A.GaussianBlur(p=1, blur_limit=(3, 5)),
+        A.RandomToneCurve(p=0.1, scale=0.05),
         A.Sharpen(p=0.1, alpha=(0.02, 0.1), lightness=(0.8, 1.0)),
-        A.Emboss(p=0.2, alpha=(0.02, 0.1), strength=(0.05, 0.2)),
-        A.MultiplicativeNoise(p=0.2, multiplier=(0.95, 1.05)),
-        A.Rotate(p=0.2, limit=5),
+        A.Emboss(p=0.1, alpha=(0.02, 0.1), strength=(0.05, 0.2)),
+        A.MultiplicativeNoise(p=0.1, multiplier=(0.95, 1.05)),
+        A.Rotate(p=0.1, limit=10, border_mode=cv2.BORDER_CONSTANT),
         A.ChromaticAberration(primary_distortion_limit=(0.01, 0.05), 
                               secondary_distortion_limit=(0.01, 0.07),
-                              mode='random', p=0.2),
+                              mode='random', p=0.1),
+        A.RGBShift(p=0.5, r_shift_limit=10, g_shift_limit=10, b_shift_limit=10),
+        A.RingingOvershoot(p=0.1, blur_limit=(5, 11)),
+        A.OpticalDistortion(p=0.1, distort_limit=0.1, shift_limit=0.1, border_mode=cv2.BORDER_CONSTANT),
+        A.GridDistortion(p=0.1, num_steps=5, distort_limit=0.3, border_mode=cv2.BORDER_CONSTANT),
+        A.Perspective(p=0.1, scale=(0.05, 0.1)),
+        A.PiecewiseAffine(p=0.1, scale=(0.01, 0.015)), 
         A.ImageCompression(quality_lower=75, p=0.0),
         ])