apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-14 04:44:00 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[Model] Update PPDetection RKNPU2 (#1323)

Browse Source 
* 更新docs

* 修正docs错误

* 更新docs

* 更新python example脚本和ppyoloe转换脚本
This commit is contained in:
Zheng-Bicheng
2023-02-14 17:34:41 +08:00
committed by GitHub
parent 41c5f2de5a
commit fc6edcc541
6 changed files with 99 additions and 126 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/vision/detection/paddledetection/rknpu2/python/infer.py
View File

@@ -45,15 +45,16 @@ if __name__ == "__main__":
# 配置runtime，加载模型
runtime_option = fd.RuntimeOption()
runtime_option.use_cpu()
runtime_option.use_rknpu2()
model = fd.vision.detection.PPYOLOE(
model_file,
params_file,
config_file,
runtime_option=runtime_option,
model_format=fd.ModelFormat.ONNX)
model_format=fd.ModelFormat.RKNN)
model.preprocessor.disable_normalize()
model.preprocessor.disable_permute()
model.postprocessor.apply_decode_and_nms()
# 预测图片分割结果