xialuxi / yolov5-car-plate

基于yolov5的车牌检测,包含车牌角点检测
GNU General Public License v3.0
152 stars 40 forks source link

不定角点检测 #23

Open zhangsiqiGit opened 1 year ago

zhangsiqiGit commented 1 year ago

hi,感谢对车牌&角点检测的分享,目前有个问题想咨询一下。

目前custume数据集,车牌被遮挡部分不标注,因此每个车辆的车牌角点个数为(0~4),当前模型跑了50个epochs左右发现几乎每张图都会预测4个角点(被遮挡部分应该不预测角点)。想问问如何解决?如果要修改网络结构部分,如何更改呢?还是对每个keypoint做一个sigmoid呢?

xialuxi commented 1 year ago

额外增加一个预测值,一个点有3个预测值:x, y, z, 其中x, y代表坐标, z代表可不可见。