模型訓(xùn)練說明
通過本網(wǎng)站,用戶可使用自己的圖片數(shù)據(jù)來定制屬于自己應(yīng)用場景的模型,生成的模型可下載到Edgeideal計算盒中運(yùn)行,方便部署使用。用戶無需編程和技術(shù)背景知識,只需要提供圖片,在網(wǎng)站上即可完成整個標(biāo)注訓(xùn)練的定制過程。
整體介紹可參考下面文章:
深度學(xué)習(xí)標(biāo)注訓(xùn)練部署一體化平臺介紹
整體流程圖如下:

本文主要介紹模型訓(xùn)練,模型測試及下載部署等步驟
圖片標(biāo)注完成之后即可進(jìn)入下一步,啟動模型訓(xùn)練。點擊“訓(xùn)練平臺”->“標(biāo)注和訓(xùn)練”菜單,進(jìn)入模型列表頁面。
選擇擬進(jìn)行訓(xùn)練的模型,點擊“訓(xùn)練”按鈕進(jìn)入模型訓(xùn)練管理界面。

點擊“啟動訓(xùn)練”按鈕,即可啟動模型訓(xùn)練后臺訓(xùn)練,按鈕將變?yōu)椤敖K止訓(xùn)練”,訓(xùn)練進(jìn)度或出錯提示會出現(xiàn)在按鈕旁邊,也可在列表的“訓(xùn)練狀態(tài)”列查看,訓(xùn)練如需中途終止,點擊“終止訓(xùn)練”按鈕即可中止。

如無錯誤,等訓(xùn)練狀態(tài)變成“訓(xùn)練已完成”,即表示本次啟動的訓(xùn)練已經(jīng)正常完成,點擊“顯示訓(xùn)練結(jié)果”即可展開訓(xùn)練結(jié)果頁面,查看各種訓(xùn)練結(jié)果數(shù)據(jù):

用戶所有的標(biāo)注樣本會被隨機(jī)劃分為訓(xùn)練集和測試集,測試集約占總樣本數(shù)的15%。測試集樣本不輸入模型訓(xùn)練過程,只用來評估模型效果。以上表格顯示了每一輪訓(xùn)練結(jié)束以后,在測試集上評估的結(jié)果。
訓(xùn)練數(shù)據(jù)按類別分行顯示,第一行為所有類別統(tǒng)計數(shù)據(jù)。下面為單獨的類別統(tǒng)計數(shù)據(jù)
下面解釋各列數(shù)據(jù)的含義:
測試圖片數(shù):測試集中的圖片數(shù)量
測試樣本數(shù):測試集中該類別的樣本數(shù)
準(zhǔn)確率,召回率,平均精度(mAP)為目標(biāo)檢測領(lǐng)域的通用指標(biāo)定義,
具體解釋可參考此鏈接:https://www.jianshu.com/p/fbb96bb49782
查看測試集結(jié)果列中可顯示每張測試集圖片預(yù)測結(jié)果框和標(biāo)注結(jié)果框,
用戶可直觀的分析每張圖片的預(yù)測效果。
測試集圖片根據(jù)預(yù)測效果被分為正確結(jié)果和錯誤結(jié)果兩類,可以分別點擊對應(yīng)鏈接展開。

用戶可點擊文件名鏈接,對應(yīng)的圖片預(yù)測效果將直接顯示在頁面中,其中紅色框為預(yù)測框,黃色框為標(biāo)注框。
如點擊“圖片中僅顯示當(dāng)前類別框”則會隱藏不屬于當(dāng)前類別的預(yù)測框和標(biāo)注框,只顯示當(dāng)前類別的,以免干擾顯示。
每個類別的正確結(jié)果和錯誤結(jié)果可調(diào)整參數(shù)來調(diào)節(jié),點擊每行右側(cè)的“參數(shù)設(shè)置”按鈕,可為每個類別設(shè)置不同參數(shù):

可設(shè)置參數(shù)包括置信度閾值和IOU閾值,
其中交并比(IOU)是度量兩個框(預(yù)測框和標(biāo)注框)的交疊程度,公式如下:

就是兩個框的交集面積除以并集面積,最大為1,最小為0
系統(tǒng)首先根據(jù)置信度閾值過慮預(yù)測框,僅保留高于該閾值的預(yù)測框,然后計算預(yù)測框和標(biāo)注框的IOU,如果該圖片所有的樣本框的IOU值都大于所設(shè)置的閾值,該圖片將顯示為預(yù)測正確結(jié)果,否則將顯示為預(yù)測錯誤結(jié)果。
通過嘗試調(diào)節(jié)參數(shù),為每個類別設(shè)置不同參數(shù),用戶可以得到更精確的模型部署準(zhǔn)確率,降低誤識別和漏識別率。
點擊"測試模型",即可上載實際測試圖片對剛才訓(xùn)練得到的模型識別效果進(jìn)行進(jìn)一步的測試評估。可選擇多張進(jìn)行測試。

如需下載測試結(jié)果圖片,點擊“下載測試結(jié)果”打包下載測試結(jié)果。
如果模型測試效果不好,可以返回修改圖片集和標(biāo)注,然后再次啟動訓(xùn)練,測試效果,反復(fù)迭代。
確認(rèn)效果滿意之后,可點擊“下載模型”下載最終得到的模型文件,提供給離線識別盒或SDK使用。