From 82b2487a9c951685f29adb75661f83c37db7e25c Mon Sep 17 00:00:00 2001 From: DannyCooler <164017819+DannyCooler@users.noreply.github.com> Date: Tue, 26 Mar 2024 09:07:43 +0200 Subject: [PATCH] Added additional augmentations --- ultralytics/data/augment.py | 25 ++++++++++++++++--------- 1 file changed, 16 insertions(+), 9 deletions(-) diff --git a/ultralytics/data/augment.py b/ultralytics/data/augment.py index 09e3e5b55fb..99bdd5a036d 100644 --- a/ultralytics/data/augment.py +++ b/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), ])