官网链接:https://github.com/open-mmlab/mmdetection
说明文档:https://mmdetection.readthedocs.io/en/latest/
预训练模型:https://github.com/open-mmlab/mmdetection/blob/master/docs/MODEL_ZOO.md
配置要求
- Linux 16
- Python 3.5+
- PyTorch 1.1 or higher
- CUDA 9.0 or higher
- NCCL 2
- GCC 4.9 or higher
- mmcv
查看方式
命令 | |
---|---|
查看 cuda 版本 | nvcc -V 或者 cat /usr/local/cuda/version.txt |
查看 python 版本 | python -V 或者 python |
查看 gcc 版本 | gcc -version |
查看系统版本号 | lsb_release -a |
查看主板信息 | lspci |
查看CPU信息 | cat /proc/cpuinfo |
查看硬盘型号 | cat /proc/ide #(IDE硬盘) 或者 cat /proc/scsi #(SCSI硬盘) |
查看内存信息 | cat /proc/meminfo |
查看显卡信息 | nvidia-smi |
查看 物理cpu个数 | grep 'physical id' /proc/cpuinfo | sort -u |
查看 核心数量 | grep 'core id' /proc/cpuinfo | sort -u | wc -l |
查看 线程数 | grep 'processor' /proc/cpuinfo | sort -u | wc -l |
安装步骤
建议新建一个环境,否则容易出错。
1、新建环境
conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab # 或 source activate open-mmlab
2、安装 pytorch
# pytorch官网:https://pytorch.org/get-started/locally/
conda install pytorch=1.0 torchvision -c pytorch
3、安装 mmcv
mmcv是mmdetection的一个CV库,提供了很多基本功能,后续你也可以根据自己的需要对这个库进行修改,但是每次修改好你都需要pip install .
, 否则修改不起作用 。
git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
pip install .
4、下载 mmdetection
# 若git速度慢 ,可手动下载解压,网址:https://github.com/open-mmlab/mmdetection
git clone https://github.com/open-mmlab/mmdetection.git
# 打开 mmdetection 所在文件夹
cd mmdetection
5、安装Cython,opencv及requirements.txt
中的相关依赖
! 注意【 安装顺序 】
由于限速问题,下载速度很慢,建议换成国内清华源
或者中科大源
。如果还是不行 ,只能离线安装,建议到 PyPI 手动下载需要的whl或tar文件,在你刚刚建立的环境下进行pip install xxx.whl
进行安装。
如果不安装,在后面编译的时候会自动安装,速度巨慢,容易报错,运行一半就卡住了。如果你的网很快,就当我没说。
# 换源
sudo gedit ~/.condarc
# 添加以下源,并保存
channels:
- https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
- https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- defaults
# 显示源
show_channel_urls: true
6、编译
两种版本的编译方法不同,注意区别。现在大多都是1.1版本。
# PyTorch version 1.1 mmdetection
python setup.py develop # or "pip install -v -e ."
# PyTorch-0.4.1 version mmdetection
./compile.sh
python setup.py install