现象:如果图像位深度为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, :, :]