diff --git a/tools/infer/predict_e2e.py b/tools/infer/predict_e2e.py index 5029d605..8ff279d7 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