From 3b931056c4b376f0a2fbfaa7f9fdc172ceabcd90 Mon Sep 17 00:00:00 2001 From: dyning Date: Thu, 14 May 2020 14:43:20 +0800 Subject: [PATCH] Update predict_system.py --- tools/infer/predict_system.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index 300ba111..8b0abd70 100755 --- a/tools/infer/predict_system.py +++ b/tools/infer/predict_system.py @@ -137,6 +137,6 @@ if __name__ == "__main__": os.makedirs(draw_img_save) cv2.imwrite( os.path.join(draw_img_save, os.path.basename(image_file)), - draw_img[:, :, -1]) + draw_img[:, :, ::-1]) print("The visualized image saved in {}".format( os.path.join(draw_img_save, os.path.basename(image_file))))