[debug]PIL.Image.open()读出来的图片只有两维

  • Post author:
  • Post category:其他


现象:如果图像位深度为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 版权协议,转载请附上原文出处链接和本声明。