基于You Only Look Once（YOLO）的目标检测器在自动脑瘤检测中展现出卓越的准确性。在本文中，我们开发了一种新的BGF-YOLO架构，通过将双层路由注意力（BRA）、广义特征金字塔网络（GFPN）和第四检测头整合到YOLOv8中来实现。BGF-YOLO包含了一个注意力机制，用于更加关注重要的特征，并使用特征金字塔网络来丰富特征表示，将高级语义特征与空间细节合并。此外，我们还研究了不同的注意力机制和特征融合、检测头架构对脑瘤检测准确性的影响。实验结果显示，与YOLOv8x相比，BGF-YOLO在mAP50上实现了4.7%的绝对增加，并在脑瘤检测数据集Br35H上达到了最先进的水平。
论文地址：https://arxiv.org/ftp/arxiv/papers/2309/2309.12585.pdf
代码地址：https://github.com/mkang315/BGF-YOLO
