Python中用PIL库批量给图片加上序号的教程
程序员文章站
2022-08-02 14:51:08
女友让我给她论文的图片上加上字母序号,本来觉得是个很简单的事情,但那个白底黑字的圆圈序号却难住了我, 试了几个常用的软件,都不行。
后来用 ps + 动作,倒是能搞出来,...
女友让我给她论文的图片上加上字母序号,本来觉得是个很简单的事情,但那个白底黑字的圆圈序号却难住了我, 试了几个常用的软件,都不行。
后来用 ps + 动作,倒是能搞出来,不过也不容易,正好那天没搞完,于是拿回自己家做,但我的电脑上又没有 ps, 所以就用 python 实现了。
效果图
这里用的图片全是 240x240 的,按文件名的首字母作为序号,pil 虽然可以计算文字的尺寸,但类似 d 这样的字符依然不能处于圆圈的正中,所以还对个别字符做了偏移设置,本来想用 aggdraw 画圆圈的,能平滑 一些,不过安装了好几次,都以失败告终,最终放弃。
#!/usr/bin/env python #-*- coding: utf-8 -*- import os, sys, fnmatch import image, imagedraw, imagefont def process_picture(filename): seq = os.path.split(filename)[-1][0].upper() img = image.open(os.path.join(input_dir, filename)) draw = imagedraw.draw(img) # 在右下角画白底黑框圆圈 draw.ellipse((215, 215, 235, 235), outline='black', fill='white') # 将字母序号写入到圆圈内 font = imagefont.truetype('fonts/times new roman.ttf', 20) # 计算文字居中的位置 text_size = draw.textsize(seq, font) x = (20 / 2) - (text_size[0] / 2) y = (20 / 2) - (text_size[1] / 2) # 字母偏移量 offsets = {'a': 1, 'b': 1, 'e': 1, 'd': 2} offset = offsets.get(seq, 0) draw.text((215 + x + offset, 215 + y), seq, font=font, fill='black') # save image img.save(os.path.join(output_dir, filename), 'jpeg') if __name__ == '__main__': if len(sys.argv) < 3: print 'usage: python drawseq.py <input_dir> <output_dir>' sys.exit(1) input_dir, output_dir = sys.argv[1:3] os.path.exists(output_dir) or os.makedirs(output_dir) for filename in os.listdir(input_dir): if fnmatch.fnmatch(filename.lower(), '*.jpg'): process_picture(filename)