search

we0091234 / Chinese_license_plate_detection_recognition

yolov5 车牌检测 车牌识别 中文车牌识别 检测 支持12种中文车牌 支持双层车牌
GNU General Public License v3.0
1.34k stars 224 forks source link

支持多类别检测吗？ #8

Open piscesdream opened 2 years ago

piscesdream commented 2 years ago

比如即检测车、也检测车牌、同时仅回归车牌的4个关键点。

we0091234 commented 2 years ago

支持啊，你加个类别就行了，没有关键点的就用-1

piscesdream commented 2 years ago

多谢大佬指教，顺便请问下您训练时采用的预训练模型是yolov5-face里面的吗？

we0091234 commented 2 years ago

训练

是的

piscesdream commented 2 years ago

我在使用yolov5s-face.pt做预训练模型时（预训练模型是分类1、5个关键点也就是（1+5+10）x3 =48, 我的网络是分类5，关键点4，（5+5+8）x3=54），出现以下错误： image 正常情况下网络不是通过网络名和shape过滤掉不匹配的网络层，然后跳过不加载该层权重，为什莫还会出现这样问题？

we0091234 commented 2 years ago

我在使用yolov5s-face.pt做预训练模型时（预训练模型是分类1、5个关键点也就是（1+5+10）x3 =48, 我的网络是分类5，关键点4，（5+5+8）x3=54），出现以下错误： image 正常情况下网络不是通过网络名和shape过滤掉不匹配的网络层，然后跳过不加载该层权重，为什莫还会出现这样问题？

换yolov5 官方的模型看看别用yolov5s-face.pt 用yolov5s.pt

piscesdream commented 2 years ago

yolov5s.pt是没问题的，但可加载权重层太少，就7层

we0091234 commented 2 years ago

yolov5s.pt是没问题的，但可加载权重层太少，就7层

或者用这句话 model.load_state_dict(state_dict, strict=False)
load_state_dict 的时候 strict设置为False