From f8bf5bff4bbb679c07daa38d7c2467560af507de Mon Sep 17 00:00:00 2001 From: pparfrey <44984488+pparfrey@users.noreply.github.com> Date: Sat, 13 Mar 2021 11:20:08 -0500 Subject: [PATCH 1/6] TEST: Added test file for yolo_to_xyxy --- tests/assets/yolo_input_file/raise_the_flag_frame00000.txt | 1 + 1 file changed, 1 insertion(+) create mode 100644 tests/assets/yolo_input_file/raise_the_flag_frame00000.txt diff --git a/tests/assets/yolo_input_file/raise_the_flag_frame00000.txt b/tests/assets/yolo_input_file/raise_the_flag_frame00000.txt new file mode 100644 index 0000000..e5ccebe --- /dev/null +++ b/tests/assets/yolo_input_file/raise_the_flag_frame00000.txt @@ -0,0 +1 @@ +8 0.640625 0.43125 0.328125 0.08333333333333333 From 819e8e261548c4c0513a17eadb74a63a934df372 Mon Sep 17 00:00:00 2001 From: pparfrey <44984488+pparfrey@users.noreply.github.com> Date: Sat, 13 Mar 2021 11:22:33 -0500 Subject: [PATCH 2/6] TEST: Added test files for convert_yolo_to_xyxy --- tests/assets/yolo_input_file/raise-the-flag_frame00000.txt | 1 + tests/assets/yolo_input_file/raise-the-flag_frame00001.txt | 1 + tests/assets/yolo_input_file/raise-the-flag_frame00002.txt | 1 + tests/assets/yolo_input_file/raise-the-flag_frame00003.txt | 1 + tests/assets/yolo_input_file/raise-the-flag_frame00004.txt | 1 + tests/assets/yolo_input_file/raise-the-flag_frame00005.txt | 1 + tests/assets/yolo_input_file/raise-the-flag_frame00006.txt | 1 + tests/assets/yolo_input_file/raise-the-flag_frame00007.txt | 1 + tests/assets/yolo_input_file/raise-the-flag_frame00008.txt | 1 + tests/assets/yolo_input_file/raise-the-flag_frame00009.txt | 1 + 10 files changed, 10 insertions(+) create mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00000.txt create mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00001.txt create mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00002.txt create mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00003.txt create mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00004.txt create mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00005.txt create mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00006.txt create mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00007.txt create mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00008.txt create mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00009.txt diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00000.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00000.txt new file mode 100644 index 0000000..f8c3ef5 --- /dev/null +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00000.txt @@ -0,0 +1 @@ +8 0.640625 0.43125 0.328125 0.08333333333333333 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00001.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00001.txt new file mode 100644 index 0000000..b7815d9 --- /dev/null +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00001.txt @@ -0,0 +1 @@ +8 0.61953125 0.43333333333333335 0.3265625 0.08333333333333333 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00002.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00002.txt new file mode 100644 index 0000000..fd472f3 --- /dev/null +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00002.txt @@ -0,0 +1 @@ +8 0.5921875 0.4375 0.325 0.08333333333333333 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00003.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00003.txt new file mode 100644 index 0000000..d41f79e --- /dev/null +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00003.txt @@ -0,0 +1 @@ +8 0.58125 0.4354166666666667 0.325 0.08333333333333333 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00004.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00004.txt new file mode 100644 index 0000000..1ff1759 --- /dev/null +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00004.txt @@ -0,0 +1 @@ +8 0.565625 0.4270833333333333 0.325 0.0875 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00005.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00005.txt new file mode 100644 index 0000000..29a24a7 --- /dev/null +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00005.txt @@ -0,0 +1 @@ +8 0.5546875 0.42291666666666666 0.325 0.0875 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00006.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00006.txt new file mode 100644 index 0000000..9819386 --- /dev/null +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00006.txt @@ -0,0 +1 @@ +8 0.540625 0.42291666666666666 0.325 0.0875 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00007.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00007.txt new file mode 100644 index 0000000..47f718b --- /dev/null +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00007.txt @@ -0,0 +1 @@ +8 0.528125 0.428125 0.325 0.08541666666666667 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00008.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00008.txt new file mode 100644 index 0000000..d1907a0 --- /dev/null +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00008.txt @@ -0,0 +1 @@ +8 0.51875 0.4322916666666667 0.325 0.08541666666666667 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00009.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00009.txt new file mode 100644 index 0000000..ae85f6b --- /dev/null +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00009.txt @@ -0,0 +1 @@ +8 0.50703125 0.42916666666666664 0.3265625 0.08333333333333333 From 04c3c069d93a76a9cfb63982c38f03c132f19680 Mon Sep 17 00:00:00 2001 From: pparfrey <44984488+pparfrey@users.noreply.github.com> Date: Sat, 13 Mar 2021 11:23:51 -0500 Subject: [PATCH 3/6] TEST: Deleting extra copy of test file --- tests/assets/yolo_input_file/raise_the_flag_frame00000.txt | 1 - 1 file changed, 1 deletion(-) delete mode 100644 tests/assets/yolo_input_file/raise_the_flag_frame00000.txt diff --git a/tests/assets/yolo_input_file/raise_the_flag_frame00000.txt b/tests/assets/yolo_input_file/raise_the_flag_frame00000.txt deleted file mode 100644 index e5ccebe..0000000 --- a/tests/assets/yolo_input_file/raise_the_flag_frame00000.txt +++ /dev/null @@ -1 +0,0 @@ -8 0.640625 0.43125 0.328125 0.08333333333333333 From bed8caec9edf1f1059a6940d55fce3879d91d078 Mon Sep 17 00:00:00 2001 From: pparfrey Date: Fri, 16 Apr 2021 12:05:36 -0400 Subject: [PATCH 4/6] TEST: Edited files to have different # annotations --- tests/assets/yolo_input_file/raise-the-flag_frame00000.txt | 1 - tests/assets/yolo_input_file/raise-the-flag_frame00001.txt | 3 +++ tests/assets/yolo_input_file/raise-the-flag_frame00003.txt | 1 + tests/assets/yolo_input_file/raise-the-flag_frame00005.txt | 1 + 4 files changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00000.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00000.txt index f8c3ef5..e69de29 100644 --- a/tests/assets/yolo_input_file/raise-the-flag_frame00000.txt +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00000.txt @@ -1 +0,0 @@ -8 0.640625 0.43125 0.328125 0.08333333333333333 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00001.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00001.txt index b7815d9..0e90c02 100644 --- a/tests/assets/yolo_input_file/raise-the-flag_frame00001.txt +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00001.txt @@ -1 +1,4 @@ 8 0.61953125 0.43333333333333335 0.3265625 0.08333333333333333 +8 0.640625 0.43125 0.328125 0.08333333333333333 +8 0.50703125 0.42916666666666664 0.3265625 0.08333333333333333 +8 0.51875 0.4322916666666667 0.325 0.08541666666666667 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00003.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00003.txt index d41f79e..fa745bc 100644 --- a/tests/assets/yolo_input_file/raise-the-flag_frame00003.txt +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00003.txt @@ -1 +1,2 @@ 8 0.58125 0.4354166666666667 0.325 0.08333333333333333 +8 0.528125 0.428125 0.325 0.08541666666666667 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00005.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00005.txt index 29a24a7..317dfb4 100644 --- a/tests/assets/yolo_input_file/raise-the-flag_frame00005.txt +++ b/tests/assets/yolo_input_file/raise-the-flag_frame00005.txt @@ -1 +1,2 @@ +8 0.540625 0.42291666666666666 0.325 0.0875 8 0.5546875 0.42291666666666666 0.325 0.0875 From ad0aa31346db18a815a5433a72d29110040b3213 Mon Sep 17 00:00:00 2001 From: pparfrey Date: Fri, 16 Apr 2021 12:12:13 -0400 Subject: [PATCH 5/6] TEST: Added test file for convert_yolo_to_xyxy.py; incomplete --- tests/test_convert_yolo_xyxy.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 tests/test_convert_yolo_xyxy.py diff --git a/tests/test_convert_yolo_xyxy.py b/tests/test_convert_yolo_xyxy.py new file mode 100644 index 0000000..f851616 --- /dev/null +++ b/tests/test_convert_yolo_xyxy.py @@ -0,0 +1,28 @@ +import filecmp +import shutil + +from utils import get_relative_path + +from universal_devkit.scripts.convert_yolo_to_xyxy import ( + convert_images_from_yolo_to_xyxy, +) + + +def test_create_logs_json(): + # The path to the folder with yolo input files + input_files_path = get_relative_path("assets/yolo_input_file") + + # The directory to be used as output + output_files_path = get_relative_path("assets/yolo_output_files") + + # Correct yolo files + correct_output_path = get_relative_path("assets/yolo_correct_files") + + # Convert the yolo input files to xyxy + convert_images_from_yolo_to_xyxy(input_files_path, output_files_path) + + comp = filecmp.dircmp(output_files_path, correct_output_path) + + assert len(comp.same_files) == 6 + + shutil.rmtree(output_files_path) From ba9fefc96ba83716c3b1a5ef4a6fa97794c29191 Mon Sep 17 00:00:00 2001 From: pparfrey Date: Fri, 16 Apr 2021 12:15:36 -0400 Subject: [PATCH 6/6] TEST: Deleted unnecessary yolo test files --- tests/assets/yolo_input_file/raise-the-flag_frame00006.txt | 1 - tests/assets/yolo_input_file/raise-the-flag_frame00007.txt | 1 - tests/assets/yolo_input_file/raise-the-flag_frame00008.txt | 1 - tests/assets/yolo_input_file/raise-the-flag_frame00009.txt | 1 - 4 files changed, 4 deletions(-) delete mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00006.txt delete mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00007.txt delete mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00008.txt delete mode 100644 tests/assets/yolo_input_file/raise-the-flag_frame00009.txt diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00006.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00006.txt deleted file mode 100644 index 9819386..0000000 --- a/tests/assets/yolo_input_file/raise-the-flag_frame00006.txt +++ /dev/null @@ -1 +0,0 @@ -8 0.540625 0.42291666666666666 0.325 0.0875 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00007.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00007.txt deleted file mode 100644 index 47f718b..0000000 --- a/tests/assets/yolo_input_file/raise-the-flag_frame00007.txt +++ /dev/null @@ -1 +0,0 @@ -8 0.528125 0.428125 0.325 0.08541666666666667 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00008.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00008.txt deleted file mode 100644 index d1907a0..0000000 --- a/tests/assets/yolo_input_file/raise-the-flag_frame00008.txt +++ /dev/null @@ -1 +0,0 @@ -8 0.51875 0.4322916666666667 0.325 0.08541666666666667 diff --git a/tests/assets/yolo_input_file/raise-the-flag_frame00009.txt b/tests/assets/yolo_input_file/raise-the-flag_frame00009.txt deleted file mode 100644 index ae85f6b..0000000 --- a/tests/assets/yolo_input_file/raise-the-flag_frame00009.txt +++ /dev/null @@ -1 +0,0 @@ -8 0.50703125 0.42916666666666664 0.3265625 0.08333333333333333