Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Faster GPU NMS operator #16542

Merged
merged 19 commits into from
Nov 5, 2019
Merged

Faster GPU NMS operator #16542

Show file tree
Hide file tree
Changes from 15 commits
Commits
Show all changes
19 commits
Select commit Hold shift + click to select a range
fe2a7d9
Adding second NMS op
ptrendx Oct 8, 2019
5d0a430
NMS kernel
ptrendx Oct 14, 2019
0829222
Removing second sort
ptrendx Oct 17, 2019
f3445f8
Optimization
ptrendx Oct 17, 2019
e653f58
Adding out-of-place ability to SortByKey
ptrendx Oct 17, 2019
49ded5a
Optimization pt2
ptrendx Oct 17, 2019
3680e1d
Optimizations pt3
ptrendx Oct 18, 2019
1978b07
Do not recompute other boxes area every time
ptrendx Oct 18, 2019
dfaa4c5
Sort only topk results during second sorting
ptrendx Oct 18, 2019
9069192
Cleaning
ptrendx Oct 18, 2019
ad60b56
Fixes from rebase
ptrendx Oct 19, 2019
eb89b07
Fix lint and more fixes from rebase
ptrendx Oct 19, 2019
2567266
Fix typo
ptrendx Oct 19, 2019
6ac371c
Early exit in Triangle kernel
ptrendx Oct 19, 2019
b0ce511
Fixes
ptrendx Oct 19, 2019
753e7c0
Fix sort
ptrendx Oct 24, 2019
8cb9a5b
Fix from rebase
ptrendx Oct 24, 2019
dc1a599
Fix for the mixed naming convention
ptrendx Oct 24, 2019
3dfca4d
Fix the index_t with int comparisoon
ptrendx Oct 25, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions src/operator/contrib/bounding_box.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ DMLC_REGISTER_PARAMETER(BoxOverlapParam);
DMLC_REGISTER_PARAMETER(BipartiteMatchingParam);
DMLC_REGISTER_PARAMETER(BoxDecodeParam);


NNVM_REGISTER_OP(_contrib_box_nms)
.add_alias("_contrib_box_non_maximum_suppression")
.describe(R"code(Apply non-maximum suppression to input.
Expand Down
Loading