[CCV] Added batch script for saliency system

Signed-off-by: Jim Martens <github@2martens.de>

ccv/saliency/tools/batch_saliency.py

@@ -0,0 +1,27 @@
+import argparse
+import os
+import subprocess
+
+
+def batch():
+    """
+    Performs a batch call to the saliency system.
+    """
+    parser = argparse.ArgumentParser(description="Calls the saliency system")
+    parser.add_argument("input", metavar="input", type=str, help="The directory of the input images.")
+    parser.add_argument("output", metavar="output", type=str, help="The directory of the saliency maps.")
+    parser.add_argument("saliency", metavar="saliency", type=str, help="The path to the saliency tool.")
+    
+    args = parser.parse_args()
+    input_dir = os.fsencode(args.input)
+    
+    for file in os.listdir(input_dir):
+        filename = os.fsdecode(file)
+        base_filename, ext = os.path.splitext(filename)
+        input_filename = os.path.join(args.input, filename)
+        output_filename = os.path.join(args.output, base_filename + "_saliency.png")
+        subprocess.run([args.saliency, input_filename, output_filename, "1"], stderr=subprocess.PIPE, check=True)
+        
+    
+if __name__ == "__main__":
+    batch()