自制VOC数据集笔记

最新推荐文章于 2024-09-30 11:10:44 发布

「已注销」

最新推荐文章于 2024-09-30 11:10:44 发布

阅读量454

点赞数 1

分类专栏： python

本文链接：https://blog.csdn.net/weixin_44412218/article/details/104332445

版权

本文记录了在自制VOC数据集中遇到的问题，包括XML文件读取失败（中文字符问题）、Keyerror: 'none'、Keyerror: 'width'以及assert错误，并提供了相应的解决方案，如XML与TXT文件的转换、检查并修正类别、删除错误缓存文件和调整边界框坐标等。

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

自制VOC数据集

环境 win10 cuda10 pytorch1.0
项目地址：faster-rcnn
参考博客：https://blog.csdn.net/weixin_43380510/article/details/83004127
data文件夹格式：

---VOC2007
    ------Annotations
    ------ImagesSet
         ---------Main
              --------trainval.txt
              --------test.txt
    ------JPEGImages

Annotations放xml文件
JPEGImages放图片
Main中存放txt文件，里面是图片名称，不带“.jpg”
手动制作txt代码如下：

import os
path_name = r'D:\Coding\python\faster-rcnn.pytorch-pytorch-1.0\data\VOCdevkit\VOC2007\JPEGImages'  
f = open('D:/Coding/python/faster-rcnn.pytorch-pytorch-1.0/data/VOCdevkit/VOC2007/ImagesSet/Main/trainval.txt', mode='w')
for item in os.listdir(path_name):
    original_name = os.path.join(path_name, item)
    print(item)#原始名称带.jpg
    str_name=item.split('.')#分割字符
    print(str_name[0