Comments (6)
@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
from yolov5face-toolkit.
not in onnxruntime, but include in lite.ai.toolkit, see the source code here:
- NMS with landmarks in C++: https://github.com/DefTruth/lite.ai.toolkit/blob/main/lite/ort/cv/yolo5face.cpp#L162
from yolov5face-toolkit.
@DefTruth is that possible export the model directly with nms baked in?
from yolov5face-toolkit.
一般情况不太可能,也不是很有必要导出NMS。NMS里面的一些for循环,哪怕能导出onnx或其他模型文件,算子也会比较琐碎和冗余。用C++写一个不是很难。
from yolov5face-toolkit.
@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~
from yolov5face-toolkit.
🤔没有很专业地在研究推理引擎,只是平时在做算法研究的时候会业余玩一玩,应用为主哈哈。
from yolov5face-toolkit.
Related Issues (7)
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from yolov5face-toolkit.