一种基于全局上下文交互的小目标检测方法

本发明属于图像检测领域，具体涉及一种基于全局上下文交互的小目标检测方法。该方法包括以下步骤：步骤1、获取小目标检测数据集；步骤2、构建SmallDet算法模型，该模型包括BCPM模块、Adown模块和VisionMamba模块；步骤3、将训练集中的小目标图像，输入到模型中进行训练，获取训练权重文件，采用损失函数通过前后向传播对训练权重文件进行迭代优化，直至模型收敛，获取训练好的模型；步骤4、将测试集中的小目标图像输入到训练好的模型中，进行并完成小目标图像检测。本发明采用BCPM模块处理图像能够提高图像分析的质量。采用Adown模块能够解决传统下采样模块中容易丢失细粒度信息的问题。采用VisionMamba模块能够解决了小目标检测中上下文信息不足、特征表达受限等问题。