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

This commit is contained in:
before31 2021-08-31 22:56:08 +08:00
parent 63ed5fcab3
commit 97987978cf
1 changed files with 1 additions and 0 deletions

View File

@ -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