最全车牌识别算法,支持11种中文车牌类型
1.单行蓝牌 2.单行黄牌 3.新能源车牌 4.白色警用车牌 5 教练车牌 6 武警车牌 7 双层黄牌 8 双层武警 9 使馆车牌 10 港澳牌车 11 双层农用车牌
12 民航车牌
环境配置:
1.python >=3.6 pytorch >=1.7
运行:
python detect_plate.py
测试文件夹imgs,结果保存再 result 文件夹中
parser.add_argument('--image_path', type=str, default='imgs', help='source') # file/folder, 0 for webcam
车牌检测训练
参考yolov5-face:
-
下载数据集:链接:https://pan.baidu.com/s/1xCYunxRoT3Xv8TeE2t1kPQ 提取码:trbl 数据从CCPD数据集中选取并转换的 数据集格式为yolo格式:
label x y w h pt1x pt1y pt2x pt2y pt3x pt3y pt4x pt4y
关键点依次是(左上,右上,右下,左下) 坐标都是经过归一化,x,y是中心点除以图片宽高,w,h是框的宽高除以图片宽高,ptx,pty是关键点坐标除以宽高
-
修改 data/widerface.yaml train和val路径
-
python3 train.py --data data/widerface.yaml --cfg models/yolov5n-0.5.yaml --weights weights/best.pt --epoch 250
结果存在run文件夹中
车牌识别参考:
crnn:
有问题可以提issues 或者加qq群:871797331 询问
支持如下:
12.民航车牌