search

xg-chu / CrowdDet

[CVPR 2020] Detection in Crowded Scenes: One Proposal, Multiple Predictions
MIT License
422 stars 85 forks source link

setnms有gpu版本吗? #71

Open Eagle104fred opened 2 years ago

Eagle104fred commented 2 years ago

最近在改setnms的gpu版本, 但是效果不太理想, 我理解原版setnms的做法是当拿到一个basement, 对剩下的框做iou并删除后, 再把number与basement相同的框复活, 不知道我理解的是否正确, 还请大佬能指点一二。

yjhtc007 commented 2 years ago

我感觉是这样的

yexiguafuqihao commented 2 years ago

Yes, you are correct.

taofuyu commented 2 years ago

Yes, you are correct.

一个anchor预测的两个框的分数是不一样的，为啥可以通过复活相同分数框的做法来找回同组的预测呢？

Eagle104fred commented 2 years ago

Yes, you are correct.

一个anchor预测的两个框的分数是不一样的，为啥可以通过复活相同分数框的做法来找回同组的预测呢？

并不是复活相同分数的框, 而是复活同一层中被抑制掉的那个框, 同层指的是双检测头的另一个检测头出来的框