diff --git a/tools/infer/predict_e2e.py b/tools/infer/predict_e2e.py index 5029d6059346a00062418d8d1b6cb029b0110643..8ff279d7437965f725082a9eb1c83e05a7ffc8a8 100755 --- a/tools/infer/predict_e2e.py +++ b/tools/infer/predict_e2e.py @@ -141,6 +141,7 @@ if __name__ == "__main__": img, flag = check_and_read_gif(image_file) if not flag: img = cv2.imread(image_file) + img = img[:, :, ::-1] if img is None: logger.info("error in loading image:{}".format(image_file)) continue