From 55492911dd74a95480d4af214c98f813eb8b0584 Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Thu, 14 May 2020 14:38:23 +0800
Subject: [PATCH] RGB 2 BGR

---
 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 48d3e54e..f4e04adc 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))))
-- 
GitLab