Skip to content

GitLab

Commit 30d56419 authored by liuhy's avatar liuhy
Browse files

修改Readme

parent 921cc06f
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
LPRNet_ORT_infer.py
View file @ 30d56419
...@@ -44,7 +44,7 @@ def LPRNetPostprocess(infer_res): ...@@ -44,7 +44,7 @@ def LPRNetPostprocess(infer_res):
return result return result
def LPRNetInference(args): def LPRNetInference(args):
if ort.get_device() == "GPU": if ort.get_device() == "GPU-MIGRAPHX":
sess = ort.InferenceSession(args.model, providers=['ROCMExecutionProvider'],) #DCU版本 sess = ort.InferenceSession(args.model, providers=['ROCMExecutionProvider'],) #DCU版本
else: else:
sess = ort.InferenceSession(args.model, providers=['CPUExecutionProvider']) # CPU版本 sess = ort.InferenceSession(args.model, providers=['CPUExecutionProvider']) # CPU版本
......
README.md
View file @ 30d56419
...@@ -81,9 +81,9 @@ LPRNet_migraphx_infer.py是基于Migraphx的推理脚本，使用需安装好Mig ...@@ -81,9 +81,9 @@ LPRNet_migraphx_infer.py是基于Migraphx的推理脚本，使用需安装好Mig
| Engine | Model Path| Model Format | Accuracy(%) | Speed(ms) | | Engine | Model Path| Model Format | Accuracy(%) | Speed(ms) |
| :------: | :------: | :------: | :------: |:------: | | :------: | :------: | :------: | :------: |:------: |
| ONNXRuntime | model/LPRNet.onnx | onnx | 91.0 | 37.62 | | ONNXRuntime | model/LPRNet.onnx | onnx | 91.0 | 2.59 |
| Migraphx | model/LPRNet.onnx |onnx | 91.0 | 3.70 | | Migraphx | model/LPRNet.onnx |onnx | 91.0 | 2.66 |
| Migraphx | model/LPRNet.mxr |mxr | 91.0 | 3.27 | | Migraphx | model/LPRNet.mxr |mxr | 91.0 | 2.49 |
## 参考 ## 参考
* [LPRNet_Pytorch](https://github.com/sirius-ai/LPRNet_Pytorch) * [LPRNet_Pytorch](https://github.com/sirius-ai/LPRNet_Pytorch)
* [license-plate-detect-recoginition](https://github.com/qzpzd/license-plate-detect-recoginition) * [license-plate-detect-recoginition](https://github.com/qzpzd/license-plate-detect-recoginition)
model/LPRNet.mxr
View file @ 30d56419
No preview for this file type
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment