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

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

12 人参与  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