遇到的问题
标注胸部和小腹的框,如果标得比较内部,没有覆盖整个胸部或者小腹,会出现检测的时候,
一个人本来是要出现一个胸部的框,但是出现了两个,一个左胸和一个右胸的,小腹的也是出现了左右两个框,这个情况,有50%情况会出现一个人一个部位出现了两个框,有约剩下50%的情况出现一个部位一个框,一个部位一个框这是我们需要的。
原本为什么标得比较内部没有覆盖整个部位,是因为要对部位的温度处理,不希望把背景框太多进来。
解决办法
1,标注的时候,覆盖整个部位,不要标得太过内部。这样nms之后会留下一个框。
2,第一个解决办法可能不可行。不用重新标注。因为用的权重是之前一个版本的权重,这个版本的也是要框出胸部和小腹,但是人的姿势不一样,
有可能是因为用了这个版本的权重作为这次模型的初始化权重,然后用新的姿势训练,一个部位就出现了两个框。
解决办法是,不用上一个版本的预训练权重,从头开始训练。
3,做了实验,解决办法1,2 都试过了不行,看代码排查出了问题,代码dataloader是需要输入的是归一化后的class_id, x_c, y_c,
w, h,而我给的输入是归一化后的class_id, x1, y1, w, h,改正输入从新训练后最终问题解决。x_c, y_c 是bbox的中心坐标,x1,y1是bbox的左上角坐标,w,h是bbox的宽高,都是归一化后的结果,x是水平方向,h是竖直方向。