Add other XML path retrieval in regions_from_dataset.py

stefanklut · Jan 23, 2024 · 4e17ed7 · 4e17ed7
1 parent dcce7c1
commit 4e17ed7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/utils/regions_from_dataset.py b/utils/regions_from_dataset.py
@@ -59,6 +59,8 @@ def main(args):
     image_paths = get_file_paths(args.input, supported_image_formats)
     xml_paths = [image_path_to_xml_path(image_path) for image_path in image_paths]
 
+    # xml_paths = get_file_paths(args.input, [".xml"])
+
     # Single thread
     # regions_per_page = []
     # for xml_path_i in tqdm(xml_paths):