mmdetection是商汤科技开源的用于深度学习目标检测的库
首先这个库是依赖pytorch的,因此需要先安装pytorch,ubuntu16上使用pip安装即可:
sudo pip3 install torch torchvision
然后安装mmdetection
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
sudo pip install cython
./compile.sh
sudo python3 setup.py install
在coco数据及上训练的话需要下载coco数据集,放置于新建的data/coco下并解压,解压前总计大小为26.8G
训练脚本为
python3 tools/train.py configs/faster_rcnn_r50_fpn_1x.py --gpus 2 --validate