極市平臺(tái):打榜經(jīng)驗(yàn)分享(卸油作業(yè)安全設(shè)施識(shí)別打榜)
從極市平臺(tái)看到,2023.2.26起,視覺AI算法實(shí)訓(xùn)加速營(yíng)本著學(xué)習(xí)和實(shí)踐的態(tài)度,對(duì)卸油作業(yè)安全設(shè)施識(shí)別(https://www.cvmart.net/topList/10585),打榜;最終過關(guān):提交算法通過,現(xiàn)把過程分享,有問題一塊解決哈:

基本過程:
注冊(cè),點(diǎn)擊打榜,相中那個(gè)參加打榜,工作臺(tái)進(jìn)行開發(fā),會(huì)有vscode的網(wǎng)頁(yè)版本,進(jìn)行代碼編寫。
Ctrl + `? :? 快捷鍵,打開 命令行
模型開發(fā):
? ?感覺還是適合yolov5,快速又準(zhǔn)確,后面再試試yolov8,看看:
?用的模型庫(kù)是實(shí)訓(xùn)老師的:目標(biāo)檢測(cè):https://github.com/daimaohui/cvmart_yolov5 ,按著步驟一步步來,分?jǐn)?shù)應(yīng)該差不了;
卸油作業(yè)安全這個(gè),把測(cè)試的ji.py,改成卸油作用的分類后,上分挺快的:
我的第一次出分是? 性能分:77.6298? f-score:0.9231
訓(xùn)練的命令行代碼是:?????
接續(xù)訓(xùn)練:# 接著 上一次 繼續(xù)訓(xùn)練
寫好,pt文件夾地址就成:
算法封測(cè):-其實(shí)就是把訓(xùn)練結(jié)果,轉(zhuǎn)到tensorrt 的C++版本下
同樣,也要看實(shí)訓(xùn)老師給的教學(xué):
https://blog.csdn.net/qq_35200479/article/details/127761384
也有視頻:https://www.bilibili.com/video/BV1bG4y1w7Pj , 看完保證你會(huì),哈哈
現(xiàn)在主要寫一下,我覺得主要 注意點(diǎn)吧:
使用的庫(kù)是:
https://gitee.com/cvmart/ev_sdk_demo4.0_pedestrian_intrusion_yolov5
復(fù)制 庫(kù),到ev_sdk文件夾下:

修改 閾值,分類類別:

改模型路徑:

編譯:
以下2個(gè)都要,進(jìn)行,不然測(cè)試和提交代碼,會(huì)報(bào)錯(cuò):
對(duì)工程,進(jìn)行編譯:
對(duì)test 文件夾,進(jìn)行編譯:
測(cè)試:?jiǎn)螐?圖片-命令行
需要把圖片,存到/project/inputs/ 文件夾下:
Nms的閾值修改:也可以像cid進(jìn)行傳參數(shù),調(diào)試

算法測(cè)試的時(shí)候,性能提不上去,可以這樣寫,去掉cv圖的輸出,但要保留 if 判斷,不然sdk規(guī)范性通過不了:

好啦,就補(bǔ)充這些,有問題請(qǐng)@我,一起解決~