NonMaxSuppression: limit output shape size by `max_output_boxes_per_class`
Open
Nobody has claimed this yet.
- Dominant language
- C++
- Stars
- 333
- Forks
- 150
- Avg merge
- 4d 19h
- Merged PRs (30d)
- 54
Description
https://onnx.ai/onnx/operators/onnx__NonMaxSuppression.html
If the max_output_boxes_per_class input is constant it can be used to limit the output size of NMS.
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the NonMaxSuppression operator implementation and its tests, then read the linked ONNX operator specification for the expected output shape. Trace how a constant max_output_boxes_per_class input is represented and identify the existing shape-inference path. Done means the output shape is bounded by that constant and relevant tests cover the behavior.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Quiet
- Clarity
- Mostly clear
- Newbie friendliness
- 55/100