【Yolov7】配置参数以及训练模型

这篇具有很好参考价值的文章主要介绍了【Yolov7】配置参数以及训练模型。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

一、下载Yolov7

github地址:https://github.com/WongKinYiu/yolov7
或者命令行下载

git clone https://github.com/WongKinYiu/yolov7.git

二、新建一个Python环境

用Acaconda新建一个干净的环境,我命名为yolov7,python版本选择3.8
【Yolov7】配置参数以及训练模型
进入到刚刚下载的yolov7的目录中,安装yolov7所需的第三方库
【Yolov7】配置参数以及训练模型

三、修改训练数据参数

打开yolov7项目,再打开data文件夹,拷贝一份coco.yaml
1、修改里面的trianvaltest的路径,改成我们自己的数据集的路径;
2、修改分类的数量,按照自己的需求修改;
3、names中改成自己的分类的名称。
【Yolov7】配置参数以及训练模型

四、下载预训练模型

github地址:https://github.com/WongKinYiu/yolov7/releases
找一个自己想要训练的模型的文件
【Yolov7】配置参数以及训练模型
在yolov7的项目中新建一个weight的文件夹,将预训练模型放到里面。
【Yolov7】配置参数以及训练模型

五、修改训练参数

打开train.py,把下面几个参数,根据自己的实际情况改一下

  • --weights刚刚下载的预训练模型文件
  • --cfg在yolov7项目文件夹下的cfg文件夹中,选择自己对应的模型
    【Yolov7】配置参数以及训练模型
  • --data刚刚修改的数据文件
    【Yolov7】配置参数以及训练模型

六、开始训练

执行命令

python train.py

训练完成之后会提示
【Yolov7】配置参数以及训练模型

七、测试

python detect.py --weights runs/train/exp/weights/best.pt --conf 0.25 --img-size 640 --source testimg.png --device 0 --save-txt

【Yolov7】配置参数以及训练模型
【Yolov7】配置参数以及训练模型文章来源地址https://www.toymoban.com/news/detail-412131.html

到了这里,关于【Yolov7】配置参数以及训练模型的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包