mmdetection训练自己的VOC数据集及常见问题

训练自己的COCO数据集及常见问题见下文：

mmdetection训练自己的COCO数据集及常见问题_不瘦8斤的妥球球饼的博客-CSDN博客

---

目录

一、环境安装

二、训练测试步骤

三、常见问题

batch size设置

学习率和epoch的修改

训练过程loss为nan的问题

GPU out of memory

如果自己的voc数据集图片是 png/tif 文件

训练生成的.pth文件占用较大内存

---

一、环境安装

代码：GitHub - open-mmlab/mmdetection: OpenMMLab Detection Toolbox and Benchmark

官方安装教程：Prerequisites — MMDetection 2.23.0 documentation

或者mmdetection/get_started.md at master · open-mmlab/mmdetection · GitHub

二、训练测试步骤

步骤一：准备数据。首先根据规范的VOC数据集导入到项目目录下，如下所示：

 VOC数据集格式介绍_不瘦8斤的妥球球饼的博客-CSDN博客_voc格式

mmdetection
├── mmdet
├── tools
├── configs
├── data
│   ├── VOCdevkit
│   │   ├── VOC2007
│   │   │   ├── Annotations
│   │   │   ├── JPEGImages
│   │   │   ├── ImageSets
│   │   │   │   ├── Main
│   │   │   │   │   ├── test.txt
│   │   │   │   │   ├── trainval.txt

推荐以软连接的方式创建：

cd mmdetection
mkdir data
ln -s $VOC2007_ROOT data/VOCdevkit

其中，$VOC2007_ROOT需改为你的VOC2007数据集根目录。

步骤二：修改config.py

在configs文件夹中选择自己要训练的config文件，例如：faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py

选好后，打开可以看到faster_rcnn_r50_fpn_1x_coco.py的基础配置如下：

下需要修改前两个文件（如红框所示）：

1. 退到上级目录找到_base_,找到faster_rcnn_r50_fpn.py，修改原VOC类的数目：