mmdetection安装和使用

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

猜你喜欢

转载自blog.csdn.net/minstyrain/article/details/83054845