💪 专业从事且热爱图像处理,图像处理专栏更新如下👇:
📝《图像去噪》
📝《超分辨率重建》
📝《语义分割》
📝《风格迁移》
📝《目标检测》
📝《图像增强》
📝《模型优化》
📝《模型实战部署》
📝《图像配准融合》
📝《数据集》
📝《高效助手》
本教程中第一章节提供的是乱序修改图片名的例子,顺序修改图片名的方法在第二章节,代码是在第一章节基础上修改。
一、修改后乱序排名
2.1 参数修改
深度学习训练前快速批量修改数据集中的图片名,使用下面代码时只需要修改两个路径即可,见下:
2.2 代码
具体代码见下:
import os
import shutil
class BatchRename():
'''
批量重命名文件夹中的图片文件
'''
def __init__(self):
self.path = 'Images/val/val2017' #表示需要命名处理的文件夹
self.save_path='Images/val/val2017 2'#保存重命名后的图片地址
def rename(self):
filelist = os.listdir(self.path) #获取文件路径
total_num = len(filelist) #获取文件长度(个数)
i = 1 #表示文件的命名是从1开始的
for item in filelist:
print(item)
if item.endswith('.jpg'): #初始的图片的格式为jpg格式的(或者源文件是png格式及其他格式,后面的转换格式就可以调整为自己需要的格式即可)
src = os.path.join(os.path.abspath(self.path), item)#当前文件中图片的地址
dst = os.path.join(os.path.abspath(self.save_path), ''+str(i) + '.jpg')#处理后文件的地址和名称,可以自己按照自己的要求改进
try:
#os.rename(src, dst)
shutil.copy(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()
二、按原图片顺序修改图片名
上面提供方法修改后图片名是乱序的,没有按照原图片顺序修改,如果需要按原图片排序修改,详见我的另外一篇博文,链接为:按原顺序修改图片名
主要就在上面代码基础上加一行代码即可,代码见下:
# filelist.sort(key=lambda x: int(x.split('.')[0])) # 先按照名称顺序排好序
filelist.sort(key=lambda x: int(x.split('(')[1].split(')')[0])) # 先按照名称顺序排好序
三、总结
以上就是快速批量修改数据集中图片名的方法,希望能帮助到你!
感谢您阅读到最后!😊总结不易,多多支持呀🌹 点赞👍收藏⭐评论✍️,您的三连是我持续更新的动力💖
关注公众号「视觉研坊」,获取干货教程、实战案例、技术解答、行业资讯!