修复 端到端算法调用inference模型时与训练模型结果不一致的问题,由rgb3个通道的顺序不一致引起
This commit is contained in:
parent
63ed5fcab3
commit
97987978cf
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue