首页 / 操作系统 / Linux / 用Python切掉图片的黑边

手机里有很多G+里的截屏，都有一些黑边。像这样：因为原图片的长宽不一定，导致黑边的大小不一定。虽然图片都是居中的，但是有些图片是宽比高大，是横着截图的，所以这样的情况是上下左右都可能有黑边。于是想到了用python的 PIL库来进行裁切。研究了一下，原代码如下 

1. from PIL import Image
2. import os
4. src_folder = "."
5. tar_folder = "tar"
6. backup_folder = "backup"
8. def isCrust（pix）:
9. return sum（pix） < 25
11. def hCheck（img, y, step = 50）:
12. count = 0
13. width = img.size[0]
14. for x in xrange（0, width, step）:
15. if isCrust（img.getpixel（（x, y）））:
16. count += 1
17. if count > width / step / 2:
18. returnTrue
19. returnFalse
21. def vCheck（img, x, step = 50）:
22. count = 0
23. height = img.size[1]
24. for y in xrange（0, height, step）:
25. if isCrust（img.getpixel（（x, y）））:
26. count += 1
27. if count > height / step / 2:
28. returnTrue
29. returnFalse
31. def boundaryFinder（img,crust_side,core_side,checker）:
32. ifnot checker（img,crust_side）:
33. return crust_side
34. if checker（img,core_side）:
35. return core_side
37. mid = （crust_side + core_side） / 2
38. while mid ！= core_side and mid ！= crust_side:
39. if checker（img,mid）:
40. crust_side = mid
41. else:
42. core_side = mid
43. mid = （crust_side + core_side） / 2
44. return core_side
45. pass
47. def handleImage（filename,tar）:
48. img = Image.open（os.path.join（src_folder,filename））
49. if img.mode ！= "RGB":
50. img = img.convert（"RGB"）
51. width, height = img.size
53. left = boundaryFinder（img, 0, width/2, vCheck）
54. right = boundaryFinder（img, width-1, width/2, vCheck）
55. top = boundaryFinder（img, 0, height/2, hCheck）
56. bottom = boundaryFinder（img, height-1, width/2, hCheck）
58. rect = （left,top,right,bottom）
59. print rect
60. region = img.crop（rect）
61. region.save（os.path.join（tar,filename）,"PNG"）
62. pass
64. def folderCheck（foldername）:
65. if foldername:
66. ifnot os.path.exists（foldername）:
67. os.mkdir（foldername）
68. print"Info: Folder "%s" created" % foldername
69. elifnot os.path.isdir（foldername）:
70. print"Error: Folder "%s" conflict" % foldername
71. returnFalse
72. returnTrue
73. pass
75. def main（）:
76. if folderCheck（tar_folder） and folderCheck（src_folder） and folderCheck（backup_folder）:
77. for filename in os.listdir（src_folder）:
78. if filename.split（"."）[-1].upper（） in （"JPG","JPEG","PNG","BMP","GIF"）:
79. handleImage（filename,tar_folder）
80. os.rename（os.path.join（src_folder,filename）,os.path.join（backup_folder,filename））
81. pass
84. if __name__ == "__main__":
85. main（）
86. #handleImage（"Screenshot_2013-10-13-21-55-14.png",""）IL import Imageimport os src_folder = "."tar_folder = "tar"backup_folder = "backup"def isCrust（pix）:return sum（pix） < 25def hCheck（img, y, step = 50）:count = 0width = img.size[0]for x in xrange（0, width, step）:if isCrust（img.getpixel（（x, y）））:count += 1if count > width / step / 2:return Truereturn Falsedef vCheck（img, x, step = 50）:count = 0height = img.size[1]for y in xrange（0, height, step）:if isCrust（img.getpixel（（x, y）））:count += 1if count > height / step / 2:return Truereturn Falsedef boundaryFinder（img,crust_side,core_side,checker）:if not checker（img,crust_side）:return crust_sideif checker（img,core_side）:return core_sidemid = （crust_side + core_side） / 2whilemid ！= core_side and mid ！= crust_side:if checker（img,mid）:crust_side = midelse:core_side = midmid = （crust_side + core_side） / 2return core_sidepassdef handleImage（filename,tar）:img = Image.open（os.path.join（src_folder,filename））if img.mode ！= "RGB":img = img.convert（"RGB"）width, height = img.sizeleft = boundaryFinder（img, 0, width/2, vCheck）right = boundaryFinder（img, width-1, width/2, vCheck）top = boundaryFinder（img, 0, height/2, hCheck）bottom = boundaryFinder（img, height-1, width/2, hCheck）rect = （left,top,right,bottom）print rectregion = img.crop（rect）region.save（os.path.join（tar,filename）,"PNG"）passdef folderCheck（foldername）:if foldername:if not os.path.exists（foldername）:os.mkdir（foldername） print "Info: Folder "%s" created" % foldernameelif not os.path.isdir（foldername）:print "Error: Folder "%s" conflict" % foldernamereturn Falsereturn Truepassdef main（）:if folderCheck（tar_folder） and folderCheck（src_folder） and folderCheck（backup_folder）:for filename in os.listdir（src_folder）:if filename.split（"."）[-1].upper（） in （"JPG","JPEG","PNG","BMP","GIF"）:handleImage（filename,tar_folder）os.rename（os.path.join（src_folder,filename）,os.path.join（backup_folder,filename））passif __name__ == "__main__":main（）#handleImage（"Screenshot_2013-10-13-21-55-14.png",""）

 代码作用是 遍历当前文件夹下的所有JPG PNG BMP GIF文件，找到内容上下左右的边距（实际上是内容矩形左上右下???点的坐标），将切出来的图片保存到目标文件夹中，并备份源文件到备份文件夹中。boundaryFinder（） 函数使用时间复杂度为O（log n）的算法来找到内容的边缘所在的线（可能是横线也可能是竖线，取决于checker使用的是哪一个函数）hCheck（） 和 vCheck（） 函数分别水平和竖直地对图片上的某条线，相等间隔（默认50）的取点。如果这些点有一半被认为属于要被切掉的边，就会返回trueisCrust（）函数判断点是否属于要切掉的边 本例中 RGB三色加起来小于25 就被认为是要黑边的内容效果如下：这个代码适用于有纯色边的图片剪切。通过修改isCrust（） 函数的算法 可以适用于不同纯色边的裁切《Python核心编程 第二版》.（Wesley J. Chun ）.[高清PDF中文版] http://www.linuxidc.com/Linux/2013-06/85425.htm《Python开发技术详解》.（ 周伟,宗杰）.[高清PDF扫描版+随书视频+代码] http://www.linuxidc.com/Linux/2013-11/92693.htmPython脚本获取Linux系统信息 http://www.linuxidc.com/Linux/2013-08/88531.htm在Ubuntu下用Python搭建桌面算法交易研究环境 http://www.linuxidc.com/Linux/2013-11/92534.htmPython 的详细介绍：请点这里
Python 的下载地址：请点这里本文永久更新链接地址：http://www.linuxidc.com/Linux/2014-07/104468.htm