Skip to content

[Frontend][TensorFlow] Support NonMaxSuppressionV5#6933

Merged
zhiics merged 1 commit intoapache:mainfrom
trevor-m:nmsv5
Nov 17, 2020
Merged

[Frontend][TensorFlow] Support NonMaxSuppressionV5#6933
zhiics merged 1 commit intoapache:mainfrom
trevor-m:nmsv5

Conversation

@trevor-m
Copy link
Copy Markdown
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
Copy Markdown
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
Copy Markdown
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