Wider Face数据集转VOC格式数据集编程实现

最新推荐文章于 2025-02-24 07:36:14 发布

知更i

最新推荐文章于 2025-02-24 07:36:14 发布

阅读量2.3k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： ssd widerface

本文链接：https://blog.youkuaiyun.com/jingtaohuang/article/details/88987364

前言

WIDER FACE: 官网地址：http://mmlab.ie.cuhk.edu.hk/projects/WIDERFace/

第一步

在WIDER FACE官网下载以下四个文件（训练集、验证集、测试集、标注文件）

下载好后解压，我的目录结构如下：

红框内是生成后的文件

第二步：

编写脚本文件dataset.py（注意文件路径中不要包含中文）

import os,cv2,sys,shutil

from xml.dom.minidom import Document

def writexml(filename,saveimg,bboxes,xmlpath):
    doc = Document()

    annotation = doc.createElement('annotation')

    doc.appendChild(annotation)

    folder = doc.createElement('folder')

    folder_name = doc.createTextNode('widerface')
    folder.appendChild(folder_name)
    annotation.appendChild(folder)
    filenamenode = doc.createElement('filename')
    filename_name = doc.createTextNode(filename)
    filenamenode.appendChild(filename_name)
    annotation.appendChild(filenamenode