yolov7數(shù)據(jù)集及項(xiàng)目部署的方法

這篇“yolov7數(shù)據(jù)集及項(xiàng)目部署的方法”文章的知識(shí)點(diǎn)大部分人都不太理解，所以小編給大家總結(jié)了以下內(nèi)容，內(nèi)容詳細(xì)，步驟清晰，具有一定的借鑒價(jià)值，希望大家閱讀完這篇文章能有所收獲，下面我們一起來(lái)看看這篇“yolov7數(shù)據(jù)集及項(xiàng)目部署的方法”文章吧。

第一步 數(shù)據(jù)集準(zhǔn)備

首先確保你有l(wèi)abelimg標(biāo)圖軟件，若無(wú)，需要自行去下一個(gè)并看一下標(biāo)圖教程。

當(dāng)你已經(jīng)標(biāo)注完成，獲得了img以及相對(duì)應(yīng)的xml之后（如圖）

接下來(lái)就是可選擇項(xiàng)：是否需要圖像增強(qiáng)來(lái)獲取更多樣本，如需要點(diǎn)擊這里下載 ，直接用enhance_img.py去增強(qiáng)。之后我的習(xí)慣是每有一個(gè)項(xiàng)目需要訓(xùn)練，則會(huì)新建一個(gè)文件夾，項(xiàng)目存放的文件如圖：

下面ImageSets\Main用于存放后續(xù)腳本文件劃分訓(xùn)練集測(cè)試集的相對(duì)應(yīng)的train.txt,test.txt。hyp.scratch.p5.yaml為yolov7的超參設(shè)置，可以直接從yolov7\data下面去拷過(guò)來(lái)放到你項(xiàng)目里。icon.yaml為你要所要訓(xùn)練的類別和相應(yīng)的類，同時(shí)也會(huì)寫(xiě)上實(shí)際訓(xùn)練時(shí)訓(xùn)練數(shù)據(jù)和測(cè)試數(shù)據(jù)。該文件如下圖。

Main.py就是劃分你的數(shù)據(jù)為訓(xùn)練集和測(cè)試集.txt的腳本。整體新建的項(xiàng)目目錄就是這樣。下面說(shuō)一下操作流程：1 首先確保是該項(xiàng)目目錄方式 2 運(yùn)行main.py腳本文件得到了ImageSets\Main下面的train.txt,test.txt。3 在yolov7的根目錄下運(yùn)行xml2txt腳本文件。main.py xml2txt.py文件點(diǎn)擊這里下載 注意要將該文件的類和項(xiàng)目名改成自己的。如圖：

此時(shí)，我們的數(shù)據(jù)準(zhǔn)備階段已經(jīng)完成，項(xiàng)目目錄如圖：

第二步 train.py載入自己的數(shù)據(jù)集并訓(xùn)練

紅色箭頭和紅框就是常見(jiàn)改動(dòng)的地方，改動(dòng)后就可以訓(xùn)練了。

第三步 將訓(xùn)練好的pt文件做成接口調(diào)用

在等待訓(xùn)練完成之后，就會(huì)在runs/train下面獲得訓(xùn)練的best.pt,你可以拿著這個(gè)pt去做接口使用了。首先，在自己的項(xiàng)目里使用必須要確保yolov7根目錄下的models和utils文件夾放到了你的項(xiàng)目根目錄。然后下載model_import.py 點(diǎn)擊這里下載 嵌入你的任何項(xiàng)目路徑下調(diào)用predict函數(shù)就可以輸出檢出結(jié)果了。

以上就是關(guān)于“yolov7數(shù)據(jù)集及項(xiàng)目部署的方法”這篇文章的內(nèi)容，相信大家都有了一定的了解，希望小編分享的內(nèi)容對(duì)大家有幫助，若想了解更多相關(guān)的知識(shí)內(nèi)容，請(qǐng)關(guān)注億速云行業(yè)資訊頻道。