统一给图片数据集重命名

最新推荐文章于 2024-02-09 11:17:48 发布

q1030460485

最新推荐文章于 2024-02-09 11:17:48 发布

阅读量233

点赞数

文章标签： python 深度学习分类算法神经网络

本文链接：https://blog.csdn.net/q1030460485/article/details/124307162

版权

本文介绍如何使用Python高效地为图像数据集进行批量重命名，这对于深度学习和分类算法的训练过程至关重要。通过示例代码，展示了如何利用os和glob库遍历文件并重命名，确保神经网络模型的输入数据一致性。

摘要由CSDN通过智能技术生成

import os
 
class BatchRename():
 
    def __init__(self):
        self.path = "E:/MobileNet/train/Tungro" #图片的路径
 
    def rename(self):
        filelist = os.listdir(self.path) 
        filelist.sort()
        total_num = len(filelist)
        i = 0
        for item in filelist:
            item = item.lower() 
            if item.endswith('.jpg'):
                src = os.path.join(self.path, item)
                s = str(i)
                s = s.zfill(2)  #Python zfill() 方法返回指定长度的字符串，原字符串右对齐，前面填充0
                dst = os.path.join(os.path.abspath(self.path), s + '.jpg')
 
 
                '''
                    概述：
                    os.rename() 方法用于重命名文件或目录，从 src 到 dst,如果dst是一个存在的目录, 将抛出OSError
                    语法：
                    rename()方法语法格式如下：os.rename(src, dst)
                    参数：
                    sr