一种基于改进YOLOv8的无人机小目标检测算法

本发明提供了一种基于改进YOLOv8的无人机小目标检测算法，包括对与目标像素点有关联的关联像素点的特征结合到该目标像素点，生成增强特征图，然后加入动态注意力的CBAM模块，动态生成注意力权重生成重分配特征图，输入YOLOv8的检测头，在每一个待检测目标处标记出对应的边界框、类别和置信度标签等步骤。本发明通过非局部机制以及引入动态CBAM模块，使得模型能够根据每幅图像的具体特征，自动调整特征关注区域，在特征提取过程中，通过建模特征图中的全局像素关联度，相比局部像素关联的模块，使得小目标的特征可以被全局视角识别，帮助模型更有效地关注到细小、易混淆的目标区域。