PASCAL VOC标注数据解析

最新推荐文章于 2023-01-17 15:16:35 发布

K5niper

最新推荐文章于 2023-01-17 15:16:35 发布

阅读量864

点赞数 1

分类专栏：计算机视觉

本文链接：https://blog.csdn.net/zhaoyin214/article/details/103859168

版权

本文介绍了PASCAL VOC标注数据的格式，包括数据文件的内容，以及如何实现和测试数据解析过程。

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

PASCAL VOC标注数据解析

1 数据格式

PASCAL VOC标注数据文件（.xml）内容：
在这里插入图片描述

2 实现

文件：pascal_voc_label_parser.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
"""
@file    :   pascal_voc_label_parser.py
@time    :   2020/01/06 10:09:43
@author  :   XiaoY
@version :   1.0
@contact :   zhaoyin214@qq.com
@license :   (c)copyright XiaoY
@desc    :   PASCAL VOC
"""

__author__ = "XiaoY"

import numpy as np
import pandas as pd
import xml.etree.ElementTree as ET
import os

from .config import ANNO_DIR, BBOX_COLUMNS, IMG_DIR

class PascalVocLabels(object):

    def __init__(self, root):

        self._root = root
        self._anno_dir = os.path.join(root, ANNO_DIR)
        self._img_dir = os.path.join(root, IMG_DIR)

    def _anno_parser(self, filename):

        # element tree<