基于MMDetection训练VOC格式数据集

一 环境说明

       基于前述安装MMDetection，数据集为VOC格式，主要版本如下：

Python：3.7.8
CUDA：11.3
cuDNN：8.4.0
torch：1.12.0
torchvision：0.13.0
mmcv-full：1.6.0
MMDetection：2.25.3

二 数据集准备

     使用VOC格式进行模型训练，利用labelimg对图像进行标注，生成xml文件，准备好图片和标注文件，并对数据集进行划分，文件目录如下图所示。

mmdetection
├── mmdet
├── tools
├── configs
├── dataset
│   ├── ACID  # 自己给数据集取名字
│   │   ├── VOC2007
│   │   │   ├── Annotations # xml文件
│   │   │   ├── JPEGImages # 图片文件
│   │   │   ├── ImageSets
│   │   │   │   ├── Main
│   │   │   │   │   ├── test.txt 
│   │   │   │   │   ├── trainval.txt

三 修改默认labels

        1、在 .\mmdetection-v2.25.3\mmdet\core\evaluation\class_names.py中，修改函数

voc_classes下面的默认类别，修改为自己标注的类别。

def voc_classes():
    #