关键词:
Focaler-EIoU
YOLOv8
水下目标检测
轻量化网络
PConv
摘要:
基于机器视觉的无人水下航行器(Unmanned Underwater Vehicle, UUV)在工作时往往面临着嵌入式设备计算资源有限、实时检测速度慢的问题,为了解决这些问题,设计了一种轻量化网络检测算法——YOLOv8-FasterECA-Slim-neck-Focaler-EIoU(YOLOv8-FESF)。在骨干网络中,基于FasterNet Block和高效通道注意力(Efficient Channel Attention, ECA)机制构建全新的C2f_Faster_ECA模块,降低特征网络的参数量和计算量,采用Slim-neck作为颈部结构,进一步压缩模型的规模;重新设计检测头,利用参数共享的思想合并特征提取模块,从而轻量化模型,提高检测速度;使用边框回归损失函数Focaler-EIoU动态调整损失值,解决难易样本不平衡的问题,以提高检测精度。实验结果证明,所提模型在RUOD数据集上表现良好,相较于YOLOv8n基准模型,参数量和计算量分别减少40%和54%,检测速度提高10.5帧/秒,平均精度均值(mean Average Precision, mAP)仅下降0.1%,适合部署在计算设备资源受限的水下目标检测平台。