PIL批量给图片加上字母序号

最新推荐文章于 2022-06-19 14:42:44 发布

转载最新推荐文章于 2022-06-19 14:42:44 发布 · 1.7k 阅读

文章标签：

#output #input #image #python #os

Python 专栏收录该内容

68 篇文章

订阅专栏

本文介绍如何使用Python的PIL库在图片上绘制指定的文字，并在圆圈中进行定位，包括处理特殊字符的偏移问题。

效果图

这里用的图片全是 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)

转自：http://www.g2w.me/2012/08/draw-char-in-picture-using-pil/