使用Python批量重命名图片工具

批量图片文件重命名

在图像处理和机器学习领域，数据标注是一个关键步骤。为了确保数据集的一致性和可管理性，图像文件的命名规范至关重要。手动重命名大量图片不仅耗时费力，还容易出错。为了解决这一问题，我开发了一款基于Python的批量重命名工具，大幅提升了标注工作的效率和准确性。

import os

def rename_images(directory, prefix, num_digits):
    # 判断目录地址
    if not os.path.isdir(directory):
        print(f"Error: Directory '{directory}' does not exist.")
        return
    # 获取目录地址
    files = os.listdir(directory)
    # 过滤图片文件
    image_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff']
    images = [f for f in files if os.path.splitext(f)[1].lower() in image_extensions]
    # 排序
    images.sort()
    # 遍历图片
    for index, image in enumerate(images, start=1):
        new_name = f"{prefix}{str(index).zfill(num_digits)}{os.path.splitext(image)[1]}"
        old_path = os.path.join(directory, image)
        new_path = os.path.join(directory, new_name)
        # 重命名
        os.rename(old_path, new_path)
        print(f"Renamed '{image}' to '{new_name}'")

if __name__ == "__main__":
    directory = './小白瓶/images'
    prefix = '小白瓶_'
    num_digits = int(5)
    rename_images(directory, prefix, num_digits)

结语

Python批量重命名图片工具通过自动化的方式，大幅提升了图像文件管理的效率和准确性。它简单易用，功能强大，能够满足多种场景下的命名需求。无论是在机器学习数据集准备、图像标注，还是日常文件管理中，这款工具都能发挥重要作用。希望通过这篇文章，能够帮助更多人了解和使用这一工具，从而提升工作效率，减少重复劳动。