当前位置:首页 » 《随便一记》 » 正文

YOLOV7实战(pycharm)-以口罩检测为例

8 人参与  2022年11月14日 12:33  分类 : 《随便一记》  评论

点击全文阅读


?欢迎来到深度学习的世界 
?博客主页:卿云阁

?欢迎关注?点赞?收藏⭐️留言?

?本文由卿云阁原创!

?本阶段属于练气阶段,希望各位仙友顺利完成突破

?首发时间:?2022年11月10日?

✉️希望可以和大家一起完成进阶之路!

?作者水平很有限,如果发现错误,请留言轰炸哦!万分感谢!


目录

? 下载代码和权重,配置环境

?准备数据集

?添加数据配置文件

?修改train.py的参数

?模型测试

下载代码和权重,配置环境

链接:https://pan.baidu.com/s/18-TslquipRAtIWqd_PSenA?pwd=61lh 
提取码:61lh

权重所放路径:

 配置环境:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple +模块名

 ?准备数据集

采用Labelme标注的数据格式,这里的数据集是我提前准备好的,需要的小伙伴可以下载哦。本文使用的是口罩的数据集,新建一个mydata文件夹,把数据放进去。

链接:https://pan.baidu.com/s/1wrsvXv5-od8ZtlvJCj9zgQ?pwd=0hti 
提取码:0hti

把它整理成这个格式:

?


?添加数据配置文件

修改yolov5/data文件夹下的内容:

内容如下所示:(这里是我用的服务器上的地址)

train: /tmp/pycharm_project_558/dataset/MaskDataSet/train/imagesval: /tmp/pycharm_project_558/dataset/MaskDataSet/valid/images nc: 2names: ['mask', 'no-mask']

其中:

path:数据集的根目录train:训练集与path的相对路径val:验证集与path的相对路径nc:类别数量,因为这个数据集只有一个类别(fire),nc即为1。names:类别名字。

?修改train.py的参数

fg参数是YoloV5 模型的配置文件,模型的文件存放在models文件夹下面,按照需求填写不同的文件。 data是配置数据集的配置文件,我们选用的是voc.yaml,所以配置data/coco.yaml 修改上面三个参数就可以开始训练了,其他的参数根据自己的需求修改。修改后的参数配置如下: parser.add_argument('--weights', type=str, default='yolov7.pt', help='initial weights path') parser.add_argument('--cfg', type=str, default='', help='model.yaml path') parser.add_argument('--data', type=str, default='data/coco.yaml', help='data.yaml path')

 


?模型测试

模型训练完成后,将runs/exp/weights下的模型(best.pt)复制在yolov7文件夹下。

开始测试。

    parser.add_argument('--weights', nargs='+', type=str, default='best.pt', help='model.pt path(s)')    parser.add_argument('--data', type=str, default='data/coco.yaml', help='*.data path')

 

 


点击全文阅读


本文链接:http://m.zhangshiyu.com/post/48333.html

<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

最新文章

  • 她的记忆停留在了最爱初恋的那年许欣柔楚临川完本_她的记忆停留在了最爱初恋的那年(许欣柔楚临川)
  • 全书浏览假千金的实习生男友霸占我办公室,我反手让他们倾家荡产(顾家明)_假千金的实习生男友霸占我办公室,我反手让他们倾家荡产(顾家明)全书结局
  • 童养夫让我给他的新欢出修复费(林嘉芝林思雅)_童养夫让我给他的新欢出修复费林嘉芝林思雅
  • 全文资助生女婿让我给他白月光付三千万月子中心钱(宋清玉宋雅)列表_全文资助生女婿让我给他白月光付三千万月子中心钱
  • 碎在时光里的谎言喻景宴秦明月完本_碎在时光里的谎言(喻景宴秦明月)
  • 旧爱剜心吻成灰席鄢之岑秋全书免费旧爱剜心吻成灰席鄢之岑秋全书免费
  • 结婚六年丈夫不碰我谁知儿子亲爹是寡头(纪清言傅司砚),结婚六年丈夫不碰我谁知儿子亲爹是寡头
  • 老公想换掉我的男胎,我笑他自不量力(宋薇于继业)_老公想换掉我的男胎,我笑他自不量力宋薇于继业
  • 给太子下了噬心蛊后,皇后找上门(小夭赵劼)全书浏览_给太子下了噬心蛊后,皇后找上门全书浏览
  • 豪门绝嗣!带球跑的夫人回来了!(谢长宴慕清杳)_豪门绝嗣!带球跑的夫人回来了!谢长宴慕清杳
  • 完美身材(李朵林之晴)_完美身材李朵林之晴
  • 离婚后,我和快穿系统绑定(白意秋陈荣周立慧)_离婚后,我和快穿系统绑定(白意秋陈荣周立慧)

    关于我们 | 我要投稿 | 免责申明

    Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1