当前位置:首页 > 神经网络 > 正文

yolov5是卷积神经网络吗

本文目录一览使用YOLOv5进行目标检测

在这篇文章中,我们将使用来自AWS上(可定)COCO数据集的图像来设置和运行YOLO。

一般来说,分类技术对于自动驾驶汽车来说并不是很有帮助,因为它们只能预测图像中的物体,而不能提供图像的位置。 物体检测在自动驾驶汽车中非常重要,它可以检测场景中的物体及其位置。 YOLO(YouOnlyLookOnce)是由JosephRedmon等创建的高速实时目标检测算法。 YOLO使用卷积神经网络(CNN)来解决概率回归问题。 然后进行了一些修改。 为了进行预测,YOLO只需要一条通过CNN的前向路径。 它输出一个具有相应边界框的对象。 它广泛用于自动驾驶汽车中检测场景中的物体。


第一步:设置账户(此步骤可跳过)

登录wandb.ai网站,复以下:

来自wandb.ai/authorize的API密钥
wandb.ai/settings中的团队名称。 默认团队名称是用户ID。

第二步:创建AWS实例(如果是本地训练,可以跳过此步骤)

创建实例时选择“LearningInAMI(Ubuntu18.04)Version47.0”—ami-01f1096e6659d38fa"AMI,因为它具有深度学习所需的库。 如果我们搜索“实例类型p3”的“AWSMachineImage(AMI)”,就可以找到这个。 .2xlarge(V100)就足够了。 为了节成本,在“配置实例”步骤下选择Spot实例

步骤3:安装依赖项

登录AWS实例后,使用以下命令创建conda环境并设置Weights&Bias。 环境变量:

第4步:训练、验证和测试

第5步:检查指标

验证实际标签集

标签预测集验证

训练损失

测试

以上所有结果将存储在yolov5runsdetectexp文件夹中

文献阅读:SSDA-YOLO:Semi-supervisedDomainAdaptiveYOLOforCross-DomainObjectDetectionSSDA-YOLO:一种基于半监督域自适应的YOLO方法,旨在解决跨域目标检测的挑战。 该方法结合了YOLOv5和域自适应,利用知识蒸馏框架和MeanTeacher模型帮助学生模型获取未标记目标域的实例级特征。 此外,通过场景样式迁移生成伪图像以补偿图像平面中的差异,并且跨域对齐预测的直观一致性认为会丢失。 SSDA-YOLO为资源受限的应用场景,尤其是自动驾驶、动作识别等实时性要求较高的场景提供了有效的解决方。
现有的基于卷积神经网络的目标检测方法在大规模基准测试上取得了显着进展,但在截然不同的目标域场景中其性能却显着下降。 SSDA-YOLO是为了解决这个问题而提出的,使用半监督学习方法,可以在数据和标注资源有限的情况下提高跨域的检测性能。 作为基础目标检测器,YOLOv5以其实时性和高精度而受到青睐,而传统的基于FasterR-CNN的方法在跨域目标检测域占据主导地位。 SSDA-YOLO结合了YOLOv5、知识蒸馏框架和MeanTeacher模型的单阶段优势,实现了未标记目标域的实例级特征学习,从而提高了检测性能。
SSDA-YOLO的整体架构包括知识蒸馏框架、生成伪训练图像、消除域之间的差异和一致性损失函数。 知识蒸馏框架通过过滤预测来强相对无偏差的更新,帮助学生模型迭代地生成强大的伪标签。 SSDA-YOLO使用CUT生成类源域假目标图像和类目标假源图像,能够解决图像级别的域差异,并通过过滤预测进行弱监督学习。 最终,一致性损失函数确保了学生模型中不同域输入的输出的一致性,通过计算最终输出之间的L2距离来实现这一目标。
实验部分表明,当以PascalVOC2007和2012数据集为源域、以Clipart1k数据集为源时,SSDA-YOLO通过标记源训练集和未标记的目标训练图像进行自适应域。 目标字段。 验证集用于评估SSDA-YOLO在目标测试集上的性能,证明其在跨域目标检测任务中的有效性。

yolov5目标检测神经网络——损失函数计算原理本文主要深入分析yolov5目标检测神经网络损失函数的计算原理,涉及精度评估、网络结构以及损失函数的组成。 目标检测任务的关键在于位置、大小和识别的准确性。 Yolov5构建了完整的损失函数,包括矩形框损失(lossrect)、置信度损失(lossobj)和分类损失(lossclc),预测每个网格的矩形框、置信度和分类概率。
损失函数的权重设置反映了不同任务的重要性。 例如,置信度损失通常是最大的,其次是矩形框和分类损失。 Yolov5使用CIOUloss来计算矩形框损失和BCEloss以实现置信度和分类损失。 mask矩阵用于确定每个网格的损失计算,仅计算预期目标的网格。
专门对矩形帧损失,yolov5使用了IOU系列损失函数,如IOU、GIOU、DIOU和CIOU。 这些损失函数考虑了矩形框的重叠和几何关系,以更准确地测量预测框和实际框之间的对应程度。 置信度损失使用CIOU作为标签,而不是简单的0-1赋值,以反映预测框与目标之间的真实关系。
分类损失计算基于one-hot码标签的BCE损失,通过平滑作降低过拟合的风险。 综上所述,yolov5损失函数的设计旨在优化目标检测任务的多方面性能,提高模型的准确性和鲁棒性。
最后,作者表示,由于个生活的变化,文章更新有所延迟,但他承诺会在空闲时间继续学习和分享知识,并祝读者新年快乐。