现象:如果图像位深度为24则正常的HWC,为8则读出来只有两维HW,没有C了。
解决:只需要convert成RGB即可解决:
mg = PIL.Image.open('img.jpg')
if img.mode != 'RGB':
img = img.convert('RGB') # this is important
img = torch.from_numpy(np.array(img))
img = img.permute(2, 0, 1)
img = img[:3, :, :]
版权声明:本文为qq_34342853原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。