Open piscesdream opened 1 year ago
支持啊,你加个类别就行了,没有关键点的就用-1
多谢大佬指教,顺便请问下您训练时采用的预训练模型是yolov5-face里面的吗?
训练
是的
我在使用yolov5s-face.pt做预训练模型时(预训练模型是分类1、5个关键点也就是(1+5+10)x3 =48, 我的网络是分类5,关键点4,(5+5+8)x3=54),出现以下错误:
正常情况下网络不是通过网络名和shape过滤掉不匹配的网络层,然后跳过不加载该层权重,为什莫还会出现这样问题?
我在使用yolov5s-face.pt做预训练模型时(预训练模型是分类1、5个关键点也就是(1+5+10)x3 =48, 我的网络是分类5,关键点4,(5+5+8)x3=54),出现以下错误:
正常情况下网络不是通过网络名和shape过滤掉不匹配的网络层,然后跳过不加载该层权重,为什莫还会出现这样问题?
换yolov5 官方的模型看看别用yolov5s-face.pt 用yolov5s.pt
yolov5s.pt是没问题的,但可加载权重层太少,就7层
yolov5s.pt是没问题的,但可加载权重层太少,就7层
或者用这句话 model.load_state_dict(state_dict, strict=False)
load_state_dict 的时候 strict设置为False
比如即检测车、也检测车牌、同时仅回归车牌的4个关键点。