YOLOv11改进 | Conv/卷积篇 | 利用MobileNetV4的UIB模块二次创新C3k2适配yolov11扩展率(全网独家首发,轻量化)

一、本文介绍

本文给大家带来的最新改进机制是利用MobileNetV4UIB模块二次创新C3k2,其中UIB模块来自2024.5月发布的MobileNetV4网络,其是一种高度优化的神经网络架构,专为移动设备设计。它最新的改动总结主要有两点采用了通用反向瓶颈(UIB,也就是本文利用的结构)和针对移动加速器优化的Mobile MQA注意力模块(一种全新的注意力机制)。我将其用于C3k2的二次创新在V11n上参数量为250W),计算量为6.0GFLOPs,非常适用于想要轻量化网络模型的读者来使用,同时本文结构为本专栏独家创新

欢迎大家订阅我的专栏一起学习YOLO! 


 专栏回顾YOLOv11改进系列专栏——本专栏持续复习各种顶会内容——科研必备

目录

一、本文介绍

二、原理介绍

三、核心代码

四、添加教程

4.1 修改一

4.2 修改二 

4.3 修改三 

4.4 修改四 

五、C2fUIB的yaml文件和运行记录

5.1 C2fUIB的yaml文件

5.2 训练代码 

5.3 C2fUIB的训练过程截图 

五、本文总结


二、原理介绍

 

官方论文地址: 官方论文地址点击此处即可跳转

官方代码地址: 官方代码地址点击此处即可跳转

### 如何使用 MobileNetV4UIB 改进 YOLOv8 的 C2f 模块 #### 实现方法 为了实现这一目标,首先需要理解 MobileNetV4 中的通用反向瓶颈(UIB)结构及其特性。UIB 结构通过减少通道数量并增加非线性操作来提高效和性能[^3]。 具体来说,在 YOLOv8 的 C2f 模块中引入 UIB 可以按照以下方式: 1. **替换原有卷积层**:将原有的标准卷积层替换成带有深度可分离卷积UIB 单元。这不仅减少了参数量还提升了计算速度。 ```python class UIB(nn.Module): def __init__(self, in_channels, out_channels, stride=1): super(UIB, self).__init__() hidden_dim = int(in_channels * expansion_ratio) layers = [ nn.Conv2d(in_channels, hidden_dim, kernel_size=1), nn.BatchNorm2d(hidden_dim), nn.ReLU6(inplace=True), nn.Conv2d(hidden_dim, hidden_dim, groups=hidden_dim, kernel_size=kernel_size, padding=(kernel_size-1)//2, stride=stride), nn.BatchNorm2d(hidden_dim), nn.ReLU6(inplace=True), nn.Conv2d(hidden_dim, out_channels, kernel_size=1), nn.BatchNorm2d(out_channels) ] self.conv = nn.Sequential(*layers) def forward(self, x): return self.conv(x) ``` 2. **集成到 C2f 模块**:修改后的 C2f 应该能够接收输入特征图并通过一系列改进过的 UIB 层处理这些数据。最终输出经过增强表示的学习结果给下一层网络继续工作。 3. **调整超参数配置文件**:根据实际需求微调模型训练过程中的学习、批量大小等设置,确保最佳效果的同时保持较低资源消耗。 #### 效果对比 采用上述方案后,相比于原始版本,新架构下的 YOLOv8 表现出显著优势: - 参数总量大幅降低至约 220W 左右,相比之前减少了近百万个参数; - 计算复杂度也有所改善,达到大约 6.2 GFLOPs; - 对于移动端应用场景而言更加友好,能够在保证检测精度的前提下极大程度节省硬件成本与功耗开销[^4]。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Snu77

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值