YoloV8改进策略:三元注意力,小参数大能力,即插即用,涨点自如

文章目录

  • 摘要
  • 三元注意力机制
  • 三元组注意力机制的优点
  • YoloV8实验结果
  • 三元注意力代码
  • 改进一:将三元注意力加入到C2f模块中,重构C2f模块
    • 改进方法
    • 测试结果
  • 改进二:将三元注意力加入到主干网络后面
    • 改进方法
    • 测试结果
  • 总结

摘要

注意力机制在计算机视觉领域得到了广泛的研究和应用,利用构建通道或空间位置之间的依赖关系的能力,有效地应用于各种计算机视觉任务。本文研究了轻量级但有效的注意力机制,并提出了一种新的计算注意力权重的方法——三元组注意力,通过一个三分支结构捕捉跨维度交互。对于输入张量,三元组注意力通过旋转操作和残差变换建立跨维度的依赖关系&#x

你可能感兴趣的:(YOLO)