From 59457048db094b058cb5460bf2e2b7ed4296573a Mon Sep 17 00:00:00 2001 From: FarisHijazi Date: Wed, 21 Sep 2022 13:18:10 +0300 Subject: [PATCH 1/5] fixed utf encoding issue --- trdg/data_generator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/trdg/data_generator.py b/trdg/data_generator.py index 75a79467fc..2ea2aacdad 100644 --- a/trdg/data_generator.py +++ b/trdg/data_generator.py @@ -274,12 +274,12 @@ def generate( final_mask.save(os.path.join(out_dir, mask_name)) if output_bboxes == 1: bboxes = mask_to_bboxes(final_mask) - with open(os.path.join(out_dir, box_name), "w") as f: + with open(os.path.join(out_dir, box_name), "w", encoding='utf8') as f: for bbox in bboxes: f.write(" ".join([str(v) for v in bbox]) + "\n") if output_bboxes == 2: bboxes = mask_to_bboxes(final_mask, tess=True) - with open(os.path.join(out_dir, tess_box_name), "w") as f: + with open(os.path.join(out_dir, tess_box_name), "w", encoding='utf8') as f: for bbox, char in zip(bboxes, text): f.write( " ".join([char] + [str(v) for v in bbox] + ["0"]) + "\n" From e6c6fe5a1d0caed4ca51d901475b49aefbffc737 Mon Sep 17 00:00:00 2001 From: FarisHijazi Date: Thu, 6 Oct 2022 08:09:49 +0300 Subject: [PATCH 2/5] feat: now also saves .txt file containing text --- trdg/data_generator.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/trdg/data_generator.py b/trdg/data_generator.py index 2ea2aacdad..b52bb1098f 100644 --- a/trdg/data_generator.py +++ b/trdg/data_generator.py @@ -270,6 +270,10 @@ def generate( # Save the image if out_dir is not None: final_image.save(os.path.join(out_dir, image_name)) + + with open(os.path.join(out_dir, "{}.txt".format(name)), "w", encoding='utf8') as f: + f.write(text) + if output_mask == 1: final_mask.save(os.path.join(out_dir, mask_name)) if output_bboxes == 1: From 68e9f02d228777ee8973b407ab57e70b032b08ac Mon Sep 17 00:00:00 2001 From: FarisHijazi Date: Thu, 6 Oct 2022 08:13:39 +0300 Subject: [PATCH 3/5] upgraded pillow version --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index c0735c5285..cc1f97a3b1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -pillow>=7.0.0 +pillow>=7.1.2 requests>=2.20.0 opencv-python>=4.2.0.32 tqdm>=4.23.0 From 539cd13cc0809ef88609b07d7d6b6713e5acd8ea Mon Sep 17 00:00:00 2001 From: Somayah Alharbi Date: Tue, 8 Nov 2022 16:52:18 +0300 Subject: [PATCH 4/5] distorsion 3 now have 50% distorsion only --- trdg/data_generator.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/trdg/data_generator.py b/trdg/data_generator.py index b52bb1098f..cbce086026 100644 --- a/trdg/data_generator.py +++ b/trdg/data_generator.py @@ -113,6 +113,11 @@ def generate( horizontal=(distorsion_orientation == 1 or distorsion_orientation == 2), ) else: + import random + if random.random() < 0.5: + distorted_img = rotated_img # Mind = blown + distorted_mask = rotated_mask + else: distorted_img, distorted_mask = distorsion_generator.random( rotated_img, rotated_mask, From 11543ba43c0731a4cb39225797fefd62041158bc Mon Sep 17 00:00:00 2001 From: Faris Hijazi <47270244+FarisHijazi@users.noreply.github.com> Date: Tue, 8 Nov 2022 20:16:04 +0300 Subject: [PATCH 5/5] fixed indentationi from distortion patch --- trdg/data_generator.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/trdg/data_generator.py b/trdg/data_generator.py index cbce086026..9490aab932 100644 --- a/trdg/data_generator.py +++ b/trdg/data_generator.py @@ -118,12 +118,12 @@ def generate( distorted_img = rotated_img # Mind = blown distorted_mask = rotated_mask else: - distorted_img, distorted_mask = distorsion_generator.random( - rotated_img, - rotated_mask, - vertical=(distorsion_orientation == 0 or distorsion_orientation == 2), - horizontal=(distorsion_orientation == 1 or distorsion_orientation == 2), - ) + distorted_img, distorted_mask = distorsion_generator.random( + rotated_img, + rotated_mask, + vertical=(distorsion_orientation == 0 or distorsion_orientation == 2), + horizontal=(distorsion_orientation == 1 or distorsion_orientation == 2), + ) ################################## # Resize image to desired format #