1、code
2、tutorial
3、添加自己的数据集
1)添加
.../mmdetection/mmdet/datasets/xxx.py
(要记得@DATASETS.register_module)
2)编辑
.../mmdetection/mmdet/datasets/__init__.py
4、添加自己的detector
1)添加
.../mmdetection/mmdet/models/detectors/xxx.py
2)编辑
.../mmdetection/mmdet/models/detectors/__init__.py
3)config文件中model[‘type’]改为2)中__init__下的模型
5、Registry操作
.../mmcv/utils/registry.py –> build_from_cfg(cfg, registry, default_args=None)
其中,registry包括
BACKBONES = Registry('backbone')
NECKS = Registry('neck')
ROI_EXTRACTORS = Registry('roi_extractor')
SHARED_HEADS = Registry('shared_head')
HEADS = Registry('head')
LOSSES = Registry('loss')
DETECTORS = Registry('detector')
6、常见的坑
1)2.0版本中,标签0~K-1代表前景,K代表背景
https://github.com/open-mmlab/mmdetection/issues/3629
https://mmdetection.readthedocs.io/en/latest/compatibility.html#codebase-conventions
2)默认freeze初始几层