Diff of /mediaug/utils.py [000000] .. [05e710]

Switch to side-by-side view

--- a
+++ b/mediaug/utils.py
@@ -0,0 +1,24 @@
+import os
+import pathlib
+
+
+def convert_array_to_poly(arr):
+    """ Convert array of (x,y) coordinates, nx2, to
+    a Polygon object for Polygon package
+    Args:
+      arr (np.array): Array of the polygon
+    Returns:
+      asn (list): The polygon in a list
+    """
+    return arr.flatten().tolist()
+
+
+def create_dirs(base_path, dir_list=[]):
+    """create dirs under the base path
+    Args:
+      base_path (str): The base of the directory paths
+      dir_list (list[str]): List of paths relative to the base dir
+    """
+    pathlib.Path(base_path).mkdir(parents=True, exist_ok=True)
+    for new_dir in dir_list:
+        pathlib.Path(os.path.join(base_path, new_dir)).mkdir(parents=True)