apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 08:37:06 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity
Files
9e20dab0d6929a223f05601c33da668d75ff6d59
FastDeploy/examples/vision/detection/paddledetection/sophgo/python
History
Yi-sir 9e20dab0d6 [Example] Merge Download Paddle Model, Paddle->ONNX->MLIR->BModel (#1643) 
* 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>
2023-03-31 15:08:01 +08:00
..
infer_picodet.py
[Example] Merge Download Paddle Model, Paddle->ONNX->MLIR->BModel (#1643)
2023-03-31 15:08:01 +08:00
infer_ppyoloe.py
[Example] Merge Download Paddle Model, Paddle->ONNX->MLIR->BModel (#1643)
2023-03-31 15:08:01 +08:00
infer_yolov8.py
[SOPHGO] Add PaddleDetection YOLOv8 example (#1165)
2023-01-30 11:47:07 +08:00
prepare_npz.py
[Example] Merge Download Paddle Model, Paddle->ONNX->MLIR->BModel (#1643)
2023-03-31 15:08:01 +08:00
README.md
[Example] Merge Download Paddle Model, Paddle->ONNX->MLIR->BModel (#1643)
2023-03-31 15:08:01 +08:00

README.md

PaddleDetection Python部署示例

在部署前，需确认以下步骤

本目录下提供infer_ppyoloe.py,infer_yolov8.py和infer_picodet.py快速完成 PP-YOLOE ,PP-YOLOV8和 PicoDet 在SOPHGO TPU上部署的示例。执行如下脚本即可完成

# 下载部署示例代码
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中

其它文档