跳至主要內容

批量图片格式转换

大林鸱小于 1 分钟研发工具图片

在图像识别项目中,样本库图片需要统一格式以及标准化文件名称。

要求

  • 图片格式统一为 jpg;
  • 文件名称格式:{n}.jpg

Python 代码

注意以下代码在处理不同的图片格式时,可能会丢失信息。建议,格式转换后,查阅新图片文件。

PIL,Python Imaging Library,是 Python 平台上的图像处理标准库。

import os
from PIL import Image as img
#tqdm模块是python进度条库
from tqdm import tqdm


path = 'D:/workspace/yolov5/sample/original/'  # 待转换格式的图片所在文件夹
path2 = 'D:/workspace/yolov5/sample/images/'  # 转换后的图片存储路径

# 如果存储路径不存在,则创建目录
if(os.path.exists(path2) == False):
    os.makedirs(path2)

files = os.listdir(path)
print('原始文件个数:{}'.format(len(files)))
for n, filename in tqdm(enumerate(files), total=len(files)):
    try:
        img.open(path + filename).save(path2 + str(n) + ".jpg")
    except:
        print('convert {} failed.'.format(filename))
上次编辑于: