1.安装
- 强烈建议使用anaconda管理python,参考,我使用python3.6的虚拟环境
- 先检查平台,一般没问题:
python3 -c "import platform;print(platform.architecture()[0]);print(platform.machine())"
第一行输出的是”64bit”,第二行输出的是”x86_64”、”x64”或”AMD64”即可
- 安装cuda10.0版本的PaddlePaddle:
python -m pip install paddlepaddle-gpu==2.0.2.post100 -f https://paddlepaddle.org.cn/whl/mkl/stable.html
,其他版本的命令在这里 - 验证PaddlePaddle安装
python3
import paddle
paddle.utils.run_check()
如果出现 PaddlePaddle is installed successfully!,说明已成功安装PaddlePaddle。
- 其他依赖安装:
pip install pycocotools
- PaddleDetection
1. git clone https://github.com/PaddlePaddle/PaddleDetection.git
2. git checkout release/0.5
3. pip install -r requirements.txt #出现timeout的可使用conda安装
4. 确认测试通过:python ppdet/modeling/tests/test_architectures.py
正确输出:
..........
----------------------------------------------------------------------
Ran 12 tests in 2.480s
OK (skipped=2)
5. 预训练模型预测: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
会在output文件夹下生成一个画有预测结果的同名图像。
2.使用