pythonPIL 批量绘制图片矩形框工具

Posted 2020-08-23

工具采用PIL：Python Imaging Library,图像处理标准库。PIL功能非常强大，但API却非常简单易用。

安装PIL

在Debian/Ubuntu Linux下直接通过apt安装

$ sudo apt-get install python-imaging

Windows平台直接通过pip安装

pip install pillow

 

批量工具脚本

默认执行方式为:              执行脚本命令 python drawline.py           1.获取当前路径下的‘png‘,‘jpg‘文件         2.绘制宽高占比为0.5,0.5的矩形框         3.保存图片至当前路径下的line文件夹

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81

# -*- coding: utf-8 -*- from PIL import Image, ImageDraw import os, sys     def drawLine(im, width, height):     ‘‘‘     在图片上绘制矩形图     :param im: 图片     :param width: 矩形宽占比     :param height: 矩形高占比     :return:     ‘‘‘     draw = ImageDraw.Draw(im)     image_width = im.size[0]     image_height = im.size[1]     line_width = im.size[0] * width     line_height = im.size[1] * height       draw.line(         ((image_width - line_width) / 2, (image_height - line_height) / 2,          (image_width + line_width) / 2, (image_height - line_height) / 2),         fill=128)     draw.line(         ((image_width - line_width) / 2, (image_height - line_height) / 2,          (image_width - line_width) / 2, (image_height + line_height) / 2),         fill=128)     draw.line(         ((image_width + line_width) / 2, (image_height - line_height) / 2,          (image_width + line_width) / 2, (image_height + line_height) / 2),         fill=128)     draw.line(         ((image_width - line_width) / 2, (image_height + line_height) / 2,          (image_width + line_width) / 2, (image_height + line_height) / 2),         fill=128)     del draw     def endWith(s, *endstring):     ‘‘‘     过滤文件扩展名     :param s: 文件名     :param endstring: 所需过滤的扩展名     :return:     ‘‘‘     array = map(s.endswith, endstring)     if True in array:         return True     else:         return False     if __name__ == ‘__main__‘:     ‘‘‘     默认执行方式为:         1.获取当前路径下的‘png‘,‘jpg‘文件         2.绘制宽高占比为0.5,0.5的矩形框         3.保存图片至当前路径下的line文件夹     ‘‘‘     line_w = 0.5     line_h = 0.5     try:         if sys.argv[1]:             line_w = float(sys.argv[1])         if sys.argv[2]:             line_h = float(sys.argv[2])     except IndexError:         pass     current_path = os.getcwd()     save_path = os.path.join(current_path, ‘line‘)     file_list = os.listdir(current_path)     for file_one in file_list:         # endWith(file_one, ‘.png‘, ‘.jpg‘) 第二个参数后为过滤格式 以 , 分割         if endWith(file_one, ‘.png‘, ‘.jpg‘):             im = Image.open(file_one)             # drawLine(im,line_w, line_h) 后面两位参数为矩形图宽高占比             drawLine(im, line_w, line_h)             if not os.path.exists(save_path):                 os.mkdir(save_path)             im.save(                 os.path.join(save_path, str(file_one.split(‘.‘)[-2]) + ‘_line.‘ + str(file_one.split(‘.‘)[-1])))