国产 无码 综合区,色欲AV无码国产永久播放,无码天堂亚洲国产AV,国产日韩欧美女同一区二区

yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版

這篇具有很好參考價值的文章主要介紹了yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版。希望對大家有所幫助。如果存在錯誤或未考慮完全的地方,請大家不吝賜教,您也可以點擊"舉報違法"按鈕提交疑問。

一級目錄

二級目錄

三級目錄

1.下載yolov7源碼

yolov7官網(wǎng): https://github.com/WongKinYiu/yolov7
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版

2.安裝Anaconda(選裝可以不下)

1.安裝Anaconda

miniconda清華源:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/
安裝最新款的

2.創(chuàng)建虛擬環(huán)境

在開始中找到Anaconda并打開
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
1.創(chuàng)建

conda create -n yolov7 python=3.7 -y

2.激活

conda activate yolov7

yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版

3.安裝依賴(這里演示的是不使用Anaconda虛擬環(huán)境)

1.注釋掉requirements.txt中默認(rèn)的torch(默認(rèn)安裝為cpu版)

打開yolov7源碼找到最下面的requirements.txt打開并將torch那一行注釋
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
打開終端安裝依賴

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
往下滑找到這點擊速度特別慢

yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
2.百度云盤下載
鏈接: https://pan.baidu.com/s/1MHW31AIKcqHXYpt9bxenQA
提取碼1234

5.標(biāo)注數(shù)據(jù)集

1.創(chuàng)建所需文件夾

yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
創(chuàng)建一個文件夾來存放數(shù)據(jù)集然后images用來存放圖片labels用來存放標(biāo)注的文件train:val的比例推薦4:1

2.下載labelImg標(biāo)注工具

先打開Anaconda(或者終端)
下圖為Anaconda
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
安裝

pip install labelImg

安裝成功后cmd中輸入labelImg即可打開

labelImg

詳情參考
鏈接: labelImg使用教程

3.labelImg使用

yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
按下w選中你要標(biāo)注的物體即可
標(biāo)注完成后如圖所示labels中的txt文件對應(yīng)images中的圖片
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版

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(起一個你想要的名字)然后打開
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
2.復(fù)制coco.yaml文件到相同的路徑下,然后重命名,我們命名為xxx.yaml。(起一個你喜歡的名字)打開后進(jìn)行更改如下圖
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版

7.開始訓(xùn)練

如果你用的是anaconda虛擬環(huán)境先進(jìn)入yolov7文件夾cmd激活虛擬環(huán)境

conda activate yolov7

然后開始訓(xùn)練如果你沒有使用虛擬環(huán)境可跳過上一步
下面是打開cmd的方法
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
輸入命令開始訓(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)重
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
last.pt為最后一次的權(quán)重
best.pt為測試效果最好的權(quán)重

8.使用訓(xùn)練的.pt文件進(jìn)行檢測

python detect.py --weights best.pt的位置 --source 檢測的文件位置 --device 0

yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
在./runs/detect中可查看結(jié)果
yolov7訓(xùn)練自己的數(shù)據(jù)集-gpu版
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)!

本文來自互聯(lián)網(wǎng)用戶投稿,該文觀點僅代表作者本人,不代表本站立場。本站僅提供信息存儲空間服務(wù),不擁有所有權(quán),不承擔(dān)相關(guān)法律責(zé)任。如若轉(zhuǎn)載,請注明出處: 如若內(nèi)容造成侵權(quán)/違法違規(guī)/事實不符,請點擊違法舉報進(jìn)行投訴反饋,一經(jīng)查實,立即刪除!

領(lǐng)支付寶紅包贊助服務(wù)器費用

相關(guān)文章

覺得文章有用就打賞一下文章作者

支付寶掃一掃打賞

博客贊助

微信掃一掃打賞

請作者喝杯咖啡吧~博客贊助

支付寶掃一掃領(lǐng)取紅包,優(yōu)惠每天領(lǐng)

二維碼1

領(lǐng)取紅包

二維碼2

領(lǐng)紅包