目标检测YOLOV2-Xml文件解析（一）

目录

读取标签文件

读取标签文件

目标检测的标签文件一般格式形式是xml格式，可使用labelImg标注工具进行标注。

将全部标签文件与源图片路径读取出来保存到列表中，返回，以供后续处理使用。

读取文件代码：

文件名：xml_parse.py

# -*- coding: utf-8 -*-
import os, glob
import numpy as np
import xml.etree.ElementTree as ET

def paras_annotation(img_dir, ann_dir, labels):
    """
    :param img_dir: image path
    :param ann_dir: annotation xml file path
    :param labels: ("class1", "class2",...,), 背景默认为0
    :function: paras annotation info from xml file
    :return:
    """
    imgs_info = []  #存储所有图片信息的容器列表
    max_boxes = 0