From 27ae32d5e1c36a53b015a845526b25348418798d Mon Sep 17 00:00:00 2001
From: Kyrylo Antoshyn <akirill98@gmail.com>
Date: Tue, 26 Oct 2021 13:39:09 +0300
Subject: [PATCH] Added script to generate text file with dataset images
 relative paths.

---
 .../generate_dataset_images_relative_paths.py | 54 +++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 scripts/generate_dataset_images_relative_paths.py

diff --git a/scripts/generate_dataset_images_relative_paths.py b/scripts/generate_dataset_images_relative_paths.py
new file mode 100644
index 00000000000..930a9b4354f
--- /dev/null
+++ b/scripts/generate_dataset_images_relative_paths.py
@@ -0,0 +1,54 @@
+import sys
+import os
+
+
+def get_dataset_images_relative_paths(dataset_directory_name):
+    """
+    Gets list of paths to the dataset images.
+
+    :param dataset_directory_name: dataset directory name
+    :return: dataset images relative paths
+    """
+    os.chdir("..")
+    os.chdir(os.path.join("data", dataset_directory_name))
+
+    dataset_images_relative_paths = []
+    for dataset_image_name in os.listdir(os.getcwd()):
+        if dataset_image_name.endswith(".jpg"):
+            dataset_images_relative_paths.append(f"data/{dataset_directory_name}/{dataset_image_name}")
+
+    return dataset_images_relative_paths
+
+
+def write_dataset_images_relative_paths(dataset_directory_name, dataset_images_relative_paths):
+    """
+    Writes dataset images relative paths
+
+    :param dataset_directory_name: dataset directory name
+    :param dataset_images_relative_paths: dataset images relative paths
+    """
+    os.chdir("..")
+
+    with open(f"{dataset_directory_name}.txt", "w") as dataset_images_relative_paths_file:
+        for dataset_image_relative_path in dataset_images_relative_paths:
+            dataset_images_relative_paths_file.write(f"{dataset_image_relative_path}\n")
+
+
+def main():
+    """
+    Script entry point.
+    """
+    # Check whether dataset directory name is not provided
+    if len(sys.argv) != 2:
+        sys.exit("You need to provide dataset directory name!")
+
+    # Get dataset directory name
+    dataset_directory_name = sys.argv[1]
+
+    # Write dataset images relative paths
+    write_dataset_images_relative_paths(dataset_directory_name,
+                                        get_dataset_images_relative_paths(dataset_directory_name))
+
+
+if __name__ == "__main__":
+    main()