[Relax][ONNX Frontend] Does the Relax frontend support NonMaxSuppression?

Questions
1

TVM main branch has removed the Relay frontend. And it is claimed that the Relax Onnx frontend supports all operators that Relay could support: Phasing out Legacy Components - #20 by Hzfengsy

From the code: tvm/python/tvm/relax/frontend/onnx/onnx_frontend.py at 707daef83eed60222b2e4cc38c7292cdf8b217ee · apa

It seems that the parsing of NMS has been commented out, but NMS is supported in the Relay frontend.

Has the Relax ONNX frontend really supported all the operators that Relay could support?

1 Like
2

+1. Is there an issue tracking unsupported ops in Relax frontend for ONNX?

3

I guess same is the case for layers: InstanceNorm2d and Padding(Zero/Reflection).

4

yes, the newest code delete all nms implementation include topi, rencently we need the nms in relax model support, so porting one kind of nms to relax, i think we can upstream it. @ysh329

5

Okay. I will upload this AllClassNMS Op’s patch this week.

2 Likes