Remove the channel conversion of the image and keep it consistent with the training phase

4 years ago · 835bfa456c
parent 05dcfdec68
commit 835bfa456c
1 changed files with 0 additions and 1 deletions
--- 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