python 批量重命名数据集中的图片和标签

批量重命名数据集中的图片和标签

针对情况：1.当数据集中的图片和标签放在同一文件夹，图片名和标签名重复
2.需要将图片名和标签名按顺序进行排序

注：数据集中图片名和标签名一一对应，改变图片名的同时，对应的标签名也会随之变动，改变后的图片和标签仍然一一对应。

import os


# 这种的只能处理 一个样本数据的 图片名字， 有待改进
def rename(path,path1):
    filelist = os.listdir(path)  # 获取指定的文件夹包含的文件或文件夹的名字的列表
    print(filelist)
    total_num = len(filelist)  # 获取文件夹内所有文件个数

    i = 1772  # 图片名字从 0 开始
    c = 0
    for item in filelist:  # 遍历这个文件夹下的文件,即 图片
        if item.endswith('.jpg'):
            src = os.path.join(os.path.abspath(path), item)
            dst = os.path.join(os.path.abspath(path1), str(i) + '.jpg')
            sr = os.path.join(os.path.abspath(path), item)
            sr=sr.replace('.jpg','.txt')
            ds = os.path.join(os.path.abspath(path1), str(i) + '.txt')

            try:
                os.rename(src, dst)
                os.rename(sr,ds)
                print('converting %s to %s ...' % (src, dst))
                i = i + 1
                c = c + 1
            except:
                continue
    print('total %d to rename & converted %d jpgs' % (total_num, i))
    print('total %d to rename & converted %d jpgs' % (total_num, c))


if __name__ == '__main__':
    path = '/home/lbb/文档/Yolov/yolov5-master/datasets/airportdata/part6(366)/frame/'
    path1='/home/lbb/文档/Yolov/yolov5-master/datasets/airportdata/images/train/part6/'
    rename(path,path1)

path是需要修改的图片和标签的文件路径
path1是修改后的图片和标签存放的文件路径