mmdetection链接:GitHub - open-mmlab/mmdetection: OpenMMLab Detection Toolbox and Benchmark
环境配置
- 1.下载mmcv
pip install -U openmim mim install mmcv-full
- 2.手动下载代码到本地或者使用git clone https://github.com/open-mmlab/mmdetection.git
- 3.进入mmdetection目录:cd mmdetection
- 4.pip install -v -e .
如需使用 下载对应版本的mmcv
eg:
mim install mmcv-full==1.7.0
mmdetection支持coco和voc数据格式,建议使用coco数据格式
1.需要修改的地方
1.mmdet/datasets/coco.py中的classes和PALETTE需要改为自己的类别和相应的颜色
2.mmdet/core/evaluation/class_names.py中的coco_classes改为自己的类别