From abe1183dbba727c2e327a7a496d768e998d1c164 Mon Sep 17 00:00:00 2001 From: wuzewu Date: Wed, 10 Apr 2019 14:59:32 +0800 Subject: [PATCH] Fix compatibility bug in ssd module demo --- demo/ssd/processor.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/demo/ssd/processor.py b/demo/ssd/processor.py index 4b9aea86..7d55d9c9 100644 --- a/demo/ssd/processor.py +++ b/demo/ssd/processor.py @@ -29,11 +29,12 @@ def draw_bounding_box_on_image(image_path, data_list, save_path): if image.mode == 'RGB': draw.text((left, top), data['label'], (255, 255, 0)) - image_name = image_path.split('/')[-1] + image_name = os.path.split(image_path)[-1] if not os.path.exists(save_path): os.mkdir(save_path) save_path = os.path.join(save_path, image_name) - print("image with bbox drawed saved as {}".format(save_path)) + print("image with bbox drawed saved as {}".format( + os.path.abspath(save_path))) image.save(save_path) -- GitLab