yolov5改进_添加CA注意力机制

69 阅读1分钟

image.png

CA的优势

  • 不仅考虑了通道信息,还考虑了方向相关的位置信息
  • 足够的灵活和轻量,能够简单的插入到轻量级网络的核心模块中
  • 可以作为预训练模型用于多种任务中,如检测和分割,均有不错的性能提升

image.png 成功加入了CA模块