INRIA数据集转换成VOC格式

最新推荐文章于 2023-08-03 18:26:40 发布

GuanghaoChen

最新推荐文章于 2023-08-03 18:26:40 发布

阅读量3k

点赞数 1

分类专栏：行人检测

本文链接：https://blog.youkuaiyun.com/qq_33297776/article/details/80260241

版权

本文提供了一种将INRIA数据集转换为VOC格式的方法，涉及614张图片，其中Train的Pos和Annotations用于训练，Test的Pos用于测试。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

背景

下面的代码是将INRIA数据集转换成VOC格式。图片是614张。
其中，使用OinginImage文件夹里的Train中Pos和Annotations作为训练，利用Test里的Pos作为测试。

# -*- coding: UTF-8 -*-
from xml.dom.minidom import Document
import os
import re

list = os.listdir("annotations")
savePath = 'Annotations'
for oldfilename in list:
    if str(".txt") not in oldfilename:
        continue
    print oldfilename

    #raw_input(unicode('按回车键退出...','utf-8').encode('gbk'))

    fileindex = re.findall('\d+', oldfilename)

    print fileindex
    #raw_input(unicode('按回车键退出...','utf-8').encode('gbk'))

    print str(int(fileindex[0]))

    #raw_input(unicode('按回车键退出...','utf-8').encode('gbk'))
    newfilename = os.path.splitext(oldfilename)[