Python之修改图片像素值

在做语义分割项目时,标注的图片不合标准,而且类型是RGBA型,且是A的部分表示的类别,因此需要将该图片转化为RGB图片

# -*- coding:utf8 -*-
import os

from PIL import Image
im = Image.open('123.png') #打开图片
pix = im.load() #导入像素
width = im.size[0] #获取宽度
height = im.size[1] #获取长度

for x in range(width):
    for y in range(height):
        r,g,b,a = im.getpixel((x,y))	
        rgba=(r,g,b,a)
        if(a == 0):
            im.putpixel((x,y),(0,0,0,0))
        if(a == 255):
            im.putpixel((x,y),(255,255,255,255))

im = im.convert('RGB')
im.save('456.png')

批量处理的方法

# -*- coding:utf8 -*-
import os
from PIL import Image

path = 'SegmentationClass(RGBA)/'
savedpath = 'SegmentationClass/'
filelist = os.listdir(path)
for item in filelist:
    im = Image.open(path + item) #打开图片
    width = im.size[0] #获取宽度
    height = im.size[1] #获取长度

    for x in range(width):
        for y in range(height):
            r,g,b,a = im.getpixel((x,y))	
            if(a == 0):
                im.putpixel((x,y),(0,0,0,0))
            if(a == 255):
                im.putpixel((x,y),(255,255,255,255))
    im = im.convert('RGB')
    im.save(savedpath + item)
    print('item of %s is saved '%(item))

 

 
  • 5
    点赞
  • 7
    评论
  • 12
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

打赏
文章很值,打赏犒劳作者一下
相关推荐
©️2020 CSDN 皮肤主题: 游动-白 设计师:白松林 返回首页

打赏

ZONG_XP

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者