首页 > 脚本专栏 > python > Python PIL图片自定义文本

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

2024-11-28 08:47:39 作者：一晌小贪欢

Pillow 是一个 Python 的图像处理库,它是 Python Imaging Library (PIL) 的一个分支,并且增加了更多的功能,下面我们看看如何利用它实现批量在图片上写上自定义的文本吧

背景

有时候我们需要将图片上添加某一个编号，但是如果图片太多我们将会变得非常麻烦，这时候我们可以使用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)

效果图

到此这篇关于Python+PIL实现批量在图片上写上自定义文本的文章就介绍到这了,更多相关Python PIL图片自定义文本内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！