From 1216dbd5dbba67ac1b890c7aa037b1e750a22ecd Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 5 Jun 2017 11:39:53 +0200 Subject: [PATCH] [CCV] Added batch script for saliency system Signed-off-by: Jim Martens --- ccv/saliency/tools/batch_saliency.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 ccv/saliency/tools/batch_saliency.py diff --git a/ccv/saliency/tools/batch_saliency.py b/ccv/saliency/tools/batch_saliency.py new file mode 100644 index 0000000..271ec14 --- /dev/null +++ b/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()