欢迎光临
我们一直在努力

YOLO系列算法改进 | Neck改进篇 | 融合EFC层间特征相关性模块 | 通过多尺度特征交互减少冗余信息丢失即插即用

0. 前言

本文介绍了增强层间特征相关性(Enhanced Interlayer Feature Correlation,EFC),并将其集成到ultralytics最新发布的YOLO26目标检测算法中,改进Neck特征融合部分。检测无人机图像小物体具有挑战性,传统多尺度特征融合方法存在不足。EFC 模块通过引入分组的特征聚焦单元(GFF)增强特征关联性,利用多级特征重构模块(MFR)进行特征重构,减少冗余特征生成,且具有通用性和适应性。我们将 EFC 模块集成进 YOLO26,替换部分特征融合策略。该方法即插即用,有望提升 YOLO26在小物体检测任务中的性能。

专栏链接:YOLO系列算法改进专栏链接

目录

0. 前言

1. EFC增强层间特征相关性模块简介

2. EFC模块基本原理与创新点

🧠 EFC模块的基本原理

🎯 PGI 的主要创新点

3. 具体改进步骤

🍀🍀步骤1:创建EFC.py文件

🍀🍀步骤2:tasks.py文件修改

🍀🍀步骤3:创建YAML配置文件

🍀🍀步骤4:新建train.py文件训练模型

🍀🍀步骤5:模型结构打印结果


1. EFC增强层间特征相关性模块简介

鉴于无人机图像存在低分辨率以及背景混合的情况,检测其中的小物体颇具挑战性,这会造成特征信息有限。多尺度特征融合能够通过获取不同尺度上的信息来增强检测能力,然而传统方法存在一定缺陷。简单的连接或加法运算无法充分发挥多尺度融合的优势,致使特征之间的相关性不足。这一缺陷对小物体的检测造成了阻碍,尤其是在复杂背景和人口稠密地区。为解决该问题并有效利用有限的计算资源,本文提出一种基于增强层间特征相关性(EFC)的轻量级融合策略,用以替代传统特征金字塔网络(FPN)中的特征融合策略。特征金字塔中不同图层的语义表达存在不一致性。在EFC策略里,分组的特征聚焦单元(GFF)通过聚焦不同特征的上下文信息来增强各层的特征相关性。多级特征重构模块(MFR)对金字塔中各层的强弱信息进行有效的重构和变换,减少冗余的特征融合,从而保留更多的小目标信息。值得一提的是,所提出的方法具有即插即用的特性,能够广泛应用于各种基础网络。

原始论文:

赞(0)
未经允许不得转载:171主机测评 » YOLO系列算法改进 | Neck改进篇 | 融合EFC层间特征相关性模块 | 通过多尺度特征交互减少冗余信息丢失即插即用
分享到: 更多 (0)

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址