yolov7使用c++推理onnx模型cpu,gpu均可(附代碼)-創(chuàng)新互聯(lián)

成都創(chuàng)新互聯(lián)-企業(yè)級(jí)云服務(wù)器提供商,為用戶提供云服務(wù)器、CDN、云安全服務(wù)、重慶服務(wù)器托管、服務(wù)器租用、高防服務(wù)器租用等全方位云服務(wù)與各行業(yè)解決方案,幫助企業(yè)及個(gè)人極速備案,輕松上云,安全無(wú)憂。

? 先放上原圖以及推理效果圖,這份數(shù)據(jù)集是我自己標(biāo)注的,然后使用yolov7的官方代碼進(jìn)行訓(xùn)練得到了一個(gè)best.pt。最后還是用yolov7官方的export.py?導(dǎo)出得到一個(gè)onnx文件。文章下面我會(huì)附上數(shù)據(jù)集,需要自提。

要點(diǎn):

yolov7環(huán)境搭建,模型的訓(xùn)練以及導(dǎo)出模型。

代碼的編寫已經(jīng)文件的使用。

一.模型導(dǎo)出

? 我相信能能看到這里的朋友對(duì)于yolov7的使用應(yīng)該是沒(méi)有的問(wèn)題的,那么只是需要使用官方代碼導(dǎo)出一個(gè)onnx文件即可完成第一步的操作。

之后就會(huì)得到想要的onnx文件這里應(yīng)該問(wèn)題不大。

二.vs平臺(tái)環(huán)境搭建?

? 當(dāng)然cpu處理是非常簡(jiǎn)單有效的,只需要配置一個(gè)release版本的opencv即可。我這里重點(diǎn)講述的是gpu版本的處理。

? 1.首先要確定你的顯卡型號(hào),我的電腦是3060對(duì)應(yīng)的cuda是11.1,cudnn8.1.1。當(dāng)然我相信會(huì)使用yolov7的朋友對(duì)于這些是很熟悉的。我想強(qiáng)調(diào)的一點(diǎn)是由于vs平臺(tái)的特殊性,vs平臺(tái)的不同版本就對(duì)很多版本的cuda是不兼容的。例如vs2022對(duì)于老版本cuda就無(wú)法調(diào)用。同樣老版本的vs對(duì)于新cuda也無(wú)法調(diào)用。我們?cè)谙螺d的時(shí)候一定要考慮一些版本適配性。

你是否還在尋找穩(wěn)定的海外服務(wù)器提供商?創(chuàng)新互聯(lián)www.cdcxhl.cn海外機(jī)房具備T級(jí)流量清洗系統(tǒng)配攻擊溯源,準(zhǔn)確流量調(diào)度確保服務(wù)器高可用性,企業(yè)級(jí)服務(wù)器適合批量采購(gòu),新人活動(dòng)首月15元起,快前往官網(wǎng)查看詳情吧

當(dāng)前文章:yolov7使用c++推理onnx模型cpu,gpu均可(附代碼)-創(chuàng)新互聯(lián)
網(wǎng)頁(yè)地址:http://www.muchs.cn/article10/epsdo.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供軟件開(kāi)發(fā)動(dòng)態(tài)網(wǎng)站、品牌網(wǎng)站設(shè)計(jì)、商城網(wǎng)站電子商務(wù)、網(wǎng)站制作

廣告

聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請(qǐng)盡快告知,我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如需處理請(qǐng)聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時(shí)需注明來(lái)源: 創(chuàng)新互聯(lián)

成都seo排名網(wǎng)站優(yōu)化