做解析视频网站怎么赚钱,多种网站建设,做网站开发很赚钱吗,如何做一个网站设计原文链接#xff1a;
1、 详解YOLOv8网络结构/环境搭建/数据集获取/训练/推理/验证/导出
2、Yolov8的详解与实战
3、YOLOV8模型训练部署#xff08;实战#xff09;#xff08;#xff09;有具体部署和训练实现代码YOLOV8模型训练部署#xff08;实战#xff09;
1、 详解YOLOv8网络结构/环境搭建/数据集获取/训练/推理/验证/导出
2、Yolov8的详解与实战
3、YOLOV8模型训练部署实战有具体部署和训练实现代码YOLOV8模型训练部署实战
4、官方平台介绍Ultralytics Inc.
5、YOLOv8改进有效系列目录 | 包含卷积、主干、检测头、注意力机制、Neck上百种创新机制有非常好的参考和启示价值
6、源码github地址: GitHub - ultralytics/ultralytics: NEW - YOLOv8 in PyTorch ONNX OpenVINO CoreML TFLite
1、导读
YOLOv8 是 ultralytics 公司在 2023 年 1月 10 号开源的基于YOLOV5进行更新的 下一个重大更新版本目前支持图像分类、物体检测和实例分割任务鉴于Yolov5的良好表现Yolov8在还没有开源时就收到了用户的广泛关注。其主要结构如下图所示 2、YOLOv8相对于Yolov5的核心改动
从YOLOv8的网络结构可以看出,其延用了YOLOv5的网络结构思想网络还是分为三个部分 基于CSP紧凑和分离的主干网络backbone特征增强网络(neck)检测头(head) 三个部分。
改进总结
Backbone的改进YOLOV5中的C3模块被替换成了C2f模块实现了进一步的轻量化。同时保持了CSP的思想保留了YOLOv5等架构中使用的SPPF空间金字塔池化模块。特征增强网络(neck) YOLOv8使用PA-FPN路径聚合网络-特征金字塔网络的思想具体实施过程中将YOLOV5中的PA-FPN上采样阶段的卷积去除了并且将其中的C3模块替换为了C2f模块。检测头(head) 区别于YOLOV5的耦合头YOLOV8使用了Decoupled-Head使得网络的训练和推理更加高效。其它更新部分 Anchor-Free的思想抛弃了Anchor-Base采用了Anchor-Free的思想。损失函数的改进分类使用BCEloss回归使用DFL LossCIOU Loss。样本匹配方式的改进采用了Task-Aligned Assigner匹配方式。
这些改进使得YOLOv8在保持了YOLOv5网络结构的优点的同时进行了更加精细的调整和优化提高了模型在不同场景下的性能。
yolov8是个模型簇从小到大包括yolov8n、yolov8s、yolov8m、yolov8l、yolov8x等。模型参数、运行速度、参数量等详见下表 对比yolov5如下表 mAP和参数量都上升了不少。
3、YOLOv8的网络结构详解
Backbone它采用了一系列卷积和反卷积层来提取特征同时也使用了残差连接和瓶颈结构来减小网络的大小和提高性能。该部分采用了C2f模块作为基本构成单元与YOLOv5的C3模块相比C2f模块具有更少的参数量和更优秀的特征提取能力。Neck它采用了多尺度特征融合技术将来自Backbone的不同阶段的特征图进行融合以增强特征表示能力。具体来说YOLOv8的Neck部分包括一个SPPF模块、一个PAA模块和两个PAN模块。Head它负责最终的目标检测和分类任务包括一个检测头和一个分类头。检测头包含一系列卷积层和反卷积层用于生成检测结果分类头则采用全局平均池化来对每个特征图进行分类。
3.1 Backbone
由最上面的YOLOv8网络结构图我们可以看出在其中的Backbone部分由5个卷积模块和4个C2f模块和一个SPPF模块组成。 (其中浅蓝色为卷积模块,黄色为C2f模块深蓝色为SPPF模块 )
如果上图看的不够直观,我们来看一下YOLOv8的文件中的yaml文件,看一下它backbone部分的结构组成部分会更加直观。
backbone:# [from, repeats, module, args]- [-1, 1, Conv, [64, 3, 2]] # 0-P1/2- [-1, 1, Conv, [128, 3, 2]] # 1-P2/4- [-1, 3, C2f, [128, True]]- [-1, 1, Conv, [256, 3, 2]] # 3-P3/8- [-1, 6, C2f, [256, True]]- [-1, 1, Conv, [512, 3, 2]] # 5-P4/16- [-1, 6, C2f, [512, True]]- [-1, 1, Conv, [1024, 3, 2]] # 7-P5/32- [-1, 3, C2f, [1024, True]]- [-1, 1, SPPF, [1024, 5]] # 9
上面的部分就是YOLOv8的yaml文件的Backbone部分可以看到其由5个Conv模块四个C2f模块以及一个SPPF模块组成下面我们来根据每个模块的组成来进行讲解。
3.1.1 卷积模块(Conv)
在其中卷积模块的结构主要为下图 在其中主要结构为一个2D的卷积一个BatchNorm2d和一个SiLU激活函数整个卷积模块的作用为
下采样每个卷积模块中的卷积层都采用步长为2的卷积核进行下采样操作以减小特征图的尺寸并增加通道数。非线性表示每个卷积层之后都添加了Batch Normalization批标准化层和ReLU激活函数以增强模型的非线性表示能力。
在其中Batch Normalization批标准化是深度学习中常用的一种技术用于加速神经网络的训练。Batch Normalization通过对每个小批量数据进行标准化使得神经网络在训练过程中更加稳定可以使用更高的学习率并且减少了对初始化权重的依赖。Batch Normalization的基本思想是对每个小批量数据进行标准化使得每个特征的均值为0方差为1然后再通过一个可学习的缩放因子和平移因子来调整数据的分布从而使得神经网络更容易训练。
3.1.2 C2f 模块
在YOLOv8的网络结构中C2f模块算是YOLOv8的一个较大的改变与YOLOv5的C3模块相比C2f模块具有更少的参数量和更优秀的特征提取能力。下图为C2f的内部网络结构图。 在C2f模块中我们可以看到输入首先经过一个k1s1p0cc_out的卷积模块进行了处理然后经过一个split处理(在这里split和后面的concat的组成其实就是所谓的残差模块处理)经过数量为n的DarknetBottleneck模块处理以后将残差模块和主干模块的结果进行Concat拼接在经过一个卷积模块处理进行输出。 在其中提到的残差连接residual connections是一种用于构建深层神经网络的技术。它的核心思想是通过跳过层级连接来传递残差或误差。 在传统的神经网络中信息流通过一层层的网络层每一层都通过非线性激活函数进行转换和提取特征。然而随着神经网络的加深可能会出现梯度消失或梯度爆炸的问题导致网络收敛困难或性能下降。 残差连接通过引入跨层级的连接将输入的原始信息直接传递到后续层级以解决梯度消失和爆炸问题。具体而言它将网络的输入与中间层的输出相加形成了一个捷径或跳跃连接从而允许梯度更容易地传播。 数学上假设我们有一个输入x通过多个网络层进行处理后得到预测值H(x)。那么残差连接的表达式为 F(x) H(x) x 。其中F(x)为残差块的输出H(x)为经过一系列网络层处理后得到的特征表示x为输入直接连接到残差块中的跳跃连接。 通过残差连接网络可以更容易地学习残差或误差从而使网络更深层次的特征表达更准确。这对于训练深层神经网络非常有用可以提高网络的性能和收敛速度。 在C2f模块中用到的DarknetBottleneck模块其中使用多个3x3卷积核进行卷积操作提取特征信息。同时其具有add是否进行残差链接的选项。 其实整个C2f模块就是一个改良版本的Darknet
首先使用1x1卷积核将输入通道数减少到原来的1/2以减少计算量和内存消耗。然后使用多个3x3卷积核进行卷积操作提取特征信息。接着使用残差链接将输入直接加到输出中从而形成了一条跨层连接。接着再次使用1x1卷积核恢复特征图的通道数。
3.1.3 SPPF模块
YOLOv8的SPPF模块相对于YOLOv5的SPPF模块并没有任何的改变。 在传统的卷积神经网络中通常使用全连接层来将卷积和池化层提取的特征映射到固定长度的向量上以便进行分类或检测。然而全连接层需要固定大小的输入(例如640*640YOLOV5、32*32(LenNet)、224*224ImageNet)这使得网络无法处理尺寸不同的输入图像。因此SPP-Net引入了空间金字塔池化SPPSpatial Pyramid Pooling模块用于解决输入图像尺寸不同的问题并且可以在不同尺寸的输入图像上实现准确的分类和检测。 在SPP-Net中输入图像首先通过一个卷积神经网络进行特征提取。然后在SPP层中进行池化操作将卷积层的输出特征图Feature Map分成不同尺度的网格对于每个网格都对其内部进行池化操作池化输出的结果拼接成一个固定长度的特征向量。这个特征向量可以用于后续的分类任务下面的图是介绍从一张输入图上提取提特征图然后产生固定长度的特征向量然后输入到全连接层中进行分类的流程图。 下面将通过一个简单的例子来说明SPP-Net是如何解决选择性搜索问题的是如何将不同大小的特征图转换为固定长度的特征向量的。假设我们有一个输入图像大小为300✖300像素由于我们希望网络能够处理不同大小的输入图像因此我们先对输入图像进行卷积操作得到一个特征图大小为15✖ 15✖ 1024其中分别表示一个特征图的宽度、高度和每个位置上的特征向量的维数。具体来说在这个例子中15✖ 15表示特征图的宽度和高度1024表示每个位置上的特征向量的维数。然后我们使用SPP层对特征图进行池化操作将其分成1✖1、2✖2和4✖4三个尺度的网格并对每个网格内部进行池化操作。最后对于每个尺度将池化输出的结果拼接成一个固定长度的特征向量。假设我们选择的池化操作是最大池化每个网格内部的池化操作输出的结果是一个特征向量我们将这些特征向量拼接起来得到一个长度为1024✖(149)15360的特征向量。这个特征向量可以作为分类器的输入用于分类任务。
下面这个图诠释了上面的例子方框内的部分即SPP部分,其分成1✖1、2✖2和4✖4三个尺度的网格进行特征向量的提取然后进行拼接固定长度的特征向量并将其输入到全连接层中用于分类任务,需要注意的是SPP通常在一个网络结构的最后一层,输出结果直接输入给检测头进行检测,或者输入给Neck部分进行特征向量的拼接操作以将不同尺度下的特征向量进行拼接提高检测和分类效率。 SPP-Net的优点包括
对任意大小的输入图像进行处理具有较好的泛化能力和适用性可以用于多种计算机视觉任务包括图像分类、物体检测、场景分类等可以减少网络训练和测试时间提高网络的效率在多个数据集和比赛中取得了不错的表现证明了其有效性和通用性。
3.2 Neck
YOLOv8的Neck部分是该模型中的一个关键组件它在特征提取和融合方面起着重要作用。 YOLOv8的Neck部分依然采用PAN-FPN的思想下图的abc为一个Neck部分的流程示意图。 整个Neck部分的步骤如下将特征提取网络(Backbone)的输出P3P4P5输入进PAN-FPN网络结构使得多个尺度的特征图进行融合将P5经过上采样与P4进行融合得到F1将F1经过C2f层和一次上采样与P3进行融合得到T1将T1经过一次卷积层与F1经过融合得到F2将F2经过一次C2f层得到T2将T2经过一次卷积层与P5融合得到F3将F3经过一次C2f层得到T3最终得到T1、T2、T3就是整个Neck的产物
上述过程可以描述为下图在图片上做了一些标准方便理解。 上述的过程可以在代码部分看到,我们同样看YOLOv8的yaml文件能够更直观的看到这个步骤,大家可以看代码同时对应图片来进行分析: head: - [-1, 1, nn.Upsample, [None, 2, nearest]] - [[-1, 6], 1, Concat, [1]] # cat backbone P4 - [-1, 3, C2f, [512]] # 12 - [-1, 1, nn.Upsample, [None, 2, nearest]] - [[-1, 4], 1, Concat, [1]] # cat backbone P3 - [-1, 3, C2f, [256]] # 15 (P3/8-small) - [-1, 1, Conv, [256, 3, 2]] - [[-1, 12], 1, Concat, [1]] # cat head P4 - [-1, 3, C2f, [512]] # 18 (P4/16-medium) - [-1, 1, Conv, [512, 3, 2]] - [[-1, 9], 1, Concat, [1]] # cat head P5 - [-1, 3, C2f, [1024]] # 21 (P5/32-large) Neck部分的整体功能的详细分析如下:
1. Neck的作用 Neck部分在YOLOv8模型中负责对来自Backbone的特征进行进一步处理和融合以提高目标检测的准确性和鲁棒性。它通过引入不同的结构和技术将多尺度的特征图进行融合以便更好地捕捉不同尺度目标的信息。2. 特征金字塔网络Feature Pyramid Network, FPN YOLOv8的Neck部分通常采用特征金字塔网络结构用于处理来自Backbone的多尺度特征图。FPN通过在不同层级上建立特征金字塔使得模型能够在不同尺度上进行目标检测。它通过上采样和下采样操作将低层级的细节特征与高层级的语义特征进行融合以获取更全面和丰富的特征表示。3. 特征融合Feature Fusion Neck部分还包括特征融合的操作用于将来自不同层级的特征进行融合。这种特征融合有助于提高模型对目标的检测准确性尤其是对于不同尺度的目标。4. 上采样和下采样 Neck部分通常会使用上采样和下采样操作以调整特征图的尺度和分辨率。上采样操作可以将低分辨率的特征图放大到与高分辨率特征图相同的尺寸以保留更多的细节信息。而下采样操作则可以将高分辨率的特征图降低尺寸以减少计算量和内存消耗。
YOLOv8的Neck部分通过特征金字塔网络和特征融合等操作有效地提取和融合多尺度的特征从而提高了目标检测的性能和鲁棒性。这使得模型能够更好地适应不同尺度和大小的目标并在复杂场景下取得更准确的检测结果。 PAN-FPN具有特征金字塔网络的路径聚合网络是一种用于计算机视觉中对象检测的神经网络架构。它将特征金字塔网络FPN与路径聚合网络PAN相结合以提高目标检测的准确性和效率。 FPN 用于从不同比例的图像中提取特征而 PAN 用于跨网络的不同层聚合这些特征。这允许网络检测不同大小和分辨率的对象并处理具有多个对象的复杂场景。 3.3 Head
如果Backbone和Neck部分可以理解为准备工作那么Head部分就是收获的部分经过前面的准备工作我们得到了Neck部分的输出T1、T2、T3分别代表不同层级的特征图Head部分就是对这三个特征图进行处理以产生模型的的输出结果的一个过程。
YOLOv8的Head部分我们先来看一下它的网络结构。 可以看到在YOLOv8的Head部分体现了最核心的改动—解耦头(Decoupled-Head)顾名思义就是将原先的一个检测头分解成两个部分。
在Head部分的三个解耦头分别对应着Neck部分的特征图输出T1、T2、T3。、
解耦头的工作流程是 将网络得到的特征图T1T2T3分别输入解耦头头进行预测检测头的结构如下图所示其中包含4个3×3卷积与2个1×1卷积同时在检测头的回归分支中添加WIOU损失函数如图4所示回归头部需要计算预测框与真实框之间的位置偏移量然后将偏移量送入回归头部进行损失计算然后输出一个四维向量分别表示目标框的左上角坐标x、y和右下角坐标x、y。分类头部针对于每个Anchor Free提取的候选框对其进行RoI Pooling和卷积操作得到一个分类器输出张量每个位置上的值表示该候选框属于每个类别的概率在最后通过极大值抑制方式筛选出最终的检测结果 我们再从YOLOv8的yaml文件来看Head部分的作用 可以看到检测头部分的输出为15,1821分别对应着Neck部分的三个输出特征图。
到此YOLOv8的网络结构部分讲解就已经完成下面我们来看如何利用YOLOv8进行训练操作。
四、环境搭建
在配置好环境之后在之后模型获取完成之后我们可以进行配置的安装我们可以在命令行下输入如下命令进行环境的配置。 pip install -r requirements.txt 输入如上命令之后我们就可以看到命令行在安装模型所需的库了。
五、数据集获取 在我们开始训练之前我们需要一份数据集,如何获取一个YOLOv8的数据集大家可以看原作者的的另一篇博客 从YOLO官方指定的数据集网站Roboflow下载数据模型训练
超详细教程YoloV8官方推荐免费数据集网站Roboflow一键导出Voc、COCO、Yolo、Csv等格式
在上面随便下载了一个 数据集用它导出yolov8的数据集以及自动给转换成txt的格式yaml文件也已经配置好了我们直接用就可以。 六、模型获取
到这里假设你已经搭建好了环境和有了数据集那么我们就可以进行模型的下载因为yolov8目前还存在BUG并不稳定随时都有可能进行更新所以不推荐大家通过其它的途径下载最好通过下面的方式进行下载。
我们可以直接在终端命令下 (PS这里需要注意的是我们需要在你总项目文件目录下输入这个命令因为他会下载到当前目录下) pip install ultralytics 如果大家去github上直接下载zip文件到本地可能会遇到报错如下识别不了yolo命令所以推荐大家用这种方式下载 七、模型训练
我们来看一下主要的ultralytics目录结构 我门打开cfg目录下的default.yaml文件可以配置模型的参数
在其中和模型训练有关的参数及其解释如下:
参数名输入类型0taskstrYOLO模型的任务选择选择你是要进行检测、分类等操作1modestrYOLO模式的选择选择要进行训练、推理、输出、验证等操作2modelstr/optional模型的文件可以是官方的预训练模型也可以是训练自己模型的yaml文件3datastr/optional模型的地址可以是文件的地址也可以是配置好地址的yaml文件4epochsint训练的轮次将你的数据输入到模型里进行训练的次数5patienceint早停机制当你的模型精度没有改进了就提前停止训练6batchint我们输入的数据集会分解为多个子集一次向模型里输入多少个子集7imgszint/list输入的图片的大小可以是整数就代表图片尺寸为int*int或者list分别代表宽和高[wh]8savebool是否保存模型以及预测结果9save_periodint在训练过程中多少次保存一次模型文件,就是生成的pt文件10cachebool参数cache用于控制是否启用缓存机制。11deviceint/str/list/optionalGPU设备的选择cuda device0 or device0,1,2,3 or devicecpu12workersint工作的线程Windows系统一定要设置为0否则很可能会引起线程报错13namestr/optional模型保存的名字结果会保存到project/name 目录下14exist_okbool如果模型存在的时候是否进行覆盖操作15prepetrained bool 参数pretrained用于控制是否使用预训练模型。16optimizerstr优化器的选择choices[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto]17verbosebool用于控制在执行过程中是否输出详细的信息和日志。18seedint随机数种子模型中涉及到随机的时候根据随机数种子进行生成19deterministicbool用于控制是否启用确定性模式在确定性模式下算法的执行将变得可重复即相同的输入将产生相同的输出20single_clsbool是否是单标签训练21rectbool当 rect 设置为 True 时表示启用矩形训练或验证。矩形训练或验证是一种数据处理技术其中在训练或验证过程中输入数据会被调整为具有相同宽高比的矩形形状。22 cos_lr bool控制是否使用余弦学习率调度器23close_mosaicint控制在最后几个 epochs 中是否禁用马赛克数据增强24resumebool用于从先前的训练检查点checkpoint中恢复模型的训练。25ampbool用于控制是否进行自动混合精度26fractionfloat用于指定训练数据集的一部分进行训练的比例。默认值为 1.027profilebool用于控制是否在训练过程中启用 ONNX 和 TensorRT 的性能分析28freezeint/list/optinal用于指定在训练过程中冻结前 n 层或指定层索引的列表以防止它们的权重更新。这对于迁移学习或特定层的微调很有用。
7.1 训练的三种方式
7.1.1 方式一
我们可以通过命令直接进行训练在其中指定参数但是这样的方式我们每个参数都要在其中打出来。命令如下:
yolo taskdetect modetrain modelyolov8n.pt datadata.yaml batch16 epochs100 imgsz640 workers0 device0
需要注意的是如果你是Windows系统的电脑其中的Workers最好设置成0否则容易报线程的错误。
7.1.2 方式二推荐
通过指定cfg直接进行训练我们配置好ultralytics/cfg/default.yaml这个文件之后可以直接执行这个文件进行训练这样就不用在命令行输入其它的参数了。
yolo cfgultralytics/cfg/default.yaml
7.1.3 方式三
我们可以通过创建py文件来进行训练这样的好处就是不用在终端上打命令这也能省去一些工作量我们在根目录下创建一个名字为run.py的文件在其中输入代码
from ultralytics import YOLO
model YOLO(权重的地址)
data 文件的地址
model.train(datadata, epochs100, batch16)
无论通过上述的哪一种方式在控制台输出如下图片的内容就代表着开始训练成功了 八、模型验证/测试
参数名类型参数讲解1valbool用于控制是否在训练过程中进行验证/测试。2splitstr用于指定用于验证/测试的数据集划分。可以选择 val、test 或 train 中的一个作为验证/测试数据集3save_jsonbool用于控制是否将结果保存为 JSON 文件4save_hybirdbool用于控制是否保存标签和附加预测结果的混合版本5conffloat/optional用于设置检测时的目标置信度阈值6ioufloat用于设置非极大值抑制NMS的交并比IoU阈值。7max_detint用于设置每张图像的最大检测数。8halfbool用于控制是否使用半精度FP16进行推断。9dnnbool用于控制是否使用 OpenCV DNN 进行 ONNX 推断。10plotsbool用于控制在训练/验证过程中是否保存绘图结果。
验证我们划分的验证集/测试集的情况也就是评估我们训练出来的best.pt模型好与坏
yolo taskdetect modeval modelbest.pt datadata.yaml device0 九、模型推理
我们训练好自己的模型之后都会生成一个模型文件,保存在你设置的目录下,当我们再次想要实验该模型的效果之后就可以调用该模型进行推理了我们也可以用官方的预训练权重来进行推理。
推理的方式和训练一样我们这里就选一种来进行举例其它的两种方式都是一样的操作只是需要改一下其中的一些参数即可:
参数讲解
参数名 类型 参数讲解0sourcestr/optinal用于指定图像或视频的目录1showbool用于控制是否在可能的情况下显示结果2save_txtbool用于控制是否将结果保存为 .txt 文件3save_confbool用于控制是否在保存结果时包含置信度分数4save_cropbool用于控制是否将带有结果的裁剪图像保存下来5show_labelsbool用于控制在绘图结果中是否显示目标标签6show_confbool用于控制在绘图结果中是否显示目标置信度分数7vid_strideint/optional用于设置视频的帧率步长8stream_bufferbool用于控制是否缓冲所有流式帧True或返回最新的帧False9line_widthint/list[int]/optional用于设置边界框的线宽度如果缺失则自动设置10visualizebool用于控制是否可视化模型的特征11augmentbool用于控制是否对预测源应用图像增强12agnostic_nmsbool用于控制是否使用无关类别的非极大值抑制NMS13classesint/list[int]/optional用于按类别筛选结果14retina_masksbool用于控制是否使用高分辨率分割掩码15boxesbool用于控制是否在分割预测中显示边界框。 这里需要需要注意的是我们用模型进行推理的时候可以选择照片也可以选择一个视频的格式都可以。支持的视频格式有 MP4.mp4这是一种常见的视频文件格式通常具有较高的压缩率和良好的视频质量 AVI.avi这是一种较旧但仍广泛使用的视频文件格式。它通常具有较大的文件大小 MOV.mov这是一种常见的视频文件格式通常与苹果设备和QuickTime播放器相关 MKV.mkv这是一种开放的多媒体容器格式可以容纳多个视频、音频和字幕轨道 FLV.flv这是一种用于在线视频传输的流式视频文件格式 十、模型输出
当我们进行部署的时候可以进行文件导出然后在进行部署。
YOLOv8支持的输出格式有如下 1. ONNXOpen Neural Network ExchangeONNX 是一个开放的深度学习模型表示和转换的标准。它允许在不同的深度学习框架之间共享模型并支持跨平台部署。导出为 ONNX 格式的模型可以在支持 ONNX 的推理引擎中进行部署和推理。 2. TensorFlow SavedModelTensorFlow SavedModel 是 TensorFlow 框架的标准模型保存格式。它包含了模型的网络结构和参数可以方便地在 TensorFlow 的推理环境中加载和使用。 3. PyTorch JITJust-In-TimePyTorch JIT 是 PyTorch 的即时编译器可以将 PyTorch 模型导出为优化的 Torch 脚本或 Torch 脚本模型。这种格式可以在没有 PyTorch 环境的情况下进行推理并且具有更高的性能。 4. Caffe ModelCaffe 是一个流行的深度学习框架它使用自己的模型表示格式。导出为 Caffe 模型的文件可以在 Caffe 框架中进行部署和推理。 5. TFLiteTensorFlow LiteTFLite 是 TensorFlow 的移动和嵌入式设备推理框架支持在资源受限的设备上进行高效推理。模型可以导出为 TFLite 格式以便在移动设备或嵌入式系统中进行部署。 6. Core MLCore Machine LearningCore ML 是苹果的机器学习框架用于在 iOS 和 macOS 上进行推理。模型可以导出为 Core ML 格式以便在苹果设备上进行部署。 这些格式都提供了不同的优势和适用场景。选择合适的导出格式应该考虑到目标平台和部署环境的要求以及所使用的深度学习框架的支持情况。
模型输出的参数有如下
参数名类型参数解释0formatstr导出模型的格式1kerasbool表示是否使用Keras2optimizebool用于在导出TorchScript模型时进行优化以便在移动设备上获得更好的性能3int8bool用于在导出CoreML或TensorFlow模型时进行INT8量化4dynamicbool用于在导出CoreML或TensorFlow模型时进行INT8量化5simplifybool用于在导出ONNX模型时进行模型简化6opsetint/optional用于指定导出ONNX模型时的opset版本7workspaceint用于指定TensorRT模型的工作空间大小以GB为单位8nmsbool用于在导出CoreML模型时添加非极大值抑制NMS
命令行命令如下:
yolo taskdetect modeexport modelbest.pt formatonnx
到此为止本文的讲解就结束了,希望对大家对于YOLOv8模型理解有帮助
