YOLOv10改进策略【Neck】| AAAI 2025 颜色增强模块CEM 双色引导保真色 + 轻量调制提对比度,强化雾天检测,适配多模态图像信息融合
2026/4/22 23:31:21 网站建设 项目流程

一、本文介绍

本文记录的是利用CEM模块改进 YOLOv10 的多尺度特征融合

CEM(Color Enhancement Module)通过色度引导与RGB特征自适应调制结合实现YOLOv10雾天场景下颜色特征的精准增强与色调一致性保持。本文利用CEM模块,通过均值归一化去除雾霾光照干扰,再通过全局池化与Softmax生成色度权重,同时依托残差融合保留原始特征信息,对YOLOv10中雾天模糊目标的颜色与纹理特征进行针对性强化、抑制色彩失真与背景灰度干扰在特征提取阶段实现双色空间色度信息高效互补,避免传统单RGB空间颜色鲁棒性不足问题,增强模型在雾霾、烟雾等复杂天气下的检测精度与色彩分辨能力。


专栏目录:YOLOv10改进目录一览 | 涉及卷积层、轻量化、注意力、损失函数、Backbone、SPPF、Neck、检测头等全方位改进

专栏地址:YOLOv10改进专栏——以发表论文的角度,快速准确的找到有效涨点的创新点!

文章目录

  • 一、本文介绍
  • 二、CEM介绍
    • 2.1 设计出发点
    • 2.2 模块结构
    • 2.3 模块优势
  • 三、CEM实现代码
  • 四、添加步骤
    • 4.1 修改一
    • 4.2 修改二
    • 4.3 修改三
  • 五、yaml模型文件
    • 5.1 改进版本
  • 六、成功运行结果

二、CEM介绍

2.1 设计出发点

CEM(Color Enhancement Module,颜色增强模块)旨在解决雾天图像RGB颜色失真、色度信息弱、色调不一致的问题。RGB空间在雾霾下亮度被干扰、颜色特征模糊,而YCbCr空间的色度分量对雾霾不敏感,能保留更纯净的颜色信息。CEM通过提取YCbCr的色度特征来调制RGB特征,在不进行颜色空间转换的前提下,恢复自然色调、增强颜色感知,让去雾后的图像色彩更准确、对比度更自然。

2.2 模块结构

CEM采用均值归一化+全局池化+Softmax调制的轻量化结构:

  1. 均值归一化:对YCbCr特征做通道均值减法,去除全局光照与雾霾影响,突出局部颜色差异;
  2. 特征映射:通过全局平均池化+Softmax生成颜色分布权重,捕捉有效色度信息;
  3. 特征调制:用YCbCr得到的权重对RGB特征进行加权增强,再与YCbCr特征残差相加;
  4. 输出:得到色调一致、色彩鲜艳、无失真的增强特征。

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询