search

bubbliiiing / yolo3-pytorch

这是一个yolo3-pytorch的源码，可以用于训练自己的模型。
MIT License
1.99k stars 584 forks source link

关于yolo_training文件中的get_target和get_ignore--yolov3 #175

Open TianX1nYu opened 1 year ago

TianX1nYu commented 1 year ago

作者您好，有一个疑惑点希望您能帮忙解答。 按照我的理解，get_target返回的noobj_mask是根据未调整时候的先验框与真实框的iou将最大的先验框的noobj_mask置为0，即存在物体。而get_ignore则又将get_target返回的noobj_mask根据调整后的先验框又做了一遍这个工作。这样一来的话，get_target中的这个操作是不是就没有什么意义了？

bubbliiiing commented 1 year ago

get_ignore还计算了别的框呀

TianX1nYu commented 1 year ago

感谢您的回答，您说的对，get_ignore还计算了别的框， 所以我问的是get_target里面的noobj_mask[b, k, j, i] = 0是不是多余了，在get_target里将未调整的先验框与真实框的iou将最大的先验框的noobj_mask置为0。假设被选中的这个序号为2，那么这个框是不是在get_ignore方法里同样也会被再次选中一次？ 感谢您的解惑

bubbliiiing commented 1 year ago

会再次选中，但是第二次不一定会选中