一級目錄
二級目錄
三級目錄
1.下載yolov7源碼
yolov7官網(wǎng): https://github.com/WongKinYiu/yolov7
2.安裝Anaconda(選裝可以不下)
1.安裝Anaconda
miniconda清華源:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/
安裝最新款的
2.創(chuàng)建虛擬環(huán)境
在開始中找到Anaconda并打開
1.創(chuàng)建
conda create -n yolov7 python=3.7 -y
2.激活
conda activate yolov7
3.安裝依賴(這里演示的是不使用Anaconda虛擬環(huán)境)
1.注釋掉requirements.txt中默認(rèn)的torch(默認(rèn)安裝為cpu版)
打開yolov7源碼找到最下面的requirements.txt打開并將torch那一行注釋
打開終端安裝依賴
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
2.安裝gpu版本的torch
1.直接跳轉(zhuǎn)看詳細(xì)版
詳細(xì)版: https://blog.csdn.net/qq_62975494/article/details/129736414?spm=1001.2014.3001.5501
2.我也不知道叫什么版
不想寫了自己點上面連接看詳細(xì)版
4.下載權(quán)重
1.官網(wǎng)下載
yolov7官網(wǎng): https://github.com/WongKinYiu/yolov7
往下滑找到這點擊速度特別慢
2.百度云盤下載
鏈接: https://pan.baidu.com/s/1MHW31AIKcqHXYpt9bxenQA
提取碼1234
5.標(biāo)注數(shù)據(jù)集
1.創(chuàng)建所需文件夾
創(chuàng)建一個文件夾來存放數(shù)據(jù)集然后images用來存放圖片labels用來存放標(biāo)注的文件train:val的比例推薦4:1
2.下載labelImg標(biāo)注工具
先打開Anaconda(或者終端)
下圖為Anaconda
安裝
pip install labelImg
安裝成功后cmd中輸入labelImg即可打開
labelImg
詳情參考
鏈接: labelImg使用教程
3.labelImg使用
按下w選中你要標(biāo)注的物體即可
標(biāo)注完成后如圖所示labels中的txt文件對應(yīng)images中的圖片
6.配置訓(xùn)練的相關(guān)文件
總共有兩個文件需要配置,一個是/yolov7/cfg/training/yolov7.yaml,這個文件是有關(guān)模型的配置文件;一個是/yolov7/data/coco.yaml,這個是數(shù)據(jù)集的配置文件。
1.第一步,復(fù)制yolov7.yaml文件到相同的路徑下,然后重命名,我們重命名為yolov7-xxx.yaml(起一個你想要的名字)然后打開
2.復(fù)制coco.yaml文件到相同的路徑下,然后重命名,我們命名為xxx.yaml。(起一個你喜歡的名字)打開后進(jìn)行更改如下圖
7.開始訓(xùn)練
如果你用的是anaconda虛擬環(huán)境先進(jìn)入yolov7文件夾cmd激活虛擬環(huán)境
conda activate yolov7
然后開始訓(xùn)練如果你沒有使用虛擬環(huán)境可跳過上一步
下面是打開cmd的方法
輸入命令開始訓(xùn)練
python train.py --weights yolov7x.pt --cfg cfg/training/yolov7-xxx.yaml --data data/xxx.yaml --batch-size 8 --epoch 300 --device 0
對參數(shù)解釋
–cfg 接受模型配置的參數(shù)
–data 接收數(shù)據(jù)配置的參數(shù)
–device 0 訓(xùn)練類型,我是一塊GPU 所以用0
–batch-size 8 GPU內(nèi)存大小決定
–epoch 訓(xùn)練次數(shù),建議300
–weights 訓(xùn)練的權(quán)重
last.pt為最后一次的權(quán)重
best.pt為測試效果最好的權(quán)重文章來源:http://www.zghlxwxcb.cn/news/detail-469210.html
8.使用訓(xùn)練的.pt文件進(jìn)行檢測
python detect.py --weights best.pt的位置 --source 檢測的文件位置 --device 0
在./runs/detect中可查看結(jié)果
over(推薦使用線上gpu)文章來源地址http://www.zghlxwxcb.cn/news/detail-469210.html
到了這里,關(guān)于yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!