最近在制作深度学习项目的测试集,需要对整个文件夹下的图片重命名以便标注,体会到了python脚本的强大和简便。
基本思路:使用python 的os 模块, 对文件夹进行遍历(listdir), 同时使用 rename 进行改名操作。
# -*- coding:utf8 -*-
import os
class BatchRename():
'''
批量重命名文件夹中的图片文件
'''
def __init__(self):
self.path = 'F:/testset/flowers'
def rename(self):
filelist = os.listdir(self.path)
total_num = len(filelist)
i = 0
for item in filelist:
if item.endswith('.jpg'):
src = os.path.join(os.path.abspath(self.path), item)
dst = os.path.join(os.path.abspath(self.path), str(i) + '.jpg')
try:
os.rename(src, dst)
print('converting %s to %s ...' % (src, dst))
i = i + 1
except:
continue
print('total %d to rename & converted %d jpgs' % (total_num, i))
if __name__ == '__main__':
demo = BatchRename()
demo.rename()
来源:CSDN
作者:上园村蜻蜓队长
链接:https://blog.csdn.net/lgk1996/article/details/80081380