python 对文件夹下的所有图片进行重命名

在做深度学习的相关项目时,经常需要对图片数据集进行重命名。这里通过python实现遍历文件夹图片并重命名,并提供了完整代码,大家可以根据自己的需求灵活修改。

1.需求

把文件夹下的每张图片重命名为 :疫苗名称 + 固定长度的编码。如:卡介苗_1.jpg 重名为 卡介苗_00001.jpg

2.代码

import os
# 图片所在路径
root_path = "E:\\images\\卡介苗"
filename_list = os.listdir(root_path)
for filename in filename_list:
    if filename.endswith('.jpg'):
        src_img_path = os.path.join(os.path.abspath(root_path), filename)
        new_img_code = filename.split('_')[1].split('.')[0].zfill(5)

        dst_img_path = os.path.join(os.path.abspath(root_path), filename.split('_')[0] + '_' + new_img_code + '.jpg')
        try:
            os.rename(src_img_path, dst_img_path)
            print('converting %s to %s ...' % (src_img_path, dst_img_path))
        except:
            continue

运行结果: