用yolov7訓練二分類會出現一堆bbox?怎麼過濾?
只想要有紙箱的對應boundingbox.
問題如圖
用yolov7訓練二分類會出現一堆bbox?怎麼過濾?
只想要有紙箱的對應boundingbox.
問題如圖
因为以不同中心点画出的框,最后概率都差不多,所以导致大框小框全都保留了,主要还是盒子颜色差不多,我之前跟你说过opencv会更简单,如果楼主坚持用yolo,楼主我感觉你这个问题应该魔改yolov1,就是不需要预测xywh这个盒子,而是只需要预测yolov1里每个格子是盒子还是不是盒子的,倒数第一层是一个772的tensor就行,当然楼主觉得49个格子太少,可以换成15152,最终楼主再把所有格子为1的用红透明层表示,为0的用绿透明层