From 735822cffd6057693305ff2aa3199720309e97ca Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Thu, 27 Jun 2019 19:49:40 +0800 Subject: [PATCH] fix infer not visualize (#2578) --- PaddleCV/object_detection/tools/infer.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/PaddleCV/object_detection/tools/infer.py b/PaddleCV/object_detection/tools/infer.py index be43ea41..c4ee32d3 100644 --- a/PaddleCV/object_detection/tools/infer.py +++ b/PaddleCV/object_detection/tools/infer.py @@ -56,6 +56,10 @@ def get_test_images(infer_dir, infer_img): """ assert infer_img is not None or infer_dir is not None, \ "--infer_img or --infer_dir should be set" + assert infer_img is None or os.path.isfile(infer_img), \ + "{} is not a file".format(infer_img) + assert infer_dir is None or os.path.isdir(infer_dir), \ + "{} is not a directory".format(infer_dir) images = [] # infer_img has a higher priority @@ -162,8 +166,8 @@ def main(): for im_id in im_ids: image_path = imid2path[int(im_id)] image = Image.open(image_path).convert('RGB') - visualize_results(image, int(im_id), catid2name, 0.5, bbox_results, - mask_results, is_bbox_normalized) + image = visualize_results(image, int(im_id), catid2name, 0.5, + bbox_results, mask_results, is_bbox_normalized) save_name = get_save_image_name(FLAGS.output_dir, image_path) logger.info("Detection bbox results save in {}".format(save_name)) image.save(save_name) -- GitLab