栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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()
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/269323.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号