MMdetection香港中文大学MMLab最近开源了一个基于Pytorch实现的深度学习目标检测工具箱,支持Faster-RCNN,Mask-RCNN,Fast-RCNN等主流的目标检测框架,本文主要讲解MMdetection在滴滴云的机器学习工作室(DAI)下的安装方法。
滴滴云的机器学习工作室(DAI),内置了丰富的机器学习开发环境,安装MMetection框架,总共需要3大步骤:
1、切换到所需环境,确认GCC环境
新建terminal,切换到pytorch环境:
conda info -e conda activate ipykernel_py3_pytorch1.4
保证GCC>=5.0:
gcc -v
如果GCC<5.0:
sudo yum install centos-release-scl sudo yum install devtoolset-7-gcc* scl enable devtoolset-7 bash gcc -v
2.创建环境变量
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} export CUDA_HOME=/usr/local/cuda export CFLAGS="-I$CUDA_HOME/include $CFLAGS"
3.安装mmdetection:
pip install pycocotools pip install mmcv git clone https://github.com/open-mmlab/mmdetection cd mmdetection python setup.py develop