From 835bfa456c0617c31313773fa9d7a57b6216f223 Mon Sep 17 00:00:00 2001
From: WenmuZhou <zjwenmu@gmail.com>
Date: Fri, 27 Nov 2020 15:29:14 +0800
Subject: [PATCH] Remove the channel conversion of the image and keep it
 consistent with the training phase

---
 tools/infer/predict_det.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py
index 69db67db..efda0e46 100755
--- a/tools/infer/predict_det.py
+++ b/tools/infer/predict_det.py
@@ -174,7 +174,6 @@ if __name__ == "__main__":
         if img is None:
             logger.info("error in loading image:{}".format(image_file))
             continue
-        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
         dt_boxes, elapse = text_detector(img)
         if count > 0:
             total_time += elapse
-- 
GitLab