YOLOv5超详细的入门级教程(思考篇)(二)——初次训练模型的一些小白问题

想当初小小白的时候问的一些奇奇怪怪的问题,记录一下

  1. 半年前的我:一般训练的时候 epoch都设多少啊?
    导师:可以计算。要能把数据集遍历至少一遍,必要的时候多遍历几遍数据集。
    要知道你每个epoch用多少数据。
    反正要是收敛的比较快,epoch设小一点;收敛的慢 就设大一点。
    师姐:就一般loss要比刚开始的低两个小数位差不多,然后不怎么变化了 就基本收敛了
    现在的我:请打开tensorboard。

  2. 半年前的我 :Imagesize是根据数据集中的图片分辨率来设定的吗?
    师姐:是的
    现在的我:这是什么问题??

  3. 半年前的我:焊缝的那个图片是6000x2000的分辨率,怎么处理啊?
    现在的我:小batchsize,多epoch;或者切割图像再检测。

  4. 半年前的我:训练的前一般着重调整哪些超参数啊?
    师姐:学习率就一般要根据loss的收敛程度来调
    其实最好是训练几十个epoch就降低一下学习率这样 (边训练边降低学习率),初始学习率和变化策略
    Batch_size根据卡来调整
    超参的设置也很重要 阅读论文的时候有时会更加偏向于学习里面的模型结构 忽略它写的超参配置 有时候我们训不出好的效果 不一定是模型结构出了问题 超参配置稍微不一样 就会有很大的不同
    然后就是要学会善用tensorboard看指标的变化

  5. 半年前的我:ssh传大文件大概40分钟才传过去?
    师姐:你现在配的是ipv4的地址 用ipv6会更快些 但是ipv6有时候能连接成功 有时候不行
    一般除了参数文件和训练数据 没有其他太大的文件了啊
    以后要是实在太大 你可以发给我 然后我下载了以后 用u盘传一下吧
    现在的我:请在同一局域网网段下传,快到起飞。

  6. 半年前的我:训练的时候要不要写shell脚本?还有就是常用数据集要不要处理好了全部都放在一个文件夹里(感觉会很整齐)
    师姐:是的。可以不写啊 直接运行.py就可以啊。对的 我就是有一个文件夹单独放数据集
    现在的我:写个shell吧,不然做实验麻烦得要死,猪头脑子记性又不好。

  7. 半年前的我:目标检测的指标吗
    师姐:显著性对象检测 指标就有pr曲线 然后f值 mae值啥的。测这些指标 在github上是能找到代码的 有matlab版也有python版的。你可以看看相关论文 它们都测了哪些 然后在github上找。不是 就是和准确率差不多 用这些去衡量你的结果好不好。目标检测的话应该是 IoU PR AP mAP这些吧 对 这些指标就是衡量你方法好坏的标准。测这些值 在github上肯定有写好的代码的。

尾声

555,导师和师姐好耐心,我的那种蠢蛋问题也耐心的回复。
希望也能帮助到更多的小小白,早日成为炼丹小白。

  • 31
    点赞
  • 95
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 3
    评论
YOLOv5是目标检测算法中的一种,通过训练自己的数据集可以实现对自定义目标的检测。下面是YOLOv5入门实践的步骤。 第一步,准备数据集。首先收集一些与你想要检测的目标相关的图像,确保图像中的目标已标注好边界框坐标。将图像和对应的标注文件放在数据集文件夹中。 第步,配置运行环境。需要在计算机上安装Python环境,并根据YOLOv5的要求安装相应的库和依赖。可以通过pip或conda进行安装。 第三步,调整配置文件。YOLOv5提供了一个默认的配置文件,可以根据自己的需求进行修改。主要需要调整的是类别数量和路径配置。 第四步,划分训练集和验证集。将数据集中的图像和标注文件划分为训练集和验证集,一般可以按照70%的比例划分。 第五步,训练模型。在终端中运行训练命令,指定相关参数,如模型类型、数据集路径、训练集和验证集路径等。训练时可以选择使用预训练权重或从头开始。 第六步,评估模型。训练完成后,可以通过评估命令对模型进行评估,得到关于模型性能的指标,如精确度、召回率等。 第七步,使用模型进行目标检测。训练完成的模型可以用于检测自定义数据集中的目标。可以在终端中运行检测命令,指定相关参数,如模型路径、检测图像路径等。 通过以上步骤,我们可以进行YOLOv5的入门实践,训练自己的数据集,并使用训练好的模型进行目标检测。随着更多的实践和调优,可以提高模型性能和检测效果

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Vanessa Ni

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值