From 02b45f83d1748bf2e69a295817b2e0151d710174 Mon Sep 17 00:00:00 2001 From: Oleksandr Bezdieniezhnykh Date: Mon, 27 May 2024 03:16:46 +0300 Subject: [PATCH] add more composition stepss --- preprocessing.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/preprocessing.py b/preprocessing.py index 7e796a4..527e93c 100644 --- a/preprocessing.py +++ b/preprocessing.py @@ -26,7 +26,13 @@ def image_processing(img_ann: ImageLabel) -> [ImageLabel]: transforms = [ A.Compose([A.HorizontalFlip(always_apply=True)], bbox_params=A.BboxParams(format='yolo')), A.Compose([A.RandomBrightnessContrast(always_apply=True)], bbox_params=A.BboxParams(format='yolo')), - A.Compose([A.ShiftScaleRotate(shift_limit=0.1, scale_limit=0.2, rotate_limit=15, always_apply=True)], + A.Compose([A.ShiftScaleRotate(shift_limit=0.1, scale_limit=0.2, rotate_limit=-40, always_apply=True)], + bbox_params=A.BboxParams(format='yolo')), + A.Compose([A.ShiftScaleRotate(shift_limit=0.1, scale_limit=0.2, rotate_limit=-20, always_apply=True)], + bbox_params=A.BboxParams(format='yolo')), + A.Compose([A.ShiftScaleRotate(shift_limit=0.1, scale_limit=0.2, rotate_limit=20, always_apply=True)], + bbox_params=A.BboxParams(format='yolo')), + A.Compose([A.ShiftScaleRotate(shift_limit=0.1, scale_limit=0.2, rotate_limit=40, always_apply=True)], bbox_params=A.BboxParams(format='yolo')) ]