search

Stinky-Tofu / Stronger-yolo

🔥Improve yolo with latest paper
MIT License
3 stars 0 forks source link

data.py __create_label问题 #19

Open qianlinjun opened 5 years ago

qianlinjun commented 5 years ago

你好,在看data.py 中__create_label函数有点疑问,为什么先用大于0.3的anchor, 当exist_positive为false时再用最大IOU对应的best anchor,而不是直接用最大IOU对应的best anchor?这样是担心多个ground_true box对应同一个位置的同一个anchor 从而出现有gt box被挤掉的情况吗?但是这样也会出现不同anchor和同一个gt box的iou大于0.3,这没有影响吗?

Stinky-Tofu commented 5 years ago

你好,在看data.py 中__create_label函数有点疑问,为什么先用大于0.3的anchor, 当exist_positive为false时再用最大IOU对应的best anchor,而不是直接用最大IOU对应的best anchor?这样是担心多个ground_true box对应同一个位置的同一个anchor 从而出现有gt box被挤掉的情况吗?但是这样也会出现不同anchor和同一个gt box的iou大于0.3,这没有影响吗?

这是我做的一些改进，有一点点提升0.6吧，我觉得这样可以提高召回率，仅供参考。

qianlinjun commented 5 years ago

好的,谢谢!