一种多尺度的YOLOv8颈部网络架构搜索方法及系统
申请号:CN202511127841
申请日期:2025-08-13
公开号:CN120635607A
公开日期:2025-09-12
类型:发明专利
摘要
本发明公开了一种多尺度的YOLOv8颈部网络架构搜索方法及系统,属于计算机视觉和深度神经网络技术领域。该方法包括:将YOLOv8颈部网络中用于特征提取的普通卷积替换为多尺度卷积,并在特征拼接模块后加入通道注意力机制,通过动态通道加权来初步提升多尺度特征提取与融合的质量;为YOLOv8的颈部网络设计搜索空间并构建一个全连接的超网;使用可微分架构搜索方法训练超网直到目标函数收敛,得到一个最优的颈部网络架构;对搜索得到的最优YOLOv8颈部网络架构进行重训练,以验证其高效性。本发明在略微增加参数量的情况下,显著提升了模型的检测精度。
技术关键词
网络架构
搜索方法
通道注意力机制
节点
图像识别模块
深度神经网络技术
参数
拼接模块
多尺度特征提取
卷积模块
数据更新
网格
全局平均池化
搜索系统
标签
计算机视觉