diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py
index e8aea452e49c37637393d8bf98c60fe2f445bd4f..300ba111905aeb39fae2bed818c2b53611ed9242 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)
+                draw_img[:, :, -1])
             print("The visualized image saved in {}".format(
                 os.path.join(draw_img_save, os.path.basename(image_file))))