yoloV5遷移學(xué)習(xí)應(yīng)用
數(shù)據(jù)標(biāo)注結(jié)果輸出
在10000個(gè)樣本中選擇200-300個(gè)好的樣本,盡量豐富,一個(gè)類型的主題的圖片選擇2-3張,形成一個(gè)新的存放數(shù)據(jù)的文件夾。
精靈標(biāo)注助手:打開軟件后,左上角“新建”-選擇“位置標(biāo)注”-選擇存放數(shù)據(jù)的文件夾-記錄標(biāo)簽并用英文逗號(hào)分隔類標(biāo)簽-創(chuàng)建
左側(cè)選擇標(biāo)注類型-矩形框-標(biāo)注好后按ctrl+s保存
精靈標(biāo)注助手快捷鍵:→控制下一個(gè)標(biāo)注對象
標(biāo)簽數(shù)據(jù)的導(dǎo)出:左下角有個(gè)“導(dǎo)出”按鍵,保存為pascal-voc格式。-“確定導(dǎo)出”。 默認(rèn)的標(biāo)簽數(shù)據(jù)文件夾導(dǎo)出的位置是在標(biāo)注的存放數(shù)據(jù)文件夾的下一級(jí)目錄下,也可以改文件夾。其中每個(gè)標(biāo)簽的名字與被標(biāo)注的圖片名字同名。
之后要把標(biāo)注的數(shù)據(jù)名稱進(jìn)行編號(hào),獨(dú)立一個(gè)python項(xiàng)目進(jìn)行操作。(如果文件名字不是數(shù)字的話就這樣處理)

數(shù)據(jù)集制作
將voc格式數(shù)據(jù)轉(zhuǎn)換成yolo v5格式的數(shù)據(jù)
需要將數(shù)據(jù)集分為訓(xùn)練集、驗(yàn)證集、測試集,比例8:1:1
里面注意改路徑
將標(biāo)注的框voc格式轉(zhuǎn)化為txt
這兩段代碼分別存儲(chǔ)為兩個(gè).py,存放在下面的路徑:

然后分別一次run每一個(gè)文件,然后在data-dataset-imagesets里面就會(huì)生成下面的.txt文件,這些是數(shù)據(jù)集中訓(xùn)練集、測試集(驗(yàn)證集,因?yàn)闇y試集要訓(xùn)練完后獲?。┑臉?biāo)簽。

2.模型訓(xùn)練
在data.yaml里面檢查修改path路徑、檢車下面的類數(shù)量以及類別

然后打開項(xiàng)目文件根目錄下的train.py? ?,? 查看一下訓(xùn)練參數(shù)有沒有需要改的

然后運(yùn)行這個(gè)文件。之后會(huì)在這個(gè)目錄下生成訓(xùn)練好的模型。

然后要測試集,之后看一下說明文件。有可能是運(yùn)行項(xiàng)目文件下的test.py
參考資料:https://blog.csdn.net/weixin_46751388/article/details/126070967
另外的全局性的參考資料:https://blog.csdn.net/FK2016/article/details/108000297

模型打包部署到邊緣端
遠(yuǎn)程登陸邊緣設(shè)備:在邊緣端的查看ip地址(點(diǎn)擊下方塢的cloudid),pc端瀏覽器上輸入邊緣端的IP地址鏈接到邊緣端。
邊緣端代碼存放在文件夾的home目錄

里面有個(gè)aidlux文件夾

把我們訓(xùn)練好的模型和監(jiān)測需要用的檢測detect數(shù)據(jù)上傳到aidlux文件目錄里面,就是這里

點(diǎn)這個(gè)上傳按鈕就能上傳文件(選左邊)。

然后更改.py文件(不一定是run.py)下面對應(yīng)的模型名稱以及detect數(shù)據(jù)的路徑(這里因?yàn)槭峭?jí)目錄,所以檢查一下名稱就行了)

如果模型訓(xùn)練時(shí)候的圖片大小更改過,上面in_shape, out_shape也需要更改,其他的就不需要更改了.
然后就是在linux終端執(zhí)行.py文件(cd到.py文件的目錄,然后)

可參考https://blog.csdn.net/weixin_47702737/article/details/121721523
細(xì)節(jié):
使用VOC格式數(shù)據(jù)的時(shí)候檢查VOC.yaml文件里面的路徑、模型類別數(shù)、類別的名稱。然后path里面不能在最后的VOC處加斜杠/。其實(shí)也可以用絕對路徑來替換掉。配置好后,訓(xùn)練模型的時(shí)候會(huì)根據(jù)這個(gè)自動(dòng)去尋找訓(xùn)練集、測試集等所在路徑。

yolov5在加載數(shù)據(jù)的時(shí)候,目錄是固定死的,數(shù)據(jù)在images目錄下,標(biāo)簽在labels中。
訓(xùn)練數(shù)據(jù)的時(shí)候這個(gè)JPEGimages名稱改為images
