位置: IT常识 - 正文

手把手教你用Yolov5 (v6.2) 训练分类模型 基于《Kaggle猫狗大战》案例(手把手教你用气焊视频)

编辑:rootadmin
手把手教你用Yolov5 (v6.2) 训练分类模型 基于《Kaggle猫狗大战》案例 文章目录0 前言1 v6.2 项目结构改动2 快速上手🌟2.1 train2.2 val2.3 predict3 重要参数解析🚀3.1 "--data"🍀3.1.1 划分数据集3.1.2 修改超参数3.2 "--seed" 🍀4 模型推理5 Test集验证🎉更多内容导航有问题欢迎大家指正,如果感觉有帮助的话请点赞支持下👍📖🌟0 前言

推荐整理分享手把手教你用Yolov5 (v6.2) 训练分类模型 基于《Kaggle猫狗大战》案例(手把手教你用气焊视频),希望有所帮助,仅作参考,欢迎阅读内容。

文章相关热门搜索词:手把手教你用荣耀手机,手把手教你用u盘安装系统,手把手教你用电脑上网络,手把手教你用电脑写文章,手把手教你用气焊视频,手把手教你用砖砌橱柜,手把手教你用砖砌橱柜,手把手教你用二氧化氯给鱼缸消毒,内容如对您有帮助,希望把文章链接给更多的朋友!

在8月17日晚上,YOLOv5官方发布了v6.2版本,v6.2版本支持分类模型训练、验证、预测和导出;v6.2版本的推出使得训练分类器模型变得超级简单!

下一个版本v6.3计划于9月发布,将为YOLOv5带来官方实例分割支持,今年晚些时候将发布一个主要的v7.0版本,更新所有3个任务的架构——分类、检测和分割。

并且Yolov5官方同样提供了大量的预训练权重供大家选择;

权重都是在ImageNet数据集上训练90 Epoch后得到的,并且YOLOv5官方将所有型号导出到ONNX FP32进行CPU速度测试,导出到 TensorRT FP16进行GPU速度测试,最终得到的各项指标如下图所示:

分类任务的工作流与先前版本的检测任务工作流基本一致,所以熟悉检测任务的同学应该会很容易上手分类任务

1 v6.2 项目结构改动

v6.2版本项目结构并无太大改变,主要是增加了classify文件夹以及predict.py train.py val.py 这三个文件;那么这三个文件也分别对应着分类模型的推理、训练和验证。

2 快速上手🌟

YOLOv5分类训练支持使用--data参数自动下载MNIST、Fashion-MNIST、CIFAR10、CIFAR100、Imagenette、Imagewoof 和 ImageNet 数据集。 例如,要开始在MNIST上进行训练,只需要给–data参数设置为“MNIST”

2.1 train# Single-GPUpython classify/train.py --model yolov5s-cls.pt --data cifar100 --epochs 5 --img 224 --batch 128# Multi-GPU DDPpython -m torch.distributed.run --nproc_per_node 4 --master_port 1 classify/train.py --model yolov5s-cls.pt --data imagenet --epochs 5 --img 224 --device 0,1,2,32.2 valbash data/scripts/get_imagenet.sh --val # download ImageNet val split (6.3G, 50000 images)python classify/val.py --weights yolov5m-cls.pt --data ../datasets/imagenet --img 224 # validate2.3 predictpython classify/predict.py --weights yolov5s-cls.pt --data data/images/bus.jpg3 重要参数解析🚀

下面来看一下train.py具体的参数信息 分类任务的参数与检测任务的参数差的并不是很多,大多数在参数的含义在6.1版本我已经讲过了,我这里就挑出几个重点的说一下

v6.1版本的参数解析可以参考我这两篇博文 手把手带你调参Yolo v5 (v6.1)(一) 手把手带你调参Yolo v5 (v6.1)(二)

3.1 "–data"🍀

Yolov5(v6.2)版本的分类数据集信息不再通过配置文件的形式载入,而是通过文件夹的名字自动匹配,官方提供了一些数据集的url可以自行下载;

本篇博文数据集以kaggle猫狗大战数据集为例子,从零开始训练自己的分类模型

3.1.1 划分数据集

kaggle猫狗大战数据集包含12500张猫🐱图片与12500张🐶狗图片,我按照如下方式划分

kaggle-DogVsCat├─train│ ├─Cat 8750│ └─Dog 8750├─val│ ├─Cat 2500│ └─Dog 2500└─test ├─Cat 1250 └─Dog 1250

分类数据集的路径是自动匹配的,所以我们要按照这个命名规则划分,不然会提示你找不到数据集

3.1.2 修改超参数

划分好数据集后做如下改动:--data default=“你分类数据集的名字”

手把手教你用Yolov5 (v6.2) 训练分类模型 基于《Kaggle猫狗大战》案例(手把手教你用气焊视频)

随后就可以训练了

分类模型较检测模型训练速度快非常多,可能是训练图像尺寸设置的比较小的原因;可以通过--imgsz参数设置

3.2 “–seed” 🍀

这是v6.2版本更新的一个非常重要的参数,使用torch>=1.12.0的单GPU训练完全可再现

4 模型推理

推理方式与检测任务几乎没有区别,修改一下权重路径就好

分类任务推理出的图片是这样子的,会显示准确率top5的类别

5 Test集验证

这里也没什么可说的,修改一下权重路径和数据集就好

不知道是不是我没用明白,还没看出来怎么修改网络结构,所以说出个分类模型的意义是什么?

🎉更多内容导航

1.手把手带你调参Yolo v5 (v6.2)(一)🌟强烈推荐

2.手把手带你调参Yolo v5 (v6.2)(二)🚀

3.如何快速使用自己的数据集训练Yolov5模型

4.手把手带你Yolov5 (v6.1)添加注意力机制(一)(并附上30多种顶会Attention原理图)🌟

5.手把手带你Yolov5 (v6.1)添加注意力机制(二)(在C3模块中加入注意力机制)

6.Yolov5如何更换激活函数?

7.Yolov5 (v6.1)数据增强方式解析

8.Yolov5更换上采样方式( 最近邻 / 双线性 / 双立方 / 三线性 / 转置卷积)

9.Yolov5如何更换EIOU / alpha IOU / SIoU?

10.Yolov5更换主干网络之《旷视轻量化卷积神经网络ShuffleNetv2》🍀

11.YOLOv5应用轻量级通用上采样算子CARAFE🍀

12.空间金字塔池化改进 SPP / SPPF / ASPP / RFB / SPPCSPC🍀

13.用于低分辨率图像和小物体的新 CNN 模块SPD-Conv🍀

14.持续更新中

有问题欢迎大家指正,如果感觉有帮助的话请点赞支持下👍📖🌟
本文链接地址:https://www.jiuchutong.com/zhishi/296110.html 转载请保留说明!

上一篇:2023年noc指导教师认证测评参考试题(ccf noi指导教师认证考核)

下一篇:ConvNeXt V2学习笔记

  • 屏幕旋转设置在什么地方(屏幕旋转咋调)

    屏幕旋转设置在什么地方(屏幕旋转咋调)

  • 华为p30pro看视频模糊(华为p30pro看视频卡顿)

    华为p30pro看视频模糊(华为p30pro看视频卡顿)

  • 微信激活成功了为什么还是不能登录(微信激活失败怎么办)

    微信激活成功了为什么还是不能登录(微信激活失败怎么办)

  • word文档艺术字文字环绕方式在哪里(word文档艺术字样式16在哪)

    word文档艺术字文字环绕方式在哪里(word文档艺术字样式16在哪)

  • 苹果手机用户正忙是什么意思(苹果手机用户正忙是不是拉黑了)

    苹果手机用户正忙是什么意思(苹果手机用户正忙是不是拉黑了)

  • 全民k歌来自微信分享是什么意思(全民k歌显示微信好友,但是不知道是哪位)

    全民k歌来自微信分享是什么意思(全民k歌显示微信好友,但是不知道是哪位)

  • 魔音耳机和beats区别(魔音耳机和森海塞尔哪个好)

    魔音耳机和beats区别(魔音耳机和森海塞尔哪个好)

  • qq托腮是什么意思(qq托腮表情大图)

    qq托腮是什么意思(qq托腮表情大图)

  • ipad开机键坏了怎么开机(iPad开机键坏了开不了机怎么办)

    ipad开机键坏了怎么开机(iPad开机键坏了开不了机怎么办)

  • 小米手机短信闪退怎么回事(小米手机短信闪灯怎么关闭)

    小米手机短信闪退怎么回事(小米手机短信闪灯怎么关闭)

  • 快手极速版审核中的钱要多久才能提现(快手极速版审核的钱迟迟不到)

    快手极速版审核中的钱要多久才能提现(快手极速版审核的钱迟迟不到)

  • 手机导航老是显示信号弱是怎么回事(手机导航老是显示位于室内怎么办)

    手机导航老是显示信号弱是怎么回事(手机导航老是显示位于室内怎么办)

  • 下架的app怎么迁移(下架的app怎么迁移到新手)

    下架的app怎么迁移(下架的app怎么迁移到新手)

  • 微信评论区怎么发图片(微信评论区怎么发图)

    微信评论区怎么发图片(微信评论区怎么发图)

  • 锐龙r73750h相当于i几(锐龙r7 3750h性能)

    锐龙r73750h相当于i几(锐龙r7 3750h性能)

  • 华为微信怎么开美颜(华为微信怎么开启抖音权限)

    华为微信怎么开美颜(华为微信怎么开启抖音权限)

  • word里替换数字字体(word替换数字通配符)

    word里替换数字字体(word替换数字通配符)

  • vivo手机root权限在哪(vivo手机root权限怎么设置)

    vivo手机root权限在哪(vivo手机root权限怎么设置)

  • 手机怎么设置用户正忙(手机怎么设置用笔触屏)

    手机怎么设置用户正忙(手机怎么设置用笔触屏)

  • 电脑怎么弄浏览器快捷栏(电脑怎么弄浏览器打开)

    电脑怎么弄浏览器快捷栏(电脑怎么弄浏览器打开)

  • 电脑强开手机usb调试(电脑强制打开手机usb)

    电脑强开手机usb调试(电脑强制打开手机usb)

  • 确认收货前后退款区别(确认收货前退款与收货后退款区别)

    确认收货前后退款区别(确认收货前退款与收货后退款区别)

  • 电脑怎么清除数据(电脑怎么清除数据恢复出厂设置)

    电脑怎么清除数据(电脑怎么清除数据恢复出厂设置)

  • 什么叫应纳税所得额
  • 养猪场死猪处理方法
  • 应纳税所得额可以扣除哪些
  • 财务台账包含什么
  • 费用发票跨月可以入帐吗
  • 滞留票税务局会罚款多少
  • 个税申报时提示扣缴单位无有效的税费种认定信息
  • 外商投资企业啥意思
  • 预付账款转出分录
  • 现房销售土增税怎么缴纳
  • 以前月度费用当期怎么入账合适?
  • 自己做的房子可以办房产证吗
  • 实收资本印花税怎么申报税目
  • 电子发票缩小多少比例打印
  • 退役士兵创业就业政策
  • 税务领取发票后怎么操作
  • 提高并购成本
  • 收回购货方前欠货款属于什么
  • 银行年费计入管理费用还是财务费用
  • 增值税发票普票和专票有什么区别
  • 企业注销时资本公积怎么处理
  • 代扣代缴增值税可以抵扣进项税?
  • 房地产公司未售房产出租账务处理
  • Win11 Dev Build 22000.65开发预览版推送(附更新修复已知问题汇总)
  • linux 将一个文件的内容给另一个文件
  • 取得进项税发票分录
  • window清除临时存储文件夹
  • 重命名电脑名称有什么用
  • ubuntu 安装vs code
  • 人工智能导论报告
  • php中字符串函数
  • framework教程
  • 无票收入什么时候确认收入
  • 办公室装修入账
  • 不良贷款转让需遵循的原则
  • php email
  • 4月满月是几号
  • 减征增值税会计处理
  • Web渗透测试新手实操详解
  • 卖废品收入计入需要交税吗
  • vue2.
  • 宣传活动费用什么发票开?
  • php无限分类实现不实用递归
  • 企业有外币账户怎样做账
  • 社保费用如何做账,社保费会计分录如何写
  • java集合详解以及底层实现
  • python搞自动化
  • 印刷行业成本核算程序
  • 计提和结转的例子
  • 企业将重组债务转为资本的会计处理
  • 外购商品发放给员工 进项税额能不能抵扣
  • 股东不任职,是否可以参与企业经营
  • 服装改制注意事项
  • 营业税改征增值税跨境应税行为增值税免税管理办法
  • 销售自行开发的房地产项目的增值税
  • 进项大于销项的原因
  • 母子公司间借款利息免增值税
  • 公交充值卡发票能报销吗
  • 其他货币资金包含的内容
  • sqlserver排序规则怎么看
  • mysql 5.7 8.0
  • Linux系统下Qt的基本安装和配置
  • wibdows任务管理器
  • srv32.exe - srv32进程是什么文件 有何作用
  • xp调出输入法
  • django项目中遇到的难点
  • unity3D游戏开发
  • js中的filter方法和map方法
  • js对话框确定事件
  • python不能用
  • unity arkit
  • Javascript字符串出现次数
  • python怎么用
  • python约瑟夫问题最高效算法
  • won10安装
  • 欧美 房产税
  • 家庭农场有什么用途
  • 航天信息的金税盘开票要怎么调格式
  • 欠税多少构成犯罪
  • 土地税是怎么缴纳按什么缴纳
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

    网站地图: 企业信息 工商信息 财税知识 网络常识 编程技术

    友情链接: 武汉网站建设