Case insensitive search on regex image extensions

sglvladi · Jun 8, 2021 · 4ab0937 · 4ab0937
1 parent c2cc56c
commit 4ab0937
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/source/scripts/partition_dataset.py b/docs/source/scripts/partition_dataset.py
@@ -32,7 +32,7 @@ def iterate_dir(source, dest, ratio, copy_xml):
         os.makedirs(test_dir)
 
     images = [f for f in os.listdir(source)
-              if re.search(r'([a-zA-Z0-9\s_\\.\-\(\):])+(.jpg|.jpeg|.png)$', f)]
+              if re.search(r'([a-zA-Z0-9\s_\\.\-\(\):])+(?i)(.jpg|.jpeg|.png)$', f)]
 
     num_images = len(images)
     num_test_images = math.ceil(ratio*num_images)
@@ -95,4 +95,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main()