admin管理员组

文章数量:1593471

安装说明

Paddledetection百度官方的安装示例对于Win10系统存在不成功的情况,主要是COCO-API的安装方式,这里加入了从源码编译安装COCO-API的方法

目录

  • 安装PaddlePaddle
  • 安装COCO-API
  • PaddleDetection

安装PaddlePaddle

环境需求:

  • 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

PaddleDetection 依赖 PaddlePaddle 版本关系:

PaddleDetection版本PaddlePaddle版本备注
v0.3>=1.7
v0.4>= 1.8.4PP-YOLO依赖1.8.4

首先使用Anaconda创建一个PaddlePaddle安装的环境,这样方便进行版本管理

# create paddlepaddle environment
conda create -n your_env_name python=3.6

# activate environment
activate your_env_name

# install paddlepaddle
pip install paddlepaddle -i https://mirror.baidu/pypi/simple

# 如果您的机器安装的是CUDA9,请运行以下命令安装
pip install paddlepaddle-gpu==1.8.4.post97 -i https://mirror.baidu/pypi/simple

如果您的机器安装的是CUDA10,请运行以下命令安装
pip install paddlepaddle-gpu==1.8.4.post107 -i https://mirror.baidu/pypi/simple

如果您的机器是CPU,请运行以下命令安装
pip install paddlepaddle==1.8.4 -i https://mirror.baidu/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__)"

安装成功后效果如下

安装COCO-API

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/philferriere/cocoapi.git#subdirectory=PythonAPI
    
  • 方案三 下载源码,自行编译,需要事先安装VS工具

    # 下载源码,自行编译,或者下载压缩包解压
    git clone https://github/philferriere/cocoapi.git
    
    # install pycocotools locally
    python setup.py build_ext --inplace
    
    # install pycocotools to the Python site-packages
    python setup.py build_ext install
    
    • 如果出现vcvarsall.bat找不到,修改visual 2017 installer 添加 适用于桌面 的 VC++ 2015.3 v14.00 (v140) 工具集https://devblogs.microsoft/python/unable-to-find-vcvarsall-bat/
    • 如果出现LINK : fatal error LNK1158: 无法运行 ‘rc.exe’,添加环境变量C:\Program Files (x86)\Windows Kits\8.1\bin\x86
      安装完成后,验证是否安装成功:
python -c "import pycocotools"

如果未出现任何报错,就说明安装成功了【注意:命令不会弹出任何提示信息】

PaddleDetection

安装Python依赖库:

Python依赖库在requirements.txt 中给出,可通过如下命令安装:

pip install -r requirements.txt

克隆PaddleDetection库:

您可以通过以下命令克隆PaddleDetection:

cd <path/to/clone/PaddleDetection>
git clone https://github/PaddlePaddle/PaddleDetection.git

提示:

也可以通过 https://gitee/paddlepaddle/PaddleDetection 克隆。

cd <path/to/clone/PaddleDetection>
git clone https://gitee/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/object_detection/ppyolo.pdparams --infer_img=demo/000000014439_640x640.jpg

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

结果如下图:

参考来源百度官方安装示例

本文标签: 环境飞桨Paddledetection