azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 12:26:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove fog and shadow augmentations

Browse Source 
add install script
This commit is contained in:
zxsanny
2025-05-19 08:30:25 +03:00
parent 292809593c
commit 96056f53ad
5 changed files with 19 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
augmentation.py
+3 -10
View File
@@ -24,17 +24,10 @@ class Augmentator:
self.transform = A.Compose([
# Flips, rotations and brightness
A.HorizontalFlip(p=0.6),
A.RandomBrightnessContrast(p=0.4, brightness_limit=(-0.1, 0.1), contrast_limit=(-0.1, 0.1)),
A.Affine(p=0.7, scale=(0.8, 1.2), rotate=(-20, 20), shear=(-10, 10), translate_percent=0.2),
A.RandomBrightnessContrast(p=0.4, brightness_limit=(-0.3, 0.3), contrast_limit=(-0.05, 0.05)),
A.Affine(p=0.8, scale=(0.8, 1.2), rotate=(-35, 35), shear=(-10, 10)),
# Weather
A.RandomFog(p=0.3, fog_coef_range=(0, 0.3)),
A.RandomShadow(p=0.2),
# Image Quality/Noise
A.MotionBlur(p=0.2, blur_limit=(3, 5)),
# Color Variations
A.MotionBlur(p=0.1, blur_limit=(1, 2)),
A.HueSaturationValue(p=0.4, hue_shift_limit=10, sat_shift_limit=10, val_shift_limit=10)
], bbox_params=A.BboxParams(format='yolo'))