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
ModelZoo
LPRNet_pytorch
Commits
c4f4abde
Commit
c4f4abde
authored
Mar 03, 2023
by
刘明贵
Browse files
修改readme
parent
967cf1b8
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
3 deletions
+7
-3
README.md
README.md
+7
-3
No files found.
README.md
View file @
c4f4abde
...
...
@@ -2,7 +2,7 @@
*
@Author: liuhy
*
@email: liuhy6@sugon.com
*
@Date: 2023-03-03 10:17:07
*
@LastEditTime: 2023-03-03 10:2
5:36
*
@LastEditTime: 2023-03-03 10:2
6:40
*
@FilePath:
\l
pr
\R
EADME.md
-->
# License-Plate-Recoginition(LPR)
...
...
@@ -12,6 +12,7 @@ LPR是一个基于深度学习技术的车牌识别模型,主要识别目标
模型采用LPRNet,模型结构主要包含三部分:一个轻量级CNN主干网络、基于预定位置的字符分类头部、基于贪婪算法的序列解码。此外模型使用CTC Loss和RMSprop优化器。参考论文
[
LPRNet: License Plate Recognition via Deep Neural Networks
](
https://arxiv.org/pdf/1806.10447v1.pdf
)
## 数据集
推荐使用一个车牌数据集
[
CCPD
](
https://github.com/detectRecog/CCPD
"CCPD官网GitHub"
)
,该数据集由中科大收集,可用于车牌的检测与识别。
我们提供了一个脚本cut_ccpd.py用于剪裁出CCPD数据集中的车牌位置,以便用于LPR模型的训练,在cut_ccpd.py中修改img_path和save_path即可,分别是CCPD数据集中ccpd_base文件夹的路径和剪裁出的图像保存路径。LPR用于训练的数据文件名就是图像的标签。
**数据集使用固定的大小94x24。**
## 训练及推理
### 环境配置
...
...
@@ -22,9 +23,12 @@ LPR是一个基于深度学习技术的车牌识别模型,主要识别目标
### 训练与Fine-tunning
LPR模型的训练程序是train.py,初次训练模型使用以下命令:
`python train.py \
`
python train.py
\
--train_img_dirs 训练集文件夹路径
\
--test_img_dirs 验证集文件夹路径`
--test_img_dirs 验证集文件夹路径
`
Fine-tunning使用以下命令:
python train.py \
...
...
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