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