image.open 和 cv2.imread打开的图像格式不同!!!在跑模型的时候数据加载遇到的bug解决


tsm的dataloader,报错就是因为这个,原来的代码是image.open,我用的cv,转格式解决。

转换:

1.Image对象->np.adarray

img = Image.open(path)

img_array = np.array(img)

 

2.np.adarray->Image对象

img = cv2.imread(path)

img_Image = Image.fromarray(np.uint8(img))