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

efficientNMSPlugin: support class-independent nms with new parameter class_agnostic #2593

Closed
wants to merge 1 commit into from

Conversation

yili123123
Copy link

Some object detection networks/architectures like YOLO series need to use class independent nms operations. Now we can do class independent nms when the class_agnostic is enabled, otherwise, different class would do nms respectively.

@yili123123
Copy link
Author

yili123123 commented Jan 29, 2023

@rajeevsrao @pranavm-nvidia Hi, can you take your time to review this.

@NVIDIA NVIDIA deleted a comment from yili123123 Feb 7, 2023
@yili123123 yili123123 force-pushed the main branch 2 times, most recently from 46f58b7 to d9619ff Compare February 7, 2023 04:03
Signed-off-by: yili123123 <[email protected]>

Co-authored-by: Samurdhi Karunaratne <[email protected]>
@yili123123
Copy link
Author

@samurdhikaru ,sorry for closing the previous pull request by mistake. Can you take your time to review the new PR?

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.

2 participants