yolov8工程实践

1、yolov8语义分割的模型输出信息
输入为800*800*3
输入为640*640*3
以13800*800输入为例,模型最后有两个输出output0和output1:
[1,37,13125]
1 - batch size
37 - the first 5 values “x, y, width, height and probability” of the detected object, next 32 values are the probability for the mask
13125 - the maximum number of possible detected objects(阈值判断,检测框/掩膜计算,非极大值抑制)

[1,32,200,200] array of 32 masks measuring 200 by 200 pixels
1 - batch size
32 - index of the mask
200 - x points
200 - y points

因此,要获得检测到的对象的输出掩码,您需要首先从第一个输出中选择最佳概率并记住对象掩码的索引。
然后根据掩膜索引从第二个输出 [1,mask of index,200,200] 获取对象掩模 200x200 像素,并将 x 点和 y 点插值到输入图像乘以 4(800 / 200 = 4)。

1、    results = model.track(source="F:/A_bubble/bubble.mp4",
                          show=False,
                          save=True,
                          save_dir=save_path,
                          save_txt=False,
                          line_width=1,
                          stream=True,
                          retina_masks=True)
      # 如果要自定义推理结果的保存路径,需要用save_dir参数指定;但是该参数使用前,必须在D:\workspace\ultralytics-main\ultralytics\cfg\default.yaml配置文件中加入对应的参数名:save_dir:    # peanut add
  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Yolov5 是一种基于深度学习的目标检测算法,它是由Ultralytics公司开发的一种轻量级目标检测模型,并且在实时性能和准确性方面表现出色。下面是 Yolov5 工程代码的一些详解: 1. 数据准备:在使用 Yolov5 进行目标检测之前,需要准备好训练和测试数据。数据需要按照一定的格式进行组织,通常是将图片和对应的标注信息放在同一个文件夹中,并生成对应的标签文件。 2. 模型定义:Yolov5 的模型定义主要包括网络结构、损失函数和评估指标等。在 Yolov5 的代码中,网络结构使用了一种叫做CSPDarknet53的骨干网络,并在其上加上了多个特征金字塔层(PANet),用于提取不同尺度的特征。损失函数方面,Yolov5 使用了一种称为YOLOv5 Loss的多任务损失函数,该损失函数包括了目标分类损失、边界框回归损失和目标置信度损失。 3. 数据加载与增强:在 Yolov5 中,数据加载与增强是通过 PyTorch 的数据加载器(DataLoader)和数据增强库(Albumentations)来完成的。数据加载器用于从硬盘上加载数据,并进行一定的预处理操作,如缩放、裁剪等。数据增强库则用于对加载的数据进行一系列的增强操作,如随机翻转、旋转、亮度调整等。 4. 训练过程:Yolov5 的训练过程主要包括模型的初始化、数据加载与增强、前向传播、损失计算、反向传播和参数更新等。训练过程中,模型会根据给定的训练数据进行一定的迭代训练,以优化模型参数,使其能够更好地适应目标检测任务。 5. 推理过程:Yolov5 的推理过程主要是将训练好的模型应用到新的测试数据上,以实现目标检测的功能。推理过程中,模型会将输入图像传入网络中进行前向传播,然后根据网络输出的结果进行后处理,如非极大值抑制(NMS)等,最终得到目标检测结果。 这些是 Yolov5 工程代码的一些主要部分,通过对这些部分的理解和实践,可以更深入地了解 Yolov5 的工作原理和使用方法。当然,这只是一个简单的概述,具体的代码实现还需要参考 Yolov5 的官方代码和文档。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值