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][TFLite] Add support for NonMaxSuppressionV5 op #12003

Merged
merged 2 commits into from
Jul 5, 2022

Conversation

blackkker
Copy link
Contributor

When I import EfficientDet_lite model, the NonMaxSuppressionV5 op is not supported yet. I implemented it with reference to nms of tensorflow and TFLite_Detection_PostProcess of TFLite.
Also, I constructed the test case with reference to the model.

test_case_nms_v5
test_case_nms_v5

model_nms_v5
model_nms_v5

@blackkker blackkker changed the title [FRONTEND][TFLITE] Add support for TFLite_Detection_PostProcess [Frontend][TFLite] Add support for TFLite_Detection_PostProcess Jul 4, 2022
@blackkker blackkker changed the title [Frontend][TFLite] Add support for TFLite_Detection_PostProcess [Frontend][TFLite] Add support for NonMaxSuppressionV5 op Jul 4, 2022
@blackkker
Copy link
Contributor Author

blackkker commented Jul 4, 2022

@AndrewZhaoLuo @masahi

@masahi masahi merged commit 83b310d into apache:main Jul 5, 2022
blackkker added a commit to blackkker/tvm that referenced this pull request Jul 7, 2022
* add nms_v5 op for TFLite

* add a test for the TFLite nms_v5 op
masahi pushed a commit to masahi/tvm that referenced this pull request Jul 15, 2022
* add nms_v5 op for TFLite

* add a test for the TFLite nms_v5 op
mikeseven pushed a commit to mikeseven/tvm that referenced this pull request Sep 27, 2023
* add nms_v5 op for TFLite

* add a test for the TFLite nms_v5 op
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