[人工智能]mmdetection学习日记

想复现Oriented R-CNN的代码并做一些改动，下载下来发现是使用的mmdetection框架，于是边学习记录一些遇到的问题。

1、训练步骤

（1）在configs中选自己想要用的模型，可以直接修改，我习惯新建一个.py文件，将model、dataset等config文件全部复制过来，方便改动，即开头的_base_中所有文件都拷贝到新.py中。

（2）修改后命令行内运行

python tools/train.py configs/xxx.py

2、无法下载预训练模型，报错"HTTP Error 403: Forbidden"

换成国内源的网址即可正常下载

可以查询：mmcv/model_zoo/open_mmlab.json · master · mirrors / open-mmlab / mmcv · CODE CHINA (gitcode.net)https://gitcode.net/mirrors/open-mmlab/mmcv/-/blob/master/mmcv/model_zoo/open_mmlab.json

选取想要的模型，将地址复制下来，在configs/default_runtime.py中修改load_from为新地址即可。

3、用mmdetection中现成模型训练自己的数据集

如训练COCO格式的自制数据集，类别与COCO不一致

（1）在data文件夹中放入分好train/val/test的图片和标签。

（2）自己的config文件中，将num_classes修改为自己的类别数（加背景），如model ——> roi_head ——> bbox_head ——> num_classes，如果有mask分支则mask_head中也有一项num_classes。

（3）mmdet/datasets/coco.py 、 mmdet/core/evaluation/class_names.py? 都将原CLASSES注释掉，换成新的类别名称。

4、ValueError: need at least one array to concatenate

一般都是数据集的标签里出了问题，查看标签文件里是否没有生成bbox坐标或label等