重點(diǎn)(本質(zhì)結(jié)構(gòu)重參數(shù)化)
我們可以看到基于YoloV7訓(xùn)練的cfg有兩種yaml文件,一個(gè)是training文件夾,一個(gè)是deploy文件夾,這兩種文件夾有啥不一樣呢???
大家可以看下下面別人的issuse,,記住這個(gè)很關(guān)鍵,就是你選擇哪個(gè)yaml訓(xùn)練對(duì)你后面導(dǎo)出的onnx是很關(guān)鍵的,后面我們會(huì)說到。
1、training中的yaml文件最后是采用IDetect,而deploy中的yaml文件采用的是Detect
2、IDetect是在最后一個(gè)C5結(jié)構(gòu)輸出增加一個(gè)add操作,之后在進(jìn)行conv,然后在mul操作,而Detect則是和v5一樣的操作,沒有add mul。
3、一句話:deploy模型中自帶后處理和nms(非極大值抑制),不需要自己在編寫,強(qiáng)烈建議使用deploy進(jìn)行訓(xùn)練,以下教程根據(jù)deploy配置文件進(jìn)行編寫
4、否則會(huì)報(bào)錯(cuò):too many values to unpack (expected 7)
,則是因?yàn)槟P洼敵霾皇牵?code>float32[Concatoutput_dim_0,7],而是 float32[1,25200,85]
,需要自己編寫后處理,也可以去yolov5代碼中復(fù)制:文章來源:http://www.zghlxwxcb.cn/news/detail-650451.html
5、類似paddlepaddle框架中也有配置如下圖,需要yolov5,onnx預(yù)處理和后處理代碼的私信我
文章來源地址http://www.zghlxwxcb.cn/news/detail-650451.html
到了這里,關(guān)于python使用onnx模型進(jìn)行推理的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!