mmdetection

1、本人的系统环境:
Ubuntu 16.04+Cuda 9.0 + Cudnn 7.0.5+Python 3.6 (mmdetection要求Python版本需要3.4+)+Anaconda 3 +spyder
2、在安装mmdetection之前,需要安装以下几个依赖库:
PyTorch 0.4.1
torchvision
Cython
mmcv
(1)新建conda环境

conda create -n pytouch python=3.6
source activate pytouch
conda install yaml  # 否则 在mmdection 的python setup.py install 会有 fatal error

(2)安装Pytorch 0.4.1和torchvision -GPU

conda install pytorch=0.4.1 torchvision -c pytorch

安装好以后,进入Python环境,输入以下代码测试是否安装成功,不报错则说明安装成功

import torch
import torchvision

(3)安装Cython

conda install  cython

(4)安装mmcv

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install .

(5)安装mmdetection

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
./compile.sh
python setup.py install

到此,我们就完成了mmdetection及其依赖库的安装
(6)pytouch 环境下安装spyder

sudo pip3 install spyder
sudo apt-get install python3-pyqt*

(7)测试Demo

你可能感兴趣的:(mmdetection)