Chinaunix首页 | 论坛 | 博客
  • 博客访问: 3257011
  • 博文数量: 284
  • 博客积分: 0
  • 博客等级: 民兵
  • 技术积分: 2099
  • 用 户 组: 普通用户
  • 注册时间: 2019-10-28 13:40
文章分类

全部博文(284)

文章存档

2022年(57)

2021年(155)

2020年(50)

2019年(22)

我的朋友

分类: Python/Ruby

2022-05-19 12:29:48

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()

阅读(128) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~