在進(jìn)行目標(biāo)檢測部署時,通常需要自行編碼實(shí)現(xiàn)對模型預(yù)測結(jié)果的解碼及與預(yù)測結(jié)果的nms操作。所幸現(xiàn)在的各種部署框架對算子的支持更為靈活,可以在模型內(nèi)實(shí)現(xiàn)預(yù)測結(jié)果的解碼,但仍然需要自行編碼實(shí)現(xiàn)對預(yù)測結(jié)果的nms操作。其實(shí)在onnx opset===11版本以后,其已支持將nms操作嵌入到模型中,具體可以參考https://hpg123.blog.csdn.net/article/details/131585808,修改代碼實(shí)現(xiàn)yolov8模型onnx下的無nms導(dǎo)出。本博文主要介紹基于onnxruntime在c#下部署含nms的yolov8模型。
1、運(yùn)行環(huán)境安裝
安裝運(yùn)行依賴opencv與onnxruntime
在nuget中安裝onnxruntime與opencv,一一共安裝以下4個庫。
安裝步驟參考
https://blog.csdn.net/qq_36694133/article/details/128209770
opencv安裝步驟參考
https://blog.csdn.net/sinat_37281674/article/details/119987327文章來源:http://www.zghlxwxcb.cn/news/detail-658809.html
2、部署模型
2.1 完整代碼
代碼中模型下載地址為:https://gitcode.net/a486259/model/-/blob/master/yolov8s_nms.onnx
也可以換成自己導(dǎo)出的模型。文章來源地址http://www.zghlxwxcb.cn/news/detail-658809.html
using
到了這里,關(guān)于pytorch 42 C#使用onnxruntime部署內(nèi)置nms的yolov8模型的文章就介紹完了。如果您還想了解更多內(nèi)容,請?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!