caffe2 detectron训练pascal voc 2007

参考链接：https://blog.csdn.net/mr_health/article/details/80676799

 

一、数据准备

我的数据格式是voc，而detectron要求的数据格式是json，因此首先要进行格式的转换。

1.数据放置

官方网站上的要求的数据存放见左图，也就是说无论我们自己要训练什么数据，文件夹的名称必须是VOC+year。但是为什么要是VOC+year这样的形式呢，我们再翻看一下datasets/dadataset_catalog.py这个函数（右图），可以看到里面有各种数据集，包括voc2007、voc2012。实际上这些就是数据接口，也就是说我们把自己的数据按照接口的形式准备好，detectron就会调用。

一般情况下，我们的数据集可以分为三部分，一部分是训练数据（train），第二部分验证数据（val）、第三部分是测试数据（test）。实际上验证数据和测试数据都可以被称为测试数据，只是功能有所不同。验证数据（val）由于其带标签，在测试中往往会输出mAP，用以评估模型的性能，而测试数据（test）不含有标签，实际上就是当我们训练好了模型，那就在我的一个数据库上试试呗，看检测出来好玩不，测试数据（test）就这样一个数据库。右图中voc_2007_train就是用于训练，voc_2007_val用于验证，voc_2007_test用于测试。

但是实际上在我们训练模型的时候，只将数据集分为训练数据和测试数据，只是这时候的测试数据功能上是对模型的评估，所以在代码的书写上都以val指代，称呼上就是测试数据（下文就这么称呼了）。

 

注意：只有train,val可以生成AP,AP50,mAP结果，只要包含test这些结果为-1

 

dataset_catalog.py当中的文件，按照这个文件名命名

 

 

 

 

 

 

 

 

文件结构：在/home/test/detectron/detectron/datasets/data当中新建一个文件夹：VOC2007

 

1）annotations里面有两个json文件，分别是voc_2007_train.json,voc2007_val.json

将annotations文件夹当中的xml拷贝出来，复制到train，val两个文件夹当中，利用上述代码，生成train,val这两个json文件

修改相应的路径

#coding=utf-8
#fork from https://github.com/CivilNet/Gemfield/blob/master/src/python/pascal_voc_xml2json/pascal_voc_xml2json.py
import xml.etree.ElementTree as ET
import os
import json

coco = dict()
coco['images'] = []
coco['type'] = 'instances'
coco['annotations'] = []
coco['categories'] = []

category_set = dict()
image_set = set()

category_item_id = 0
image_id = 20180000000
annotation_id = 0


def addCatItem(name):
    global category_item_id
    category_item = dict()
    category_item['supercategory'] = 'none'
    category_item_id += 1
    category_item['id'] = category_item_id
    category_item['name'] = name
    coco['categories'].append(category_item)
    category_set[name] = category_item_id
    return category_item_id


def addImgItem(file_name, size):
    global image_id
    if file_name is None:
        raise Exception('Could not find filename tag in xml file.')
    if size['width'] is None:
        raise Exception('Could not find width tag in xml file.')
    if size['height'] is None:
        raise Exception('Could not find height tag in xml file.')
    image_id += 1
    image_item = dict()
    image_item['id'] = image_id
    image_item['file_name'] = file_name
    image_item['width'] = size['width']
    image_item['height'] = size['height']
    coco['images'].append(image_item)
    image_set.add(file_name)
    return image_id


def addAnnoItem(object_name, image_id, category_id, bbox):
    global annotation_id
    annotation_item = dict()
    annotation_item['segmentation'] = []
    seg = []
    # bbox[] is x,y,w,h
    # left_top
    seg.append(bbox[0])
    seg.append(bbox[1])
    # left_bottom
    seg.append(bbox[0])
    seg.append(bbox[1] + bbox[3])
    # right_bottom
    seg.append(bbox[0] + bbox[2])
    seg.append(bbox[1] + bbox[3])
    # right_top
    seg.append(bbox[0] + bbox[2])
    seg.append(bbox[1])

    annotation_item['segmentation'].append(seg)

    annotation_item['area'] = bbox[2] * bbox[3]
    annotation_item['iscrowd'] = 0
    annotation_item['ignore'] = 0
    annotation_item['image_id'] = image_id
    annotation_item['bbox'] = bbox
    annotation_item['category_id'] = category_id
    annotation_id += 1
    annotation_item['id'] = annotation_id
    coco['annotations'].append(annotation_item)


def parseXmlFiles(xml_path):
    for f in os.listdir(xml_path):
        if not f.endswith('.xml'):
            continue

        print(f)
        name =f.split('.')[0]
        valPath = '/home/test/detectron/detectron/datasets/data/VOC2007/VOCdevkit2007/VOC2007/ImageSets/Main/'
        valPath = os.path.join(valPath,"val.txt")
        with open(valPath,'a+') as ff:
            ff.write(name)
            ff.write('\n')

        bndbox = dict()
        size = dict()
        current_image_id = None
        current_category_id = None
        file_name = None
        size['width'] = None
        size['height'] = None
        size['depth'] = None

        xml_file = os.path.join(xml_path, f)
        print(xml_file)#输出xml文件的全路径

        tree = ET.parse(xml_file)
        root = tree.getroot() #抓根结点元素

        if root.tag != 'annotation': #根节点标签
            raise Exception('pascal voc xml root element should be annotation, rather than {}'.format(root.tag))

        # elem is , , ,