前言
此代码为博主进行深度学习的数据集批量更改名字所用。网上大多都是批量更改目标路径文件的所有文件,对分类多的数据集太不友好了!于是此代码就此诞生。。。
另:后面也有针对单文件批量修改名字的代码
一、使用方法
目录示例:
— train
丨— cat
—丨— cat_1.jpg
—丨— cat_2.jpg
丨— dog
—丨— dog_1.jpg
—丨— dog_2.jpg
—丨— dog_3.jpg
—丨— dog_4.jpg
丨— pig
···
丨— fish
具体流程:
- 将paths设置为如上所示的train目录作为根目录
- 运行代码就会遍历根目录的所有子目录,并获得该目录名
- 子目录的所有文件以目录名为前缀重新命名,结果如目录实例所示
二、代码示例
代码如下(示例):
import os
paths = r"E:\myproject\datasets\train"
# 获取根目录下所有子目录,存入列表中
fileLists = os.listdir(paths)
for path in fileLists:
fileList = os.listdir(paths+'\\'+path)
n = 0
# 遍历子目录的所有文件
for i in fileList:
# 设置旧文件名(就是路径+文件名)
oldname = paths + os.sep + path + os.sep + fileList[n] # os.sep添加系统分隔符
# 设置新文件名
newname = paths + os.sep + path + os.sep + path + '_' + str(n + 0) + '.jpg'
os.rename(oldname, newname) # 用os模块中的rename方法对文件改名
print(oldname, '===>', newname)
n += 1
另:下面这部分是只针对单文件夹的修改文件名
import os
path = input('请输入文件路径:')
# 获取该目录下所有文件,存入列表中
fileList = os.listdir(path)
n = 0
for i in fileList:
# 设置旧文件名(就是路径+文件名)
oldname = path + os.sep + fileList[n] # os.sep添加系统分隔符
# 设置新文件名
newname = path + os.sep + 'pig_' + str(n) + '.jpg'
os.rename(oldname, newname) # 用os模块中的rename方法对文件改名
print(oldname, '======>', newname)
n += 1