yolov8代码梳理 训练自己的数据 最终版

1.总结一下

  1. 最开始为了检测不规则的麻包袋,所以采用了目标检测。yolov3,fasterrcnn,ssd。这种矩形框还是可以用。后面检测的物体变成了规则的纸箱,我们还用目标检测发现,没有旋转角度,因为箱子的摆放不是正的。只能通过opencv的minarea去找到最小矩形框去寻找角度。但是opencv的方式首先对物体要和背景颜色区分,其次不够优美,毕竟算是2步走。
  2. 后面又尝试多训练一个角度,也就是把角度分成180个类别去分类,这个方式及其不稳定,也可能是我代码写的不好,但是后面发现,有人在做旋转矩形检测
  3. 旋转矩形框检测, 说实话,我把杨雪提供的框架跑了一遍,效果很差,角度的回归就像没用一样,反正很不如意。
  4. 实在没辙,只好采用了实力分割,maskrcnn。这个是真好用,大概标定50个样本,训练100epoch效果就很棒了。我们一直采用这种方式大概用了2年。
  5. 但是从我测试了rotated_rtmdet后。我真的觉得醍醐灌顶的爽,一直采用实力分割去做箱子的识别,总让我感觉很别扭。现在终于有一个可以落地的旋转目标检测。我就觉得公司的3D拆垛已经终于走上正轨了。
  6. 后面yolo又推出了,v4v5v6v7 其中还有ppyolo,yolox,yoloe , 包括现在的v8。我也不断的尝试更新算法。但是说实话目标检测真的算是走到头了。当然也是我个人的最终版。因为公司原因,个人将不在做视觉类的项目了,所以视觉类的目标检测和实力分割或者分类也就不在继续研究了。

2.yolov8代码梳理

参考
在这里插入图片描述

说实话,我不太懂他们改变的原因,看代码而言并不困难。基本和v5差不多。而且解耦检测头也是和yolox一样的。
总体来说,这个模型的精度和速度是对做工程和做应用的非常友好,我用自己的数据集,训练50个epoch,没有修改任何东西,map都跑到了99.5.非常惊艳。

3. 训练自己的数据集

  1. 数据集还是yolo格式。如果是labelme标注的json格式,参考 修改json标注为txt文件。或者就直接采用labelImg标注,生成yolo格式的标注文件参考
  2. 修改数据配置文件,也就是复制一份coco.yaml 。修改路径为自己的数据集路径

在这里插入图片描述
文件放在G:\sick\SH_visionary-s\ultralytics\ultralytics\yolo\v8\detect\coco.yaml

  1. 下载yolov8n.pt 放在和train.py同目录下。
    在这里插入图片描述

  2. 修改训练参数,路径在G:\sick\SH_visionary-s\ultralytics\ultralytics\yolo\cfg\default.yaml ,
    在这里插入图片描述
    我就修改了这些

  3. 执行train.py的文件就好

  • 6
    点赞
  • 30
    收藏
    觉得还不错? 一键收藏
  • 7
    评论
YOLOv8是一种目标检测算法,它是YOLO(You Only Look Once)系列算法的最新版本。如果你想要训练自己的数据集,可以按照以下步骤进行操作: 1. 数据准备:首先,你需要准备自己的数据集。数据集应包含标注好的图像和相应的标签。标签通常以XML或JSON格式存储,并包含每个目标的类别、边界框坐标等信息。 2. 数据预处理:在训练之前,你需要对数据进行预处理。这包括将图像调整为统一的大小、归一化像素值等操作。此外,还可以进行数据增强操作,如随机裁剪、旋转、翻转等,以增加数据的多样性。 3. 配置文件:YOLOv8训练过程需要一个配置文件来指定网络结构、超参数等信息。你可以根据自己的需求创建一个配置文件,并设置相关参数,如类别数、输入图像大小、学习率、批大小等。 4. 模型训练:使用准备好的数据集和配置文件,可以开始进行模型训练了。你可以使用开源的YOLOv8代码库,如Darknet或PyTorch实现的YOLOv5,根据其提供的训练脚本进行训练。在训练过程中,模型会根据输入的数据进行前向传播和反向传播,不断更新网络参数以提高检测性能。 5. 模型评估:在训练完成后,你可以使用测试集对训练好的模型进行评估。评估指标通常包括精确率、召回率、平均精确率均值(mAP)等,用于衡量模型的性能。 6. 模型应用:训练好的YOLOv8模型可以用于目标检测任务。你可以使用该模型对新的图像或视频进行目标检测,并获取目标的类别和位置信息。 希望以上步骤对你有所帮助!

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值