修复端到端算法调用inference模型时与训练模型结果不一致的问题，由rgb3个通道的顺序不一致引起

2021-08-31 22:56:08 +08:00 · 2021-08-31 22:56:08 +08:00 · 97987978cf
parent 63ed5fcab3
commit 97987978cf
1 changed files with 1 additions and 0 deletions
--- 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