forked from PulseFocusPlatform/PulseFocusPlatform
135 lines
4.2 KiB
Markdown
135 lines
4.2 KiB
Markdown
English | [简体中文](INSTALL_cn.md)
|
||
|
||
# Installation
|
||
|
||
|
||
This document covers how to install PaddleDetection and its dependencies
|
||
(including PaddlePaddle), together with COCO and Pascal VOC dataset.
|
||
|
||
For general information about PaddleDetection, please see [README.md](https://github.com/PaddlePaddle/PaddleDetection/tree/develop).
|
||
|
||
## Requirements:
|
||
|
||
- PaddlePaddle 2.1
|
||
- OS 64 bit
|
||
- Python 3(3.5.1+/3.6/3.7/3.8/3.9),64 bit
|
||
- pip/pip3(9.0.1+), 64 bit
|
||
- CUDA >= 10.1
|
||
- cuDNN >= 7.6
|
||
|
||
|
||
Dependency of PaddleDetection and PaddlePaddle:
|
||
|
||
| PaddleDetection version | PaddlePaddle version | tips |
|
||
| :----------------: | :---------------: | :-------: |
|
||
| release/2.1 | >= 2.1.0 | Dygraph mode is set as default |
|
||
| release/2.0 | >= 2.0.1 | Dygraph mode is set as default |
|
||
| release/2.0-rc | >= 2.0.1 | -- |
|
||
| release/0.5 | >= 1.8.4 | Cascade R-CNN and SOLOv2 depends on 2.0.0.rc |
|
||
| release/0.4 | >= 1.8.4 | PP-YOLO depends on 1.8.4 |
|
||
| release/0.3 | >=1.7 | -- |
|
||
|
||
|
||
## Instruction
|
||
|
||
### 1. Install PaddlePaddle
|
||
|
||
```
|
||
|
||
# CUDA10.1
|
||
python -m pip install paddlepaddle-gpu==2.1.0.post101 -f https://paddlepaddle.org.cn/whl/mkl/stable.html
|
||
|
||
# CPU
|
||
python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple
|
||
```
|
||
|
||
- For more CUDA version or environment to quick install, please refer to the [PaddlePaddle Quick Installation document](https://www.paddlepaddle.org.cn/install/quick)
|
||
- For more installation methods such as conda or compile with source code, please refer to the [installation document](https://www.paddlepaddle.org.cn/documentation/docs/en/install/index_en.html)
|
||
|
||
Please make sure that your PaddlePaddle is installed successfully and the version is not lower than the required version. Use the following command to verify.
|
||
|
||
```
|
||
# check
|
||
>>> import paddle
|
||
>>> paddle.utils.run_check()
|
||
|
||
# confirm the paddle's version
|
||
python -c "import paddle; print(paddle.__version__)"
|
||
```
|
||
|
||
**Note**
|
||
|
||
1. If you want to use PaddleDetection on multi-GPU, please install NCCL at first.
|
||
|
||
|
||
### 2. Install PaddleDetection
|
||
|
||
PaddleDetection can be installed in the following two ways:
|
||
|
||
#### 2.1 Install via pip
|
||
|
||
**Note:** Installing via pip only supports Python3
|
||
|
||
```
|
||
# Install paddledet via pip
|
||
pip install paddledet==2.1.0 -i https://mirror.baidu.com/pypi/simple
|
||
|
||
# Download and use the configuration files and code examples in the source code
|
||
git clone https://github.com/PaddlePaddle/PaddleDetection.git
|
||
cd PaddleDetection
|
||
```
|
||
|
||
#### 2.2 Compile and install from Source code
|
||
|
||
```
|
||
# Clone PaddleDetection repository
|
||
cd <path/to/clone/PaddleDetection>
|
||
git clone https://github.com/PaddlePaddle/PaddleDetection.git
|
||
|
||
# Compile and install paddledet
|
||
cd PaddleDetection
|
||
python setup.py install
|
||
|
||
# Install other dependencies
|
||
pip install -r requirements.txt
|
||
|
||
```
|
||
|
||
**Note**
|
||
|
||
1. If you are working on Windows OS, `pycocotools` installing may failed because of the origin version of cocoapi does not support windows, another version can be used used which only supports Python3:
|
||
|
||
```pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI```
|
||
|
||
2. If you are using Python <= 3.6, `pycocotools` installing may failed with error like `distutils.errors.DistutilsError: Could not find suitable distribution for Requirement.parse('cython>=0.27.3')`, please install `cython` firstly, for example `pip install cython`
|
||
|
||
After installation, make sure the tests pass:
|
||
|
||
```shell
|
||
python ppdet/modeling/tests/test_architectures.py
|
||
```
|
||
|
||
If the tests are passed, the following information will be prompted:
|
||
|
||
```
|
||
.....
|
||
----------------------------------------------------------------------
|
||
Ran 5 tests in 4.280s
|
||
OK
|
||
```
|
||
|
||
## Inference demo
|
||
|
||
**Congratulation!** Now you have installed PaddleDetection successfully and try our inference demo:
|
||
|
||
```
|
||
# Predict an image by GPU
|
||
export CUDA_VISIBLE_DEVICES=0
|
||
python tools/infer.py -c configs/ppyolo/ppyolo_r50vd_dcn_1x_coco.yml -o use_gpu=true weights=https://paddledet.bj.bcebos.com/models/ppyolo_r50vd_dcn_1x_coco.pdparams --infer_img=demo/000000014439.jpg
|
||
```
|
||
|
||
An image of the same name with the predicted result will be generated under the `output` folder.
|
||
The result is as shown below:
|
||
|
||
![](../images/000000014439.jpg)
|