Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

removing point_nms leads to near 0 AP #45

Closed
twangnh opened this issue May 12, 2020 · 2 comments
Closed

removing point_nms leads to near 0 AP #45

twangnh opened this issue May 12, 2020 · 2 comments

Comments

@twangnh
Copy link

twangnh commented May 12, 2020

Hi! XinLong, i noted in #33, the point nms is only used to reduce memory usage, but when I try to remove it, the performance drop to near 0, could you please give some advice?

@WXinlong
Copy link
Owner

@twangnh Hi, I guess you have just removed L163. It's not correct. You can replace that line with cate_pred = cate_pred.sigmoid().permute(0, 2, 3, 1) for removing point nms. You will get 35.9 AP for SOLO_R50_3x model.

@twangnh
Copy link
Author

twangnh commented May 13, 2020

thanks, it works.

@twangnh twangnh closed this as completed May 13, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants