From 657a87972e7d4d500ffd3dd862ab1bdc1e4f9d34 Mon Sep 17 00:00:00 2001 From: jerrywgz Date: Tue, 21 Jan 2020 09:10:56 +0000 Subject: [PATCH] sort image name when inferring dir --- tools/infer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/infer.py b/tools/infer.py index 5e303da4c..35660e591 100644 --- a/tools/infer.py +++ b/tools/infer.py @@ -87,7 +87,8 @@ def get_test_images(infer_dir, infer_img): exts = ['jpg', 'jpeg', 'png', 'bmp'] exts += [ext.upper() for ext in exts] for ext in exts: - images.extend(glob.glob('{}/*.{}'.format(infer_dir, ext))) + image_list = glob.glob('{}/*.{}'.format(infer_dir, ext)) + images.extend(sorted(image_list)) assert len(images) > 0, "no image found in {}".format(infer_dir) logger.info("Found {} inference images in total.".format(len(images))) -- GitLab