一、下載Yolov7
github地址:https://github.com/WongKinYiu/yolov7
或者命令行下載
git clone https://github.com/WongKinYiu/yolov7.git
二、新建一個Python環(huán)境
用Acaconda新建一個干凈的環(huán)境,我命名為yolov7,python版本選擇3.8
進入到剛剛下載的yolov7的目錄中,安裝yolov7所需的第三方庫
三、修改訓練數(shù)據(jù)參數(shù)
打開yolov7項目,再打開data文件夾,拷貝一份coco.yaml
1、修改里面的trian
、val
、test
的路徑,改成我們自己的數(shù)據(jù)集的路徑;
2、修改分類的數(shù)量,按照自己的需求修改;
3、names
中改成自己的分類的名稱。
四、下載預訓練模型
github地址:https://github.com/WongKinYiu/yolov7/releases
找一個自己想要訓練的模型的文件
在yolov7的項目中新建一個weight
的文件夾,將預訓練模型放到里面。
五、修改訓練參數(shù)
打開train.py
,把下面幾個參數(shù),根據(jù)自己的實際情況改一下
-
--weights
剛剛下載的預訓練模型文件 -
--cfg
在yolov7項目文件夾下的cfg
文件夾中,選擇自己對應的模型 -
--data
剛剛修改的數(shù)據(jù)文件
六、開始訓練
執(zhí)行命令
python train.py
訓練完成之后會提示文章來源:http://www.zghlxwxcb.cn/news/detail-412131.html
七、測試
python detect.py --weights runs/train/exp/weights/best.pt --conf 0.25 --img-size 640 --source testimg.png --device 0 --save-txt
文章來源地址http://www.zghlxwxcb.cn/news/detail-412131.html
到了這里,關于【Yolov7】配置參數(shù)以及訓練模型的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章,希望大家以后多多支持TOY模板網(wǎng)!