0006-目标检测-制作VOC数据集-生成Main下的txt文件

柔光的暖阳◎ 2023-01-07 07:19 125阅读 0赞

import os
    import random
    
    #其中trainval=train+val
    trainval_percent = 0.7  # 可以自己设置
    train_percent = 0.8  # 可以自己设置
    
    xmlfilepath = f"/Annotations"  # 这里填写自己的标注目录
    txtsavepath = f"/ImageSets/Main"
    total_xml = os.listdir(xmlfilepath)
    num = len(total_xml)
    list = range(num)
    tv = int(num * trainval_percent)
    tr = int(tv * train_percent)
    trainval = random.sample(list, tv)
    train = random.sample(trainval, tr)
    
    ftrainval = open(txtsavepath + '/trainval.txt', 'w')
    ftest = open(txtsavepath + '/test.txt', 'w')
    ftrain = open(txtsavepath + '/train.txt', 'w')
    fval = open(txtsavepath + '/val.txt', 'w')
    
    for i in list:
        name = total_xml[i][:-4] + '\n'
        if i in trainval:
            ftrainval.write(name)
            if i in train:
                ftrain.write(name)
            else:
                fval.write(name)
        else:
            ftest.write(name)
    
    ftrainval.close()
    ftrain.close()
    fval.close()
    ftest.close()
    print('Well finshed')

发表评论取消回复

表情：

评论列表（有 0 条评论，125人围观）

还没有评论，来说两句吧...

相关阅读

相关 0006-目标检测-制作VOC数据集-生成Main下的txt文件

import os import random 其中trainval=train+val trainval_percent = 0.7

柔光的暖阳◎/ 2023年01月07日 07:19/ 0 赞/ 126 阅读

相关目标检测数据集PASCAL VOC笔记

PASCAL VOC 数据集的应用领域有Object Classification 、Object Detection、Object Segmentation、Human La

小鱼儿/ 2022年12月26日 11:25/ 0 赞/ 234 阅读

相关目标检测数据集MSCOCO详解

1. 前言介绍一下目标检测领域另外一个比较有名的数据集 [MS COCO][] (Microsoft COCO: Common Objects in Context)

以你之姓@/ 2022年10月22日 04:06/ 0 赞/ 653 阅读

相关目标检测数据集PASCAL VOC详解

1. 前言 PASCAL VOC挑战赛（The [PASCAL][] Visual Object Classes ）是一个世界级的计算机视觉挑战赛,[PASCAL][P

末蓝、/ 2022年10月22日 02:57/ 0 赞/ 341 阅读

相关使用Yolov5进行目标检测并训练自己的VOC格式数据集

使用Yolo v5进行目标检测并训练自己的VOC格式数据集一、前提准备二、下载代码及配置环境三、下载预训练模型四、预测五、训练

桃扇骨/ 2022年09月02日 08:40/ 0 赞/ 312 阅读

相关 python对目标检测数据集xml文件操作

博主在制作目标检测数据集时，需要将之前的标签进行替换，将person替换为rider，使用python批量处理xml格式文件如下： import os, sys

淩亂°似流年/ 2022年04月25日 09:32/ 0 赞/ 397 阅读

相关实践目标检测--读取数据集

描述对于普通的图像分类，label只用表示图片的类别就行，而目标检测，不仅仅包括了类别的判断，还包含了类别的位置信息。所以在神经网络的构造上和数据的读取上都大不相同。

小咪咪/ 2022年04月18日 05:59/ 0 赞/ 415 阅读

相关制作VOC数据集

[https://blog.csdn.net/xlz2017/article/details/80426887][https_blog.csdn.net_xlz2017_art

系统管理员/ 2022年01月15日 00:57/ 0 赞/ 340 阅读

相关目标检测数据集处理脚本

功能一：将图像文件和标注文件对应，删去无标注文件 import os import sys input_folder=r'C:\Users\roc

古城微笑少年丶/ 2021年12月24日 00:53/ 0 赞/ 363 阅读

相关 Citypersons数据集转VOC标准格式（YOLO 目标检测txt格式）

CItyscapes城市数据集包含一组不同的立体视频序列中记录来自50个不同城市的街景,高质量的进行像素级的注释。数据集下载地址（需要申请注册，通过申请才能下载）\[[http

你的名字/ 2021年11月09日 14:31/ 0 赞/ 1490 阅读