SSD训练自己的数据集（一）: 制作VOC数据集

最新推荐文章于 2024-08-08 07:42:54 发布

_金戈铁马_

最新推荐文章于 2024-08-08 07:42:54 发布

阅读量8.2k

点赞数 1

分类专栏：目标检测文章标签： SSD 目标检测 VOC数据格式

本文链接：https://blog.csdn.net/xlz2017/article/details/80426887

版权

本文介绍了如何为SSD目标检测训练创建VOC格式的数据集。首先，需要将图片重命名为特定格式，并创建Annotations、JPEGImages和ImageSets文件夹。ImageSets下设Main文件夹，用于存放txt文件。通过调整代码中trainval_percent和train_percent的值来设定训练集和验证集的比例。接着，将重命名后的图片放入JPEGImages文件夹，使用提供的Windows标记软件生成XML标注文件，存入Annotations文件夹，从而完成VOC数据集的制作。参考了CSDN上的两篇文章。

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

准备自己的图片，最好重命名为“000001.jpg”的格式，重命名Python代码如下：

import os
path = "./123"
filelist = os.listdir(path) #该文件夹下所有的文件（包括文件夹）
count=0
for file in filelist:
    print(file)
for file in filelist:   #遍历所有文件
    Olddir=os.path.join(path,file)   #原来的文件路径
    if os.path.isdir(Olddir):   #如果是文件夹则跳过
        continue
    filename=os.path.splitext(file)[0]   #文件名
    filetype=os.path.splitext(file)[1]   #文件扩展名
    Newdir=os.path.join(path,str(count).zfill(6)+filetype)  #用字符串函数zfill 以0补全所需位数
    os.rename(Olddir,Newdir)#重命名
    count+=1

VOC下一共有五个文件夹，其中只用到前三个文件夹，Annotations、JPEGImages和ImageSets