Skip to content

GitLab

Commit 66498f8a authored by liuhy's avatar liuhy
Browse files

修改readme

parent 7cb703c4
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
README.md
View file @ 66498f8a
...@@ -6,6 +6,11 @@ LPR是一个基于深度学习技术的车牌识别模型，主要识别目标 ...@@ -6,6 +6,11 @@ LPR是一个基于深度学习技术的车牌识别模型，主要识别目标
## 数据集 ## 数据集
推荐使用一个车牌数据集[CCPD](https://github.com/detectRecog/CCPD "CCPD官网GitHub")，也可参考[CCPD](https://blog.csdn.net/LuohenYJ/article/details/117752120 "CCPD中文版介绍")，该数据集由中科大收集，可用于车牌的检测与识别。我们提供了一个脚本cut_ccpd.py用于剪裁出CCPD数据集中的车牌位置，以便用于LPR模型的训练，在cut_ccpd.py中修改img_path和save_path即可，分别是CCPD数据集中ccpd_base文件夹的路径和剪裁出的图像保存路径。LPR用于训练的数据文件名就是图像的标签。**数据集使用固定的大小94x24。** 推荐使用一个车牌数据集[CCPD](https://github.com/detectRecog/CCPD "CCPD官网GitHub")，也可参考[CCPD](https://blog.csdn.net/LuohenYJ/article/details/117752120 "CCPD中文版介绍")，该数据集由中科大收集，可用于车牌的检测与识别。我们提供了一个脚本cut_ccpd.py用于剪裁出CCPD数据集中的车牌位置，以便用于LPR模型的训练，在cut_ccpd.py中修改img_path和save_path即可，分别是CCPD数据集中ccpd_base文件夹的路径和剪裁出的图像保存路径。LPR用于训练的数据文件名就是图像的标签。**数据集使用固定的大小94x24。**
## 训练及推理 ## 训练及推理
### 环境配置
在[光源](https://www.sourcefind.cn/#/service-details)可拉取docker镜像，在[光合开发者社区](https://cancon.hpccube.com:65024/4/main/)可下载Migraphx和ONNXruntime安装包。LPR推荐的镜像、Migraphx以及ONNXruntime版本如下：
* 镜像：docker pull image.sourcefind.cn:5000/dcu/admin/base/vscode-pytorch:1.10.0-centos7.6-dtk-22.10-py37-patch4
* Migraphx-2.5.0-dtk-22.10
* ORT-1.14.0-dtk-22.10
### 训练与Fine-tunning ### 训练与Fine-tunning
LPR模型的训练程序是train.py，初次训练模型使用以下命令： LPR模型的训练程序是train.py，初次训练模型使用以下命令：
......
requirement.txt 0 → 100644
View file @ 66498f8a
imutils==0.5.4
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