YOLOv10模型改进-注意力机制-第34篇:YOLOv10改进策略【注意力机制】| SimAM注意力机制

📅 2026/7/1 23:44:06
YOLOv10模型改进-注意力机制-第34篇:YOLOv10改进策略【注意力机制】| SimAM注意力机制
一、本文介绍本文记录的是利用SimAM注意力机制改进YOLOv10的特征提取部分。SimAM通过计算神经元的重要性得分,实现无参数的自适应特征增强。二、SimAM注意力机制介绍2.1 设计出发点传统注意力机制需要额外参数,SimAM通过计算神经元的重要性得分,实现无参数的自适应特征增强。2.2 模块结构SimAM:神经元重要性计算:通过能量函数计算每个神经元的重要性权重归一化:归一化权重特征增强:应用权重到特征三、SimAM注意力机制的实现代码importtorchimporttorch.nnasnnclassSimAM