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

 

posted @ 2022-03-25 17:31  Tomorrow1126  阅读(143)  评论(0编辑  收藏  举报