From 9686071b35d3fa009a8cab3de564399e925ea58b Mon Sep 17 00:00:00 2001 From: Samir Aguiar Date: Fri, 16 Oct 2020 18:14:15 -0300 Subject: [PATCH] Ignore non-files in the input directory --- Global/detection.py | 6 +++++- Global/test.py | 8 ++++++-- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/Global/detection.py b/Global/detection.py index fb52c65d..87402f59 100644 --- a/Global/detection.py +++ b/Global/detection.py @@ -106,7 +106,11 @@ def main(config): print("processing", image_name) results = [] - scratch_image = Image.open(os.path.join(config.test_path, image_name)).convert("RGB") + scratch_file = os.path.join(config.test_path, image_name) + if not os.path.isfile(scratch_file): + print("Skipping non-file %s" % image_name) + continue + scratch_image = Image.open(scratch_file).convert("RGB") w, h = scratch_image.size diff --git a/Global/test.py b/Global/test.py index 6e162216..cfa808bf 100644 --- a/Global/test.py +++ b/Global/test.py @@ -108,7 +108,7 @@ def parameter_set(opt): dataset_size = 0 input_loader = os.listdir(opt.test_input) - dataset_size = len(os.listdir(opt.test_input)) + dataset_size = len(input_loader) input_loader.sort() if opt.test_mask != "": @@ -124,7 +124,11 @@ def parameter_set(opt): for i in range(dataset_size): input_name = input_loader[i] - input = Image.open(os.path.join(opt.test_input, input_name)).convert("RGB") + input_file = os.path.join(opt.test_input, input_name) + if not os.path.isfile(input_file): + print("Skipping non-file %s" % input_name) + continue + input = Image.open(input_file).convert("RGB") print("Now you are processing %s" % (input_name))