Python-png转rgb

Posted 2021-06-28 Devil love genius

在针对深度学习方面，数据集的好与坏时时刻刻决定着你的模型的好与坏；

png是有透明度，不能用于模型训练

图像处理更是十分的重要，在机器学习（深度学习）当中70%的时间在数据预处理，20%在模型的调优调参，10%在于模型的检验；
因此数据预处理十分重要的；
下面将为大家讲解图像格式的转换

PNG-----》RBG

from PIL import Image
import cv2 as cv
import os
 
def PNG_JPG(PngPath):
    img = cv.imread(PngPath, 0)
    w, h = img.shape[::-1]
    infile = PngPath
    outfile = os.path.splitext(infile)[0] + ".jpg"
    img = Image.open(infile)
    img = img.resize((int(w / 2), int(h / 2)), Image.ANTIALIAS)
    try:
        if len(img.split()) == 4:
            # prevent IOError: cannot write mode RGBA as BMP
            r, g, b, a = img.split()
            img = Image.merge("RGB", (r, g, b))
            img.convert('RGB').save(outfile, quality=70)
            os.remove(PngPath)
        else:
            img.convert('RGB').save(outfile, quality=70)
            os.remove(PngPath)
        return outfile
    except Exception as e:
        print("PNG转换JPG 错误", e)

转化一张图：

img='1.png'
Path='./'
if img.endswith('.png'):
    PngPath= Path + img
    PNG_JPG(PngPath)

循环遍历多张图像和转换：

Path="../../signs"
img_dir = os.listdir(Path)
for img in img_dir:
            if img.endswith('.png'):
                PngPath= os.path.join(path_2,img)
                print(PngPath)
                PNG_JPG(PngPath)

希望这篇文章对你有用！
谢谢点赞评论！