Yolov5制作自己的数据集，图片标签重命名程序

制作自己的数据集，应该先进行格式重命名，以便于后期处理

import os

class BatchRename():
    '''
    批量重命名文件夹中的图片文件

    '''
    def __init__(self):
        self.path = r'E:\python学习资料\python数据处理\xiangduiying'  #表示需要命名处理的文件夹

    def rename(self):
        filelist = os.listdir(self.path) #获取文件路径
        total_num = len(filelist) #获取文件长度（个数）
        i = 1  #表示文件的命名是从1开始的
        for item in filelist:
            if item.endswith('.jpg'):  #初始的图片的格式为jpg格式的（或者源文件是png格式及其他格式，后面的转换格式就可以调整为自己需要的格式即可）
                src = os.path.join(os.path.abspath(self.path), item)
                # dst = os.path.join(os.path.abspath(self.path), '2_' + str(i) + '.jpg')#处理后的格式也为jpg格式的，当然这里可以改成png格式
                # dst = os.path.join(os.path.abspath(self.path), '002_' + str(i) + '.jpg')

                dst = os.path.join(os.path.abspath(self.path), '0' + format(str(i), '0>3s') + '.jpg')    #这种情况下的命名格式为0000000.jpg形式，可以自主定义想要的格式
                print(dst)
                try:
                    os.rename(src, dst)
                    print('converting %s to %s ...' % (src, dst))
                    i = i + 1
                except:
                    continue
        print('total %d to rename & converted %d pngs' % (total_num, i))

if __name__ == '__main__':
    demo = BatchRename()
    demo.rename()

当没有提前编号时，后期处理会带来一些麻烦，不过也有代码可以帮助大家

import os

class BatchRename():
    '''
    批量重命名文件夹中的图片文件

    '''
    def __init__(self):
        self.path1 = r'E:\python学习资料\python数据处理\xiangduiying'  #表示需要命名处理的文件夹
        self.path2 = r'E:\python学习资料\python数据处理\labels_my-project-name_2021-09-23-10-49-03'  #表示需要命名处理的文件夹

    def rename(self):
        filelist1 = os.listdir(self.path1) #获取文件路径
        filelist2 = os.listdir(self.path2) #获取文件路径
        total_num = len(filelist1) #获取文件长度（个数）
        i = 1  #表示文件的命名是从1开始的
        for item in filelist1:
            if item.endswith('.jpg'):  #初始的图片的格式为jpg格式的（或者源文件是png格式及其他格式，后面的转换格式就可以调整为自己需要的格式即可）
                src1 = os.path.join(os.path.abspath(self.path1), item)
                src2 = os.path.join(os.path.abspath(self.path2), item[:-4]+'.txt')
                # dst = os.path.join(os.path.abspath(self.path), '2_' + str(i) + '.jpg')#处理后的格式也为jpg格式的，当然这里可以改成png格式
                # dst = os.path.join(os.path.abspath(self.path), '002_' + str(i) + '.jpg')

                dst1 = os.path.join(os.path.abspath(self.path1), '0' + format(str(i), '0>3s') + '.jpg')    #这种情况下的命名格式为0000000.jpg形式，可以自主定义想要的格式
                dst2 = os.path.join(os.path.abspath(self.path2), '0' + format(str(i), '0>3s') + '.txt')    #这种情况下的命名格式为0000000.jpg形式，可以自主定义想要的格式

                os.rename(src1, dst1)
                os.rename(src2, dst2)
                print('converting %s to %s ...' % (src1, dst1))
                print('converting %s to %s ...' % (src2, dst2))
                i = i + 1

        print('total %d to rename & converted %d jpg' % (total_num, i))

if __name__ == '__main__':
    demo = BatchRename()
    demo.rename()