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

目录
二、原理介绍

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

本文详细介绍了如何使用MobileNetV4的通用反转瓶颈(UIB)模块对YOLOv8进行二次创新,形成了C2fUIB模型。UIB结合了MobileNetV2、ConvNext和ViT的元素,旨在提高轻量化网络的性能。Mobile MQA注意力机制进一步优化了移动加速器的推理速度。文章提供修改教程和C2fUIB的yaml文件,适用于追求模型轻量化和高性能的读者。
订阅专栏 解锁全文
2980





