Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
huastant
LPR
Commits
296a1a18
You need to sign in or sign up before continuing.
Commit
296a1a18
authored
Mar 03, 2023
by
刘明贵
Browse files
修改readme
parent
6be6ff06
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
9 deletions
+4
-9
README.md
README.md
+4
-9
No files found.
README.md
View file @
296a1a18
...
...
@@ -2,7 +2,7 @@
*
@Author: liuhy
*
@email: liuhy6@sugon.com
*
@Date: 2023-03-03 10:17:07
*
@LastEditTime: 2023-03-03 11:1
2:5
7
*
@LastEditTime: 2023-03-03 11:1
4:1
7
*
@FilePath:
\l
pr
\R
EADME.md
-->
# License-Plate-Recoginition(LPR)
...
...
@@ -41,12 +41,7 @@ Fine-tuning时只训练从起始epoch到最大epoch。
LPR模型用test.py对训练出的模型进行测试,使用方法如下:
`
python test.py
\
--model 需要测试的pth模型路径
\
--imgpath 测试集路径 # 单张图像或文件夹皆可
--export_onnx 该参数用于选择是否需要将pth模型转为onnx模型
--dynamic 该参数用于选择onnx模型是否使用动态的batch size
python test.py --model 需要测试的pth模型路径 --imgpath 测试集路径(文件夹或图像皆可) --export_onnx True/False(该参数用于选择是否需要将pth模型转为onnx模型) --dynamic True/False(该参数用于选择onnx模型是否使用动态的batch size)
`
### 推理
我们分别提供了基于OnnxRuntime(ORT)和Migraphx的推理脚本,版本依赖:
...
...
@@ -56,13 +51,13 @@ python test.py \
LPRNet_ORT_infer.py是基于ORT的的推理脚本,使用方法:
`
python LPRNet_ORT_infer.py --model onnx模型路径 --imgpath 数据路径(文件夹图像皆可)
python LPRNet_ORT_infer.py --model onnx模型路径 --imgpath 数据路径(文件夹
或
图像皆可)
`
#### Migraphx
LPRNet_migraphx_infer.py是基于Migraphx的推理脚本,使用需安装好Migraphx,支持onnx模型和mxr模型推理,mxr模型是migraphx将onnx模型保存成的离线推理引擎,初次使用onnx模型会保存对应的mxr模型。使用方法:
`
python LPRNet_migraphx_infer.py --model mxr/onnx模型路径 --imgpath 数据路径(文件夹图像皆可) --savepath mxr模型的保存路径以及模型名称
python LPRNet_migraphx_infer.py --model mxr/onnx模型路径 --imgpath 数据路径(文件夹
或
图像皆可) --savepath mxr模型的保存路径以及模型名称
`
## 性能和准确率数据
测试数据使用的是
[
LPRNet_Pytorch
](
https://github.com/sirius-ai/LPRNet_Pytorch/tree/master/data/test
)
,使用的加速卡是DCU Z100。
**mxr格式的模型是migraphx创建的onnx模型的离线引擎。**
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment