浏览代码

Pass data modulo as int

Jérôme BUISINE 3 年之前
父节点
当前提交
e9a28342dc
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      run/extract_specific_modulo.py

+ 1 - 1
run/extract_specific_modulo.py

@@ -8,7 +8,7 @@ def main():
     parser = argparse.ArgumentParser(description="Extract specific samples indices")
 
     parser.add_argument('--folder', type=str, help='folder with all rawls files', required=True)
-    parser.add_argument('--modulo', type=str, help='every image to take', required=True)
+    parser.add_argument('--modulo', type=int, help='every image to take', required=True)
     parser.add_argument('--output', type=str, help='folder with all png files', required=True)
 
     args = parser.parse_args()