Parcourir la source

use of image index instead

Jérôme BUISINE il y a 3 ans
Parent
commit
3950fbb546
1 fichiers modifiés avec 4 ajouts et 2 suppressions
  1. 4 2
      run/extract_specific_modulo.py

+ 4 - 2
run/extract_specific_modulo.py

@@ -19,9 +19,11 @@ def main():
 
     images_path = glob.glob(f"{p_folder}/**/**/*.rawls")
 
-    for i, img in enumerate(sorted(images_path)):
+    for img in sorted(images_path):
 
-        if ((i + 1) % p_modulo) == 0:
+        img_index = int(img.split('-').split('.')[0])
+
+        if ((img_index + 1) % p_modulo) == 0:
 
             output_path = img.replace(p_folder, p_output)
             output_folder, _ = os.path.split(output_path)