Mar, 2024

Fostc3net: 一个基于网络结构优化的轻量级 YOLOv5

TL;DR本文介绍了一种针对移动设备的增强轻量级 YOLOv5 技术,用于识别与输电线相关的物体。该方法通过引入 C3Ghost 模块和 FasterNet 模块改进了 YOLOv5 的性能,并采用 wIoU v3 LOSS 损失函数解决了数据集中简单和挑战样本的不平衡问题。实验证明,与现有的 YOLOv5 相比,该模型在检测准确性上提高了 1%,降低了 13% 的 FLOPs,模型参数减少了 26%。而在消融实验中,Fastnet 模块和 CSghost 模块提高了原始 YOLOv5 基准模型的精度,但导致了 [email protected] 度量的下降,而 wIoUv3 损失函数的改进显著减轻了该度量的下降。