INSTALL_cn.md 3.5 KB
Newer Older
1
[English](INSTALL.md) | 简体中文
Q
qingqing01 已提交
2 3


4
# 安装文档
Q
qingqing01 已提交
5

6
本文档包含了如何安装PaddleDetection以及相关依赖
Q
qingqing01 已提交
7

W
wangguanzhong 已提交
8
其他更多PaddleDetection信息,请参考[README.md](https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.0).
Q
qingqing01 已提交
9

10
## 环境要求
Q
qingqing01 已提交
11

12
- PaddlePaddle 2.0.1
Q
qingqing01 已提交
13 14
- OS 64位操作系统
- Python 3(3.5.1+/3.6/3.7),64位版本
F
Feng Ni 已提交
15
- pip/pip3(9.0.1+),64位版本
Q
qingqing01 已提交
16 17 18 19
- CUDA >= 9.0
- cuDNN >= 7.6


20
## 安装说明
Q
qingqing01 已提交
21

22
建议使用docker环境安装PaddleDetection并开启你的目标检测之旅。请按照如下步骤说明进行安装,如果您希望使用本机环境,可以跳过步骤1.
Q
qingqing01 已提交
23

24
### 1. (推荐)准备docker环境
Q
qingqing01 已提交
25

26
已CUDA10.1, CUDNN7.6为例
Q
qingqing01 已提交
27

28 29 30
```bash
# 首先拉去PaddlePaddle镜像
sudo docker pull paddlepaddle/paddle:latest-dev-cuda10.1-cudnn7-gcc82
Q
qingqing01 已提交
31

32 33
# 切换到工作目录
cd /home/work
Q
qingqing01 已提交
34

35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
# 创建ppdet容器
# 将存放数据的当前目录映射到容器中的/ppdet目录中
sudo nvidia-docker run --name ppdet -v $PWD:/paddle --privileged --shm-size=4G --network=host -it paddlepaddle/paddle:latest-dev-cuda10.1-cudnn7-gcc82 /bin/bash
```

可以在[DockerHub](https://hub.docker.com/r/paddlepaddle/paddle/tags/) 中找到匹配您机器环境的镜像

```
# ctrl+P+Q 退出容器, 使用如下命令重新进入docker环境:
sudo docker exec -it ppdet /bin/bash
```

其他更多docker用法,请参考PaddlePaddle[文档](https://www.paddlepaddle.org.cn/documentation/docs/zh/install/docker/fromdocker.html)


### 安装PaddlePaddle

```
# CUDA9.0
python -m pip install paddlepaddle-gpu==2.0.1.post90 -i https://mirror.baidu.com/pypi/simple

# CUDA10.1
python -m pip install paddlepaddle-gpu==2.0.1.post101 -f https://mirror.baidu.com/pypi/simple

# CPU
python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple
```
Q
qingqing01 已提交
62

63
更多安装方式例如conda或源码编译安装方法,请参考PaddlePaddle[安装文档](https://www.paddlepaddle.org.cn/documentation/docs/zh/install/index_cn.html)
Q
qingqing01 已提交
64

65
请确保您的PaddlePaddle安装成功并且版本不低于需求版本。使用以下命令进行验证。
Q
qingqing01 已提交
66

67 68 69 70 71 72 73 74 75 76
```
# 在您的Python解释器中确认PaddlePaddle安装成功
>>> import paddle
>>> paddle.utils.run_check()

# 确认PaddlePaddle版本
python -c "import paddle; print(paddle.__version__)"
```
**注意**
1. 如果您希望在多卡环境下使用PaddleDetection,请首先安装NCCL
Q
qingqing01 已提交
77

78
### 3. 安装PaddleDetection
Q
qingqing01 已提交
79 80

```
81
# 克隆PaddleDetection仓库
Q
qingqing01 已提交
82 83
cd <path/to/clone/PaddleDetection>
git clone https://github.com/PaddlePaddle/PaddleDetection.git
84 85 86 87 88 89 90

# 安装其他依赖
pip install -r requirements.txt

# 安装PaddleDetection
cd PaddleDetection
python setup.py install
Q
qingqing01 已提交
91 92
```

93 94 95 96 97 98 99 100 101
**注意**

1. 由于原版cocoapi不支持windows,采用第三方实现版本,该版本仅支持Python3

    ```pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI```


安装后确认测试通过:

Q
qingqing01 已提交
102
```
103
python ppdet/modeling/tests/test_architectures.py
Q
qingqing01 已提交
104
```
K
Kaipeng Deng 已提交
105

106
测试通过后会提示如下信息:
K
Kaipeng Deng 已提交
107 108

```
109 110 111 112
..........
----------------------------------------------------------------------
Ran 12 tests in 2.480s
OK (skipped=2)
K
Kaipeng Deng 已提交
113
```
F
Feng Ni 已提交
114

115
## 快速体验
F
Feng Ni 已提交
116

117
**恭喜!** 您已经成功安装了PaddleDetection,接下来快速体验目标检测效果
F
Feng Ni 已提交
118 119

```
120 121 122
# 在GPU上预测一张图片
export CUDA_VISIBLE_DEVICES=0
python tools/infer.py -c configs/ppyolo/ppyolo.yml -o use_gpu=true weights=https://paddlemodels.bj.bcebos.com/object_detection/ppyolo.pdparams --infer_img=demo/000000014439.jpg
F
Feng Ni 已提交
123 124 125 126 127 128
```

会在`output`文件夹下生成一个画有预测结果的同名图像。

结果如下图:

129
![](../images/000000014439.jpg)