python 图片处理之转换为RGB图片

文章介绍了在进行语义分割时遇到的图片处理问题,由于深度学习模型通常只支持RGB格式,当图片包含透明通道(如PNG的alpha通道)时,需要将图片转换为只有RGB三个通道的形式。提供了一段Python代码,使用PIL库来检查并转换图片模式为RGB,确保模型能正确处理。代码来源网络,无误且可供日后参考使用。
摘要由CSDN通过智能技术生成
今天在对照片进行语义分割的时候,发现图片分割失败了。看了一下报错信息,才知道大部分的深度学习处理图片的模型由于一般默认支持RGB格式的图片(主要是数组矩阵的格式设置)。如果图片中包含其他的通道(比如透明度),这个时候就需要将图片转换为只有RGB三个通道的照片。这个代码完全可以解决图片转换问题,没有一点错误,这个代码是网上一个大神的,我只是记录分享一下,以后自己能用,如有侵权,请联系我,谢谢。
import os
from PIL import Image
from tqdm import tqdm

img_path = 'F:/bishe/flower_photos/八点灰灯蛾'  # 填入图片所在文件夹的路径
img_Topath = 'F:/bishe/flower_photos/八点灰灯蛾'  # 填入图片转换后的文件夹路径

for item in tqdm(os.listdir(img_path)):
    arr = item.strip().split('*')
    img_name = arr[0]
    image_path = os.path.join(img_path, img_name)
    img = Image.open(image_path)
    if (img.mode != 'RGB'):
        img = img.convert("RGB")
        img.save(img_Topath + '/' + img_name)
  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值