From baa3b89ac59e905e697acdf8187b527bda0fddd4 Mon Sep 17 00:00:00 2001 From: JiaQi Xu <47347516+bubbliiiing@users.noreply.github.com> Date: Tue, 12 May 2020 11:01:44 +0800 Subject: [PATCH] Update utils.py --- utils/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index e82e597..73f9272 100644 --- a/utils/utils.py +++ b/utils/utils.py @@ -96,7 +96,7 @@ def merge_bboxes(bboxes, cutx, cuty): merge_bbox.append(tmp_box) return merge_bbox -def get_random_data_with_Mosaic(annotation_line, input_shape, random=True, max_boxes=50, hue=.1, sat=1.5, val=1.5, proc_img=True): +def get_random_data_with_Mosaic(annotation_line, input_shape, max_boxes=50, hue=.1, sat=1.5, val=1.5): '''random preprocessing for real-time data augmentation''' h, w = input_shape min_offset_x = 0.4 @@ -203,7 +203,7 @@ def get_random_data_with_Mosaic(annotation_line, input_shape, random=True, max_b return new_image, box_data -def get_random_data(annotation_line, input_shape, random=True, max_boxes=50, jitter=.3, hue=.1, sat=1.5, val=1.5, proc_img=True): +def get_random_data(annotation_line, input_shape, max_boxes=50, jitter=.3, hue=.1, sat=1.5, val=1.5): '''random preprocessing for real-time data augmentation''' line = annotation_line.split() image = Image.open(line[0]) -- GitLab