diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index 300ba11190..8b0abd706c 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))))