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')) ]