Python+PIL实现批量在图片上写上自定义文本_Python

背景

有时候我们需要将图片上添加某一个编号，但是如果图片太多我们将会变得非常麻烦，这时候我们可以使用python对图片进行自然排序，接着在利用pil 对图片进行编辑，添加上自增的编号

1、库的介绍

pillow 是一个 python 的图像处理库，它是 python imaging library (pil) 的一个分支，并且增加了更多的功能。pillow 库使得加载、操作和保存多种图像文件格式变得非常简单。它支持大量的图像格式，包括常见的如 jpeg、png、bmp、gif、ppm、tiff 和一些更少见的格式。

主要特性

图像处理: pillow 提供了一系列的方法来对图像进行基本的操作，例如裁剪、旋转、缩放、颜色转换等。

图像过滤器: 包括模糊、轮廓检测、边缘增强等滤镜效果。

文本和图形绘制: 可以在图片上添加文字或绘制简单的图形。

图像序列处理: 支持处理 gif 或其他格式的多帧图像。

扩展性: 用户可以编写自己的图像解码器或格式处理器。

色彩模式支持: 支持多种色彩模式，比如 rgb、rgba、cmyk 等。

2、库的安装

库	用途	安装
pliiow	处理图片	pip install pillow -i https://pypi.tuna.tsinghua.edu.cn/simple/
natsort	自然排序	pip install natsort -i https://pypi.tuna.tsinghua.edu.cn/simple/
os	获取绝对路径	内置库无需安装

3、核心代码

自然排序

 # 获取所有图片文件 使用自然排序
 image_files = natsort.natsorted(os.listdir(input_folder))

创建自定义文本(包括样式)

 # 创建绘图对象
draw = imagedraw.draw(img)

 # 设置字体
 font_size = max(30, int(img_width * 0.03))
 font = imagefont.truetype("simhei.ttf", font_size)

 # 文本内容和颜色
 text = f"编号:{index + 1}"
 text_color = (0, 0, 0)  # 黑色字体
 shadow_color = (0, 0, 0)  # 黑色阴影

 # 获取文本尺寸
 bbox = draw.textbbox((0, 0), text, font=font)
 text_width = bbox[2] - bbox[0]
 text_height = bbox[3] - bbox[1]

 # 计算文本位置
 x = img_width - text_width - 190
 y = img_height - text_height - 30

 # 绘制文本阴影
 draw.text((x + 1, y + 1), text, font=font, fill=shadow_color)

 # 绘制文本
 draw.text((x, y), text, font=font, fill=text_color)

4、测试图片创建脚本

# -*- coding: utf-8 -*-
'''
@project ：测试 
@file    ：创建随机图片.py
@ide     ：pycharm 
@author  ：一晌小贪欢（278865463@qq.com）
@date    ：2024/9/27 下午11:04 
'''

import random
import os
from pil import image, imagefont, imagedraw

# 图片命名为 name_list 中的随机一个，如张三_1.png，张三_2.png
name_list = ['张三', '李四', '王五', '赵六', '田七']

try:
    os.mkdir('图片/图片')
except fileexistserror:
    print('文件夹已存在')

# 清空文件夹 图片
for i in os.listdir('图片/图片'):
    os.remove('./图片/' + i)

# 创建50张随机图片
for i in range(50):
    width, height = 300, 300
    img = image.new('rgb', (width, height), color=(255, 255, 255))
    draw = imagedraw.draw(img)

    # 尝试加载字体
    try:
        font = imagefont.truetype('c:/windows/fonts/simhei.ttf', 65)
    except ioerror:
        print("font not found. using default font.")
        font = imagefont.load_default()

    random_name = random.choice(name_list)
    text = f"{random_name}_{i}"

    # 计算文本的边界框
    bbox = draw.textbbox((0, 0), text, font=font)
    text_width = bbox[2] - bbox[0]
    text_height = bbox[3] - bbox[1]

    # 计算文本的起始位置
    x = (width - text_width) / 2
    y = (height - text_height) / 2

    # 在图片上绘制文本
    draw.text((x, y), text, fill=(0, 100, 255), font=font)
    img.save(f'./图片/{random_name}_{i}.png')
    print(f"{random_name}_{i}.png")

5、完整代码

from pil import image, imagedraw, imagefont
import natsort
import os

def add_number_to_images(input_folder, output_folder):
    """
    在指定文件夹内的每张图片右下角添加编号
    :param input_folder: 输入文件夹路径
    :param output_folder: 输出文件夹路径
    """
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    # 获取所有图片文件 使用自然排序
    image_files = natsort.natsorted(os.listdir(input_folder))


    for index, file_name in enumerate(image_files):
        # 打开图片
        img_path = os.path.join(input_folder, file_name)
        img = image.open(img_path)

        # 获取图片大小
        img_width, img_height = img.size

        # 创建绘图对象
        draw = imagedraw.draw(img)

        # 设置字体
        font_size = max(30, int(img_width * 0.03))
        font = imagefont.truetype("simhei.ttf", font_size)

        # 文本内容和颜色
        text = f"编号:{index + 1}"
        text_color = (0, 0, 0)  # 黑色字体
        shadow_color = (0, 0, 0)  # 黑色阴影

        # 获取文本尺寸
        bbox = draw.textbbox((0, 0), text, font=font)
        text_width = bbox[2] - bbox[0]
        text_height = bbox[3] - bbox[1]

        # 计算文本位置
        x = img_width - text_width - 190
        y = img_height - text_height - 30

        # 绘制文本阴影
        draw.text((x + 1, y + 1), text, font=font, fill=shadow_color)

        # 绘制文本
        draw.text((x, y), text, font=font, fill=text_color)

        # 保存处理后的图片
        output_path = os.path.join(output_folder, file_name)
        img.save(output_path)

    print(f"所有图片已处理并保存到：{output_folder}")

if __name__ == "__main__":
    # 输入和输出文件夹路径
    input_dir = "图片"  # 输入图片的文件夹
    output_dir = "处理后图片"  # 处理后图片的保存文件夹
    if not os.path.exists(input_dir):
        os.mkdir(output_dir)

    add_number_to_images(input_dir, output_dir)

效果图