专栏介绍
这个专栏旨在分享我在进行深度学习算法前的一些脚本文件。包括一些常见的处理,如将单通道图片改为三通道、统一图片格式、将标签进行统一设置方便查找等等。
为文件夹内的文件添加前缀
在进行算法训练前,清晰的文件命名尤为重要。
写这个脚本的初衷是因为最近在训练yolo系列,扩充的数据集是从百度图片爬虫所得。众所周知,YOLO系列的标注文件为txt格式,原爬得的图片都是以1.jpg开始命名,因此将所有图片汇总时会出现同名文件的低级错误,因此为文件统一加不同前缀,避免出现同名错误。我一般喜欢添加的是时间。分享此脚本避免大家入坑。
import os
import sys
if __name__ == "__main__":
folder_name ="C:/Users/wkk/Desktop/wkk" # 获取该文件夹路径
file_names = os.listdir(folder_name) # 获取文件夹内所有文件的名字
for name in file_names: # 如果某个文件名在file_names内
old_name = folder_name +'/’+ name #获取旧文件的名字,注意名字要带路径名
new_name = folder_name + '/’+ '22_12_01_’+ name # 定义新文件的名字,这里给每个文件名前加了
os.rename(old_name,new_name)#用rename()函数重命名
print(new_name) #打印新的文件名字
话说编写脚本程序,还是得python啊