前言

首先，先說明我只是初步接觸yolov7,寫這篇文章的主要目的是可以讓大家快速應(yīng)用自己的數(shù)據(jù)集進(jìn)行訓(xùn)練。沒有接觸過yolov5也沒有關(guān)系，該篇文章會逐步進(jìn)行演示如何訓(xùn)練。

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

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

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

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

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

Main.py就是劃分你的數(shù)據(jù)為訓(xùn)練集和測試集.txt的腳本。整體新建的項(xiàng)目目錄就是這樣。下面說一下操作流程：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ù)據(jù)準(zhǔn)備階段已經(jīng)完成，項(xiàng)目目錄如圖：

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

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

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

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

總結(jié)

到此這篇關(guān)于保姆級官方y(tǒng)olov7訓(xùn)練自己的數(shù)據(jù)集及項(xiàng)目部署的文章就介紹到這了,更多相關(guān)yolov7訓(xùn)練自己的數(shù)據(jù)集內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論