Image.open(x).convert(‘RGB‘)

在pytorch自定义数据集的时候,需要使用到如下代码

Image.open(x).convert('RGB')

使用Image.open读出图像即可,为什么还需要使用convert(‘RGB’)转换成RGB,难道Image.open()读出的彩色图像不是RGB吗

使用如下代码进行测试:

img = Image.open('pokeman\\bulbasaur\\00000000.png').convert('RGB')
img2 = Image.open('pokeman\\bulbasaur\\00000000.png')
print(img)
print(img2)
print(img.shape)
print(img2.shape)

输出结果:

<PIL.Image.Image image mode=RGB size=900x900 at 0x15D0609AF28>
<PIL.PngImagePlugin.PngImageFile image mode=RGBA size=900x900 at 0x15D08BD6550>
(3, 900, 900)
(4, 900, 900)

可以看到,如果不使用.convert(‘RGB’)进行转换的话,读出来的图像是RGBA四通道的,A通道为透明通道,该对深度学习模型训练来说暂时用不到,因此使用convert(‘RGB’)进行通道转换。

  • 30
    点赞
  • 46
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值