Skip to content

Commit

Permalink
Fixes.
Browse files Browse the repository at this point in the history
  • Loading branch information
@folkien
folkien committed Feb 5, 2025
1 parent 4d040e9 commit a41ad37
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 18 deletions.
36 changes: 18 additions & 18 deletions yaya_tools/helpers/augmentations.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,7 @@ def transform_compression_make() -> Augumentation:
[A.ImageCompression(quality_lower=10, quality_upper=15, p=0.99)],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_degrade_make() -> Augumentation:
Expand All @@ -111,7 +111,7 @@ def transform_degrade_make() -> Augumentation:
],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_clahe_make() -> Augumentation:
Expand All @@ -120,7 +120,7 @@ def transform_clahe_make() -> Augumentation:
[A.CLAHE(p=0.999, always_apply=True)],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_equalize_make() -> Augumentation:
Expand All @@ -129,7 +129,7 @@ def transform_equalize_make() -> Augumentation:
[A.Equalize(p=0.999, always_apply=True)],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_sharpen_make() -> Augumentation:
Expand All @@ -138,7 +138,7 @@ def transform_sharpen_make() -> Augumentation:
[A.Sharpen(alpha=(0.25, 0.5), lightness=(0.7, 1.1), p=0.999, always_apply=True)],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_downsize_padding_make() -> Augumentation:
Expand Down Expand Up @@ -174,7 +174,7 @@ def transform_blur_make() -> Augumentation:
[A.Blur(blur_limit=7, p=0.99)],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_median_blur_make() -> Augumentation:
Expand All @@ -183,7 +183,7 @@ def transform_median_blur_make() -> Augumentation:
[A.MedianBlur(blur_limit=7, p=0.99)],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_colorshift_make() -> Augumentation:
Expand All @@ -201,7 +201,7 @@ def transform_colorshift_make() -> Augumentation:
],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_brighten_make() -> Augumentation:
Expand All @@ -217,7 +217,7 @@ def transform_brighten_make() -> Augumentation:
],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_darken_make() -> Augumentation:
Expand All @@ -233,7 +233,7 @@ def transform_darken_make() -> Augumentation:
],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_snow_make() -> Augumentation:
Expand All @@ -242,7 +242,7 @@ def transform_snow_make() -> Augumentation:
[A.RandomSnow(p=0.999)],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_rain_make() -> Augumentation:
Expand All @@ -251,7 +251,7 @@ def transform_rain_make() -> Augumentation:
[A.RandomRain(drop_length=10, blur_value=4, p=0.999)],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_spatter_big_make() -> Augumentation:
Expand All @@ -260,7 +260,7 @@ def transform_spatter_big_make() -> Augumentation:
[A.Spatter(p=0.999, gauss_sigma=4.5)],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_spatter_make() -> Augumentation:
Expand All @@ -269,7 +269,7 @@ def transform_spatter_make() -> Augumentation:
[A.Spatter(p=0.999)],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_spatter_small_make() -> Augumentation:
Expand All @@ -278,7 +278,7 @@ def transform_spatter_small_make() -> Augumentation:
[A.Spatter(p=0.999, gauss_sigma=0.5)],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_blackboxing_make(size: int = 50) -> Augumentation:
Expand Down Expand Up @@ -314,7 +314,7 @@ def transform_isonoise_make() -> Augumentation:
],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_gaussnoise_make() -> Augumentation:
Expand All @@ -330,7 +330,7 @@ def transform_gaussnoise_make() -> Augumentation:
],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_multinoise_make() -> Augumentation:
Expand All @@ -347,7 +347,7 @@ def transform_multinoise_make() -> Augumentation:
],
bbox_params=A.BboxParams(format="albumentations", min_area=100, min_visibility=0.3),
)
return Augumentation(transform=transformation, is_bboxes=True)
return Augumentation(transform=transformation, is_bboxes=False)


def transform_fog_make() -> Augumentation:
Expand Down
1 change: 1 addition & 0 deletions yaya_tools/helpers/image_multiprocessing.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -181,6 +181,7 @@ def multiprocess_augment(
)
else:
augmented = augumentation.transform(image=image)
new_yolo_annotations = file_annotations

# Output image : Save
output_name = f"{get_random_sha1()}.jpeg"
Expand Down

0 comments on commit a41ad37

Please sign in to comment.