摘要
本发明公开了一种基于CoMFE的小目标检测方法、装置、设备及介质,包括:获取原始图像;将原始图像输入至预训练后的小目标检测模型中,输出检测结果,小目标检测模型中,骨干网络为基于坐标多分支特征增强模块,即CoMFE模块的骨干网络,在骨干网络中引入坐标注意力模块与多分支特征增强模块,用于对原始图像进行多特征提取输出,每个坐标多分支特征增强模块的输出与骨干网络的最终输出连接至颈部网络,颈部网络用于融合骨干网络输出的不同尺度特征后输出融合特征,颈部网络与若干检测头连接,每个检测头用于对融合特征进行目标检测,输出检测结果;对检测结果采用非极大抑制算法去除冗余预测框,得到最终的小目标检测结果。
技术关键词
多分支
注意力
网络
检测头
融合特征
坐标
抑制算法
图像获取模块
全局平均池化
计算机设备
可读存储介质
多尺度
空洞
处理器
冗余
存储器
输出模块