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

Posted 2020-09-30

 

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

　　原图如图一

图一

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

图二

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

图三

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

图四

　　附上处理脚本

　　

  1 # -*- coding=utf-8 -*-
  2 from PIL import Image
  3 import os
  4 import shutil
  5 
  6 SUFIX_PNG = "png"
  7 SUFIX_JPEG = "jpeg"
  8 SUFIX_LIST = [SUFIX_PNG, SUFIX_JPEG]
  9 NEED_RESIZE_TAG = True
 10 SMALL_SIZE = (126,126)
 11 
 12 
 13 def isWhite(color):
 14     r, g, b = color[0], color[1], color[2]
 15     if r == 255 and g == 255 and  b==255:
 16         return True
 17     return False
 18 
 19 def getImgFileList(folderPath):
 20     modelFileNames = []
 21     for dirpath, dirnames, filenames in os.walk(folderPath):
 22         for filename in filenames:
 23             sufix = os.path.splitext(filename)[1][1:]
 24             if sufix in SUFIX_LIST:
 25                 modelFileNames.append(filename)
 26     modelFileNames.sort()
 27     return modelFileNames
 28 
 29 def getFileName(fileFullName):
 30     return fileFullName[:fileFullName.rindex(".")]
 31 
 32 def checkImgSize(sizeDict, widthSum, heightSum, imgCnt):
 33     # imgCnt = len(sizeDict.keys())
 34     averWidth, averHeight = widthSum/imgCnt, heightSum/imgCnt
 35     for sizeKey, sizeValue in sizeDict.items():
 36         if sizeValue[‘width‘] != averWidth:
 37             print "[ERROE],{IMG} width is wrong! please check it again!".format(IMG=sizeKey)
 38             return False
 39         if sizeValue[‘height‘] != averHeight:
 40             print "[ERROE, ]{IMG} height is wrong!please check it again!".format(IMG=sizeKey)
 41             return False
 42     return True
 43 
 44 
 45 def test():
 46     folderPath = ‘./model/‘
 47     srcPath = ‘./src/‘
 48     srcBackupPath = ‘./srcBackup/‘
 49     targetPath = ‘./done/‘
 50     currPath = ‘.‘
 51     # check model &src folder 
 52     if not os.path.exists(folderPath) or not os.path.exists(srcPath):
 53         print ‘[ERROR], dir [model] or [src] does not exist‘
 54         return
 55 
 56     # creat the target folder
 57     if not os.path.exists(targetPath):
 58         os.mkdir(targetPath)
 59 
 60     partImgNames = getImgFileList(folderPath)
 61     srcImgNames = getImgFileList(srcPath)
 62 
 63     partDict = {}
 64     srcDict = {}
 65     sizeDict = {}
 66     imgCnt = 0
 67     widthSum, heightSum = 0, 0
 68 
 69     # change src  img file sufix 把之前的png原图改成jpeg格式
 70     # 需不需要改格式
 71     # for _, srcImgName in enumerate(srcImgNames):
 72     #     if srcImgName.endswith(SUFIX_PNG):
 73     #         im = Image.open(srcPath+srcImgName)
 74     #         bg = Image.new("RGB", im.size, (255,255,255))
 75     #         newSrcImgName = srcImgName.replace(SUFIX_PNG, SUFIX_JPEG)
 76     #         bg.paste(im, im)
 77     #         bg.save(srcPath+newSrcImgName)
 78     #         if os.path.exists(srcPath+srcImgName):
 79     #             if not os.path.exists(srcBackupPath):
 80     #                 os.mkdir(srcBackupPath)
 81     #             shutil.move(srcPath+srcImgName, srcBackupPath)
 82 
 83     srcImgNames = getImgFileList(srcPath)
 84     for index, partImgName in enumerate(partImgNames):
 85         partDict[index] = Image.open(folderPath+partImgName)
 86         width, height = partDict[index].size
 87         sizeDict[partImgName] = {}
 88         sizeDict[partImgName][‘width‘] = width
 89         sizeDict[partImgName][‘height‘] = height
 90         widthSum += width
 91         heightSum += height
 92         imgCnt += 1
 93 
 94     for index, srcImgName in enumerate(srcImgNames):
 95         srcDict[index] = Image.open(srcPath+srcImgName)
 96         width, height = srcDict[index].size
 97         sizeDict[srcImgName] = {}
 98         sizeDict[srcImgName][‘width‘] = width
 99         sizeDict[srcImgName][‘height‘] = height
100         widthSum += width
101         heightSum += height
102         imgCnt += 1
103 
104     # img size check, 检查图片尺寸
105     if not checkImgSize(sizeDict, widthSum, heightSum, imgCnt):
106         return 
107 
108     for srcIndex, srcImgName in enumerate(srcImgNames):
109         tempSrcImg = srcDict[srcIndex].copy()
110         colorDataSrc1 = tempSrcImg.getdata()
111 
112         for modelIndex, partImgName in enumerate(partImgNames):
113             tempImg = partDict[modelIndex].copy() # copy img , 复制图片
114             colorData2 = tempImg.getdata()
115             colorStore = []  # to save colors
116             for index, color2 in enumerate (colorData2):
117                 if not isWhite(color2):
118                     colorStore.append(colorDataSrc1[index])
119                 else:
120                     colorStore.append((0,0,0,0))
121 
122             tempImg.putdata(colorStore)
123             # make target files names
124             newFileName = "{TARGET}{NAME}-{INDEX}.{SUFIX}".format(
125                 TARGET=targetPath, 
126                 NAME=getFileName(srcImgName),
127                 INDEX=modelIndex+1, 
128                 SUFIX=SUFIX_PNG)
129             tempImg.save(newFileName, SUFIX_PNG)
130 
131             if NEED_RESIZE_TAG:
132                 smallTargetPath = ‘./smalldone/‘
133                 if not os.path.exists(smallTargetPath):
134                     os.mkdir(smallTargetPath)
135                 smallImg = tempImg.resize(SMALL_SIZE)
136                 newFileName = "{TARGET}{NAME}-{INDEX}.{SUFIX}".format(
137                 TARGET=smallTargetPath, 
138                 NAME=getFileName(srcImgName),
139                 INDEX=modelIndex+1, 
140                 SUFIX=SUFIX_PNG)
141                 smallImg.save(newFileName, SUFIX_PNG)
142             print ‘------create {NAME} done-----‘.format(NAME=newFileName)
143     print ‘>>>>>>>>>>>>>>>>>Job Done<<<<<<<<<<<<<<<<‘
144 
145 
146 if __name__ == "__main__":
147     test()

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

图五

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

图六

 

　　Note:

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