一、使用环境
Ubuntu16.04
Cuda9.0 + cudnn7.0
Python3.7
Anaconda3
商汤科技和香港中文大学联合开源的深度学习目标检测工具箱mmdetection
二、安装
1.使用Anaconda创建虚拟环境:mmdetection
conda create -n mmdetection python=3.7 -y
source activate mmdetection
此时,虚拟环境已经被创建并且激活。由于最近Anaconda清华大学镜像服务又被重新授权,所以大家可以继续使用清华源。
2.安装Pytorch1.1
打开pytorch官网,选择对应的pytorch安装环境,例如笔者的

执行这条语句即可安装pytorch1.1,mmdetection是支持这个版本的。
安装完成后,可执行下面的语句来测试是否安装成功。
python
>>import torch
>>import torchvision
3.安装Cython
conda install cython
4.安装mmcv
git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip in

本文详细介绍了在Ubuntu16.04环境下,如何使用Anaconda3创建虚拟环境并安装mmdetection,包括Pytorch1.1、Cython和mmcv的安装。同时,讲解了如何准备PASCAL VOC格式的数据集,调整mmdetection代码以适配自定义数据集,并修改config文件。最后,提到了训练和测试的步骤。

3615

被折叠的 条评论
为什么被折叠?



