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