add resume training possibility

2026-04-22 10:46:35 +00:00 · 2025-05-19 18:32:41 +03:00
parent 96056f53ad
commit 8fb24384db
2 changed files with 11 additions and 1 deletions
@@ -22,7 +22,6 @@ class Augmentator:
        self.correct_min_bbox_size = 0.01

        self.transform = A.Compose([
-            # Flips, rotations and brightness
            A.HorizontalFlip(p=0.6),
            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)),