Skip to content

NMS bugfix - #58

Open
pfjaeger wants to merge 1 commit into
multimodallearning:masterfrom
pfjaeger:master
Open

NMS bugfix#58
pfjaeger wants to merge 1 commit into
multimodallearning:masterfrom
pfjaeger:master

Conversation

@pfjaeger

Copy link
Copy Markdown

NMS is not working correctly, since the detections fed in are not sorted:

in nms/path_nms.py line 49:

nms.gpu_nms(keep, num_out, dets_temp, thresh)

the unordered 'dets_temp' are fed in instead of the ordered 'dets'.
This is solvable by either changing line 43 to :

dets_temp = dets_temp[order].contiguous()

or just getting rid of the dets_temp and feeding in dets, since NMS should be rotation invariant.

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

Successfully merging this pull request may close these issues.

1 participant