Python实战——为人脸照片添加口罩

import os

from PIL import Image

# 源目录

# MyPath = 'E:/play/FaceMask_CelebA-master/facemask_image/'

MyPath = 'E:/play/FaceMask_CelebA-master/save/masks/'

# 输出目录

OutPath = 'E:/play/FaceMask_CelebA-master/save/Binarization/'

def processImage(filesoure, destsoure, name, imgtype):

    '''

    filesoure是存放待转换图片的目录

    destsoure是存在输出转换后图片的目录

    name是文件名

    imgtype是文件类型

    '''

    imgtype = 'bmp' if imgtype == '.bmp' else 'png'

    # 打开图片

    im = Image.open(filesoure + name)

    # =============================================================================

    #     #缩放比例

    #     rate =max(im.size[0]/640.0 if im.size[0] > 60 else 0, im.size[1]/1136.0 if im.size[1] > 1136 else 0)

    #     if rate:

    #         im.thumbnail((im.size[0]/rate, im.size[1]/rate))

    # =============================================================================

    img = im.convert("RGBA")

    pixdata = img.load()

    # 二值化

    for y in range(img.size[1]):

        for x in range(img.size[0]):

            if pixdata[x, y][0] < 90:

                pixdata[x, y] = (0, 0, 0, 255)

    for y in range(img.size[1]):

        for x in range(img.size[0]):

            if pixdata[x, y][1] < 136:

                pixdata[x, y] = (0, 0, 0, 255)

    for y in range(img.size[1]):

        for x in range(img.size[0]):

            if pixdata[x, y][2] > 0:

                pixdata[x, y] = (255, 255, 255, 255)

    img.save(destsoure + name, imgtype)

def run():

    # 切换到源目录，遍历源目录下所有图片

    os.chdir(MyPath)

    for i in os.listdir(os.getcwd()):

        # 检查后缀

        postfix = 跟单网gendan5.comos.path.splitext(i)[1]

        name = os.path.splitext(i)[0]

        name2 = name.split('.')

        if name2[1] == 'jpg-binary' or name2[1] == 'png-binary':

            processImage(MyPath, OutPath, i, postfix)

if __name__ == '__main__':

    run()