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

[Frontend][TensorFlow] Support NonMaxSuppressionV5 #6933

Merged
merged 1 commit into from
Nov 17, 2020

Conversation

trevor-m
Copy link
Contributor

  • Support NonMaxSuppressionV5 aka tf.image.non_max_suppression_with_scores which returns scores in addition to indices and size (tensorflow/tensorflow@eb10dcf)
  • "soft_nms_sigma" option is not supported yet, but it doesn't appear to be common.

NMSv5 is used by TF Object Detection API.

Copy link
Contributor

@kevinthesun kevinthesun left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@zhiics zhiics merged commit 0f5b5fe into apache:main Nov 17, 2020
@zhiics
Copy link
Member

zhiics commented Nov 17, 2020

Thanks @trevor-m @kevinthesun

TusharKanekiDey pushed a commit to TusharKanekiDey/tvm that referenced this pull request Nov 18, 2020
codeislife99 pushed a commit to codeislife99/incubator-tvm that referenced this pull request Nov 20, 2020
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 2, 2020
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 4, 2020
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Dec 4, 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

Successfully merging this pull request may close these issues.

3 participants