基于梯度流特征融合的泵站安全风险检测

Safety risk detection method of pumping station based on gradient flow feature fusion

为解决泵站场景下远距离小尺寸目标和特征遮挡等因素对泵站安全风险检测任务带来的挑战,文章提出一种融合坐标注意力机制和梯度流特征提取模块的泵站安全风险检测算法。通过在特征提取网络中采用梯度流特征提取模块以提高网络的特征提取能力,同时引入坐标注意力机制模块提升对全局信息的捕获能力,最后在特征融合模块采用BiFPN结构提高对不同尺寸目标特征的融合效果。在泵站场景下进行实验,结果表明文章所提算法在泵站场景下的安全风险检测任务中算法平均准确度达到94.78%,与常见的目标检测算法相比具有较好的效果。