这份文档介绍了如何安装PaddleDetection及其依赖项(包括PaddlePaddle)。
PaddleDetection的相关信息,请参考README.md.
环境需求:
- OS 64位操作系统
- Python2 >= 2.7.15 or Python 3(3.5.1+/3.6/3.7),64位版本
- pip/pip3(9.0.1+),64位版本操作系统是
- CUDA >= 9.0
- cuDNN >= 7.6
如果需要 GPU 多卡训练,请先安装NCCL(Windows暂不支持nccl)。
PaddleDetection 依赖 PaddlePaddle 版本关系:
PaddleDetection版本 | PaddlePaddle版本 | 备注 |
---|---|---|
v0.3 | >=1.7 | -- |
v0.4 | >= 1.8.4 | PP-YOLO依赖1.8.4 |
# install paddlepaddle
python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple
# 如果您的机器安装的是CUDA9,请运行以下命令安装
python -m pip install paddlepaddle-gpu==1.8.4.post97 -i https://mirror.baidu.com/pypi/simple
如果您的机器安装的是CUDA10,请运行以下命令安装
python -m pip install paddlepaddle-gpu==1.8.4.post107 -i https://mirror.baidu.com/pypi/simple
如果您的机器是CPU,请运行以下命令安装
python -m pip install paddlepaddle==1.8.4 -i https://mirror.baidu.com/pypi/simple
更多的安装方式如conda, docker安装,请参考安装文档中的说明进行操作
请确保您的PaddlePaddle安装成功并且版本不低于需求版本。使用以下命令进行验证。
# 在您的Python解释器中确认PaddlePaddle安装成功
>>> import paddle.fluid as fluid
>>> fluid.install_check.run_check()
# 确认PaddlePaddle版本
python -c "import paddle; print(paddle.__version__)"
PaddleDetection
在评估时若使用COCO评估标准,则需要安装COCO-API ,安装方式如下:
# 安装pycocotools。若使用了虚拟环境,请使用虚拟环境中的pip,或者指定pip绝对路径进行安装
pip install pycocotools
如果windows用户按照上面方式安装COCO-API出错,可参考以下方式:
-
安装 pycocotools-windows
# pip install pycocotools-windows pip install pycocotools-windows
- 如果您遇到ssl问题,请参考conda issue #8273
- 如果您的网络无法下载安装包:
- 设置pip源pip issue #1736 。
- 将安装包下载到本地安装。从pycocotools-windows 下载对应安装包,在本地安装:
# 例如安装 pycocotools_windows-2.0.0.2-cp37-cp37m-win_amd64.whl pip install pycocotools_windows-2.0.0.2-cp37-cp37m-win_amd64.whl
-
从
pycocotools
源码安装# 若Cython未安装,请安装Cython pip install Cython # 由于原版cocoapi不支持windows,采用第三方实现版本,该版本仅支持Python3 pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
安装完成后,验证是否安装成功:
python -c "import pycocotools"
安装Python依赖库:
Python依赖库在requirements.txt 中给出,可通过如下命令安装:
pip install -r requirements.txt
克隆PaddleDetection库:
您可以通过以下命令克隆PaddleDetection:
cd <path/to/clone/PaddleDetection>
git clone https://github.com/PaddlePaddle/PaddleDetection.git
提示:
也可以通过 https://gitee.com/paddlepaddle/PaddleDetection 克隆。
cd <path/to/clone/PaddleDetection>
git clone https://gitee.com/paddlepaddle/PaddleDetection
确认测试通过:
python ppdet/modeling/tests/test_architectures.py
测试通过后会提示如下信息:
..........
----------------------------------------------------------------------
Ran 12 tests in 2.480s
OK (skipped=2)
预训练模型预测
使用预训练模型预测图像,快速体验模型预测效果:
# use_gpu参数设置是否使用GPU
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_640x640.jpg
会在output
文件夹下生成一个画有预测结果的同名图像。
结果如下图: