voc 数据集制作 + yolo v3 训练 笔记

数据准备

课程需要一个目标检测的demo，对单个种类的目标检测，以下为训练记录。

1. 文件目录格式

• VOCdevkit
  • Annotations
    • 00001.xml
    • 00002.xml
    • ……
  • ImageSets
    • Main
      • train.txt
      • test.txt
  • JPEGImges
    • 00001.jpg
    • 00002.jpg
    • ……
  • labels
    • 00001.txt
    • 00002.txt
    • ……

1.1 Annotations文件夹下 xml 文件生成

def save_xml(image_name, bbox, save_dir='./xml', width=2666, height=2000, channel=3):
    from lxml.etree import Element, SubElement, tostring
    from xml.dom.minidom import parseString
    node_root = Element('annotation')

    node_folder = SubElement(node_root, 'folder')
    node_folder.text = 'JPEGImages'

    node_filename = SubElement(node_root, 'filename')
    node_filename.text = image_name

    node_size = SubElement(node_root, 'size')
    node_width = SubElement(node_size, 'width')
    node_width.text = '%s' % width

    node_height = SubElement(node_size, 'height')
    node_height.text = '%s' % height

    node_depth = SubElement(node_size, 'depth')
    node_depth.text = '%s' % channel
    # 2 different shape of bbox data
    # 1. x,y,w,h
    # 2.x1,y1,x2,y2
    for x1, y1, x2, y2 in bbox:
        node_object &#