diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py
index 48d3e54eb8de88e2f252bdca05baacc02e354952..f4e04adc5ad90f7cd640be5ddbdd49d535b43970 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))))