图片切碎片脚本 python PIL库实践

　　python PIL库实践运用，对图像进行切碎片操作。

　　原图如图一

图一

　　我们想要的是图片的不同部分，比如图二中1、2、3、4每一个分区单独的碎片，

图二

　　做法是做出4张跟原图大小一样的碎片模版图，白底，想要的区域涂黑（非纯白）即可。1区域的碎片模板图如图三所示：

图三

　　脚本处理图片的方法：先找出碎片模版中的非白区域（可以不规则），然后将原图中这个区域内的所有像素点的颜色都放到新的一张跟原图像素大小一样的新图上，新图的其余位置都设置成透明（可以根据自己的不同需求进行不同的调整）。 1碎片模版得到的新图碎片如图四所示：

图四

　　附上处理脚本

　　

# -*- coding=utf-8 -*-
from PIL import Image
import os
import shutil

SUFIX_PNG = "png"
SUFIX_JPEG = "jpeg"
SUFIX_LIST = [SUFIX_PNG, SUFIX_JPEG]
NEED_RESIZE_TAG = True
SMALL_SIZE = (126,126)


def isWhite(color):
    r, g, b = color[0], color[1], color[2]
    if r == 255 and g == 255 and  b==255:
        return True
    return False

def getImgFileList(folderPath):
    modelFileNames = []
    for dirpath, dirnames, filenames in os.walk(folderPath):
        for filename in filenames:
            sufix = os.path.splitext(filename)[1][1:]
            if sufix in SUFIX_LIST:
                modelFileNames.append(filename)
    modelFileNames.sort()
    return modelFileNames

def getFileName(fileFullName):
    return fileFullName[:fileFullName.rindex(".")]

def checkImgSize(sizeDict, widthSum, heightSum, imgCnt):
    # imgCnt = len(sizeDict.keys())
    averWidth, averHeight = widthSum/imgCnt, heightSum/imgCnt
    for sizeKey, sizeValue in sizeDict.items():
        if sizeValue['width'] != averWidth:
            print "[ERROE],{IMG} width is wrong! please check it again!".format(IMG=sizeKey)
            return False
        if sizeValue['height'] != averHeight:
            print "[ERROE, ]{IMG} height is wrong!please check it again!".format(IMG=sizeKey)
            return False
    return True


def test():
    folderPath = './model/'
    srcPath = './src/'
    srcBackupPath = './srcBackup/'
    targetPath = './done/'
    currPath = '.'
    # check model &src folder 
    if not os.path.exists(folderPath) or not os.path.exists(srcPath):
        print '[ERROR], dir [model] or [src] does not exist'
        return

    # creat the target folder
    if not os.path.exists(targetPath):
        os.mkdir(targetPath)

    partImgNames = getImgFileList(folderPath)
    srcImgNames = getImgFileList(srcPath)

    partDict = {}
    srcDict = {}
    sizeDict = {}
    imgCnt = 0
    widthSum, heightSum = 0, 0

    # change src  img file sufix 把之前的png原图改成jpeg格式
    # 需不需要改格式
    # for _, srcImgName in enumerate(srcImgNames):
    #     if srcImgName.endswith(SUFIX_PNG):
    #         im = Image.open(srcPath+srcImgName)
    #         bg = Image.new("RGB", im.size, (255,255,255))
    #         newSrcImgName = srcImgName.replace(SUFIX_PNG, SUFIX_JPEG)
    #         bg.paste(im, im)
    #         bg.save(srcPath+newSrcImgName)
    #         if os.path.exists(srcPath+srcImgName):
    #             if not os.path.exists(srcBackupPath):
    #                 os.mkdir(srcBackupPath)
    #             shutil.move(srcPath+srcImgName, srcBackupPath)

    srcImgNames = getImgFileList(srcPath)
    for index, partImgName in enumerate(partImgNames):
        partDict[index] = Image.open(folderPath+partImgName)
        width, height = partDict[index].size
        sizeDict[partImgName] = {}
        sizeDict[partImgName]['width'] = width
        sizeDict[partImgName]['height'] = height
        widthSum += width
        heightSum += height
        imgCnt += 1

    for index, srcImgName in enumerate(srcImgNames):
        srcDict[index] = Image.open(srcPath+srcImgName)
        width, height = srcDict[index].size
        sizeDict[srcImgName] = {}
        sizeDict[srcImgName]['width'] = width
        sizeDict[srcImgName]['height'] = height
        widthSum += width
        heightSum += height
        imgCnt += 1

    # img size check, 检查图片尺寸
    if not checkImgSize(sizeDict, widthSum, heightSum, imgCnt):
        return 

    for srcIndex, srcImgName in enumerate(srcImgNames):
        tempSrcImg = srcDict[srcIndex].copy()
        colorDataSrc1 = tempSrcImg.getdata()

        for modelIndex, partImgName in enumerate(partImgNames):
            tempImg = partDict[modelIndex].copy() # copy img , 复制图片
            colorData2 = tempImg.getdata()
            colorStore = []  # to save colors
            for index, color2 in enumerate (colorData2):
                if not isWhite(color2):
                    colorStore.append(colorDataSrc1[index])
                else:
                    colorStore.append((0,0,0,0))

            tempImg.putdata(colorStore)
            # make target files names
            newFileName = "{TARGET}{NAME}-{INDEX}.{SUFIX}".format(
                TARGET=targetPath, 
                NAME=getFileName(srcImgName),
                INDEX=modelIndex+1, 
                SUFIX=SUFIX_PNG)
            tempImg.save(newFileName, SUFIX_PNG)

            if NEED_RESIZE_TAG:
                smallTargetPath = './smalldone/'
                if not os.path.exists(smallTargetPath):
                    os.mkdir(smallTargetPath)
                smallImg = tempImg.resize(SMALL_SIZE)
                newFileName = "{TARGET}{NAME}-{INDEX}.{SUFIX}".format(
                TARGET=smallTargetPath, 
                NAME=getFileName(srcImgName),
                INDEX=modelIndex+1, 
                SUFIX=SUFIX_PNG)
                smallImg.save(newFileName, SUFIX_PNG)
            print '------create {NAME} done-----'.format(NAME=newFileName)
    print '>>>>>>>>>>>>>>>>>Job Done<<<<<<<<<<<<<<<<'


if __name__ == "__main__":
    test()

　　目录组织结构如图五所示：

图五

　　最终done文件夹的结果如图六所示：

图六

 

　　Note:

1. 原图放在src文件夹下，原图可以有很多张，脚本对图片的宽高尺寸有检查，如果批量处理的话，尺寸必须保持一致，不然的话需要注掉尺寸检查的代码。
2. 碎片模版图放在model文件夹下，所有碎片模版的尺寸必须跟原图尺寸一致。
3. 生成的碎片会存放到done文件夹里。
4. 支持resize功能，将resize后的新碎片存到smalldone文件夹下。