OpenMMLab AI 实战训练营笔记 05

MMDetection 环境搭建

这里我使用软连接进行环境配置

git clone -b 2.x git@github.com:open-mmlab/mmcv.git 
cd mmcv

# 安装 ninja 和 psutil 以加快编译速度
pip install -r requirements/optional.txt

# 仅仅编译，不安装
MMCV_WITH_OPS=1 python setup.py build_ext --inplace

# 建立软链接
cd ../mmyolo
ln -s ../mmcv/mmcv mmcv

如果是使用mmdet 3.x版本，则需要依次配置MMengine，配置步骤如上MMCV所示。

参考：https://zhuanlan.zhihu.com/p/580885852

COCO数据集格式

在 MMDetection 中改写配置文件

训练

训练一个新的模型
为了使用新的配置方法来对模型进行训练，你只需要运行如下命令。

python tools/train.py ../configs/xxxx/xxxxxxxxx.py # 改写的配置文件地址