-
Notifications
You must be signed in to change notification settings - Fork 8
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
does the onnxruntime include nms? #1
Comments
not in onnxruntime, but include in lite.ai.toolkit, see the source code here:
|
@DefTruth is that possible export the model directly with nms baked in? |
一般情况不太可能,也不是很有必要导出NMS。NMS里面的一些for循环,哪怕能导出onnx或其他模型文件,算子也会比较琐碎和冗余。用C++写一个不是很难。 |
@DefTruth if target on cpu, it can wrap NonMaxSupression as onnx op. Many forward framework support this op, such as dnn and ort. Not sure about mnn or ncnn |
haha, many thanks for your advice~ |
🤔没有很专业地在研究推理引擎,只是平时在做算法研究的时候会业余玩一玩,应用为主哈哈。 |
does the onnxruntime include nms?
The text was updated successfully, but these errors were encountered: