zzzxxxttt / pytorch_simple_CenterNet_45

A simple pytorch implementation of CenterNet (Objects as Points)
308 stars 63 forks source link

为什么test阶段需要nms? #31

Closed NeuZhangQiang closed 3 years ago

NeuZhangQiang commented 3 years ago

在看代码的时候,发现test阶段需要用到soft_nms,但是CenterNet不是号称不需要nms做后处理的嘛?我记得原文中,说是找极值,然后用前100个极值来做框。其实,我就是对这一部分有点云里雾里的。不太能理解,如果写死了取100个极值,岂不是意味着一张图就有100个框?但是,实际上一张图可能是只有3个框,4个框之类的。 作者能不能把这一步写出来呢?感谢!!!

zzzxxxttt commented 3 years ago
NeuZhangQiang commented 3 years ago

非常感谢!