yolov5和resnet比稍微麻煩了一點(diǎn),主要就是多了nms部分,還有坐標(biāo)點(diǎn)映射回原圖的yolov5_scale_coords函數(shù)。流程大致分為五部分:1)圖像等比例放縮,2)圖像預(yù)處理,3)onnx推理,4)nms后處理,5)坐標(biāo)點(diǎn)映射回原圖
創(chuàng)新互聯(lián)是專業(yè)的澤州網(wǎng)站建設(shè)公司,澤州接單;提供網(wǎng)站設(shè)計(jì)制作、做網(wǎng)站,網(wǎng)頁設(shè)計(jì),網(wǎng)站設(shè)計(jì),建網(wǎng)站,PHP網(wǎng)站建設(shè)等專業(yè)做網(wǎng)站服務(wù);采用PHP框架,可快速的進(jìn)行澤州網(wǎng)站開發(fā)網(wǎng)頁制作和功能擴(kuò)展;專業(yè)做搜索引擎喜愛的網(wǎng)站,專業(yè)的做網(wǎng)站團(tuán)隊(duì),希望更多企業(yè)前來合作!等比例放縮還是和resnet一樣的 letterbox 函數(shù),就不重復(fù)了。
圖像預(yù)處理還是和resnet一樣的,就不重復(fù)了,就cv::dnn::blobFromImage一句話。
onnx推理和之前差不多,改下輸入輸出的維度、名稱,然后resnet的一維輸出變成yolo的二維輸出,這個(gè)在 c++ 使用onnx推理 中寫過了,也不重復(fù)了。
nms后處理這個(gè)我懶得自己寫,還好opencv里面有cv::dnn::NMSBoxes函數(shù),直接用就好。
正常來說onnx推理后返回的預(yù)測值是252000 x (n+5) ,其中252000是框的總數(shù)一般是不變的, n是等于你設(shè)置的類別數(shù),向量含義是(center_x, center_y, width, height, conf_框, conf_類別1,conf_類別2,…,,conf_類別n)。
這個(gè)函數(shù)主要就是把preds轉(zhuǎn)成opencv_nms函數(shù)需要的輸入 cv::Rect(left, top, w
你是否還在尋找穩(wěn)定的海外服務(wù)器提供商?創(chuàng)新互聯(lián)www.cdcxhl.cn海外機(jī)房具備T級流量清洗系統(tǒng)配攻擊溯源,準(zhǔn)確流量調(diào)度確保服務(wù)器高可用性,企業(yè)級服務(wù)器適合批量采購,新人活動首月15元起,快前往官網(wǎng)查看詳情吧