mmdetection安装和使用

然后安装mmdetectionmmdetection是商汤科技开源的用于深度学习目标检测的库,不得不说商汤还是挺挫的,居然不支持Windows,也不支持MAC,吹的牛逼震天,结果啥也用不上,这不扯淡么;python2也不支持,这NM不太好吧.

我们现看下配置要求

  • Linux (Windows is not officially supported)
  • Python 3.5+ (Python 2 is not supported)
  • PyTorch 1.1 or higher
  • CUDA 9.0 or higher
  • NCCL 2
  • GCC(G++) 4.9 or higher
  • mmcv

首先这个库是依赖pytorch的,因此需要先安装pytorch,ubuntu16上使用pip安装即可:

sudo pip3 install torch torchvision

然后克隆代码安装 

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
sudo pip3 install Cython
sudo pip3 install -r requirements.txt 
sudo python3 setup.py develop

在coco数据及上训练的话需要下载coco数据集,放置于新建的data/coco下并解压,解压前总计大小为26.8G

训练脚本为

python3 tools/train.py configs/faster_rcnn_r50_fpn_1x.py --gpus 2 --validate

 

你可能感兴趣的:(深度学习)