mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
9e20dab0d6
* fix infer.py and README * [Example] Merge Download Paddle Model, Paddle->Onnx->Mlir->Bmodel and inference into infer.py. Modify README.md * modify pp_liteseg sophgo infer.py and README.md * fix PPOCR,PPYOLOE,PICODET,LITESEG sophgo infer.py and README.md * fix memory overflow problem while inferring with sophgo backend * fix memory overflow problem while inferring with sophgo backend --------- Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com> Co-authored-by: xuyizhou <yizhou.xu@sophgo.com>
45 lines
2.2 KiB
Markdown
45 lines
2.2 KiB
Markdown
# PaddleDetection Python部署示例
|
||
|
||
在部署前，需确认以下步骤
|
||
|
||
- 1. 软硬件环境满足要求，参考[FastDeploy环境要求](../../../../../../docs/cn/build_and_install/sophgo.md)
|
||
|
||
本目录下提供`infer_ppyoloe.py`,`infer_yolov8.py`和`infer_picodet.py`快速完成 PP-YOLOE ,PP-YOLOV8和 PicoDet 在SOPHGO TPU上部署的示例。执行如下脚本即可完成
|
||
|
||
```bash
|
||
# 下载部署示例代码
|
||
git clone https://github.com/PaddlePaddle/FastDeploy.git
|
||
cd FastDeploy/examples/vision/detection/paddledetection/sophgo/python
|
||
|
||
# 下载图片
|
||
wget https://gitee.com/paddlepaddle/PaddleDetection/raw/release/2.4/demo/000000014439.jpg
|
||
|
||
# 推理
|
||
# ppyoloe推理示例
|
||
# 指定--auto True，自动完成模型准备、转换和推理，需要指定PaddleDetection路径
|
||
python3 infer_ppyoloe.py --auto True --pp_detect_path {Path to PaddleDetection} --model_file '' --config_file '' --image ''
|
||
|
||
# 指定--auto False，需要用户指定模型、配置文件和图片路径，不需要指定PaddleDetection路径。
|
||
python3 infer_ppyoloe.py --auto False --pp_detect_path '' --model_file model/ppyoloe_crn_s_300e_coco_1684x_f32.bmodel --config_file model/infer_cfg.yml --image ./000000014439.jpg
|
||
|
||
# picodet推理示例
|
||
# 指定--auto True，自动完成模型准备、转换和推理，需要指定PaddleDetection路径
|
||
python3 infer_picodet.py --auto True --pp_detect_path {Path to PaddleDetection} --model_file '' --config_file '' --image ''
|
||
|
||
# 指定--auto False，需要用户指定模型、配置文件和图片路径，不需要指定PaddleDetection路径。
|
||
python3 infer_picodet.py --auto False --pp_detect_path '' --model_file model/ppyoloe_crn_s_300e_coco_1684x_f32.bmodel --config_file model/infer_cfg.yml --image ./000000014439.jpg
|
||
|
||
# yolov8推理示例
|
||
python3 infer_yolov8.py --model_file model/yolov8s_s_300e_coco_1684x_f32.bmodel --config_file model/infer_cfg.yml --image ./000000014439.jpg
|
||
# 运行完成后返回结果如下所示
|
||
可视化结果存储在sophgo_result.jpg中
|
||
```
|
||
|
||
## 其它文档
|
||
- [PP-YOLOE C++部署](../cpp)
|
||
- [PicoDet C++部署](../cpp)
|
||
- [YOLOV8 C++部署](../cpp)
|
||
- [转换PicoDet SOPHGO模型文档](../README.md)
|
||
- [转换PP-YOLOE SOPHGO模型文档](../README.md)
|
||
- [转换YOLOV8 SOPHGO模型文档](../README.md)
|