基于多尺度特征融合和注意力机制的水面死鱼检测方法

作     者：杨帅鹏 李贺 刘金江 付主木 张锐 贾会梅 YANG Shuaipeng;LI He;LIU Jinjiang;FU Zhumu;ZHANG Rui;JIA Huimei

作者机构：南阳师范学院计算机科学与技术学院河南南阳473016 河南中光学集团有限公司河南南阳473003 河南科技大学信息工程学院河南洛阳471000 

基　　金：国家自然科学基金项目(62002180) 河南省科技攻关项目(202102210362,232102210149) 河南省高等学校重点科研项目(24A520030) 南阳师范学院实验室开放项目(SYKF2021029) 

出 版 物：《郑州大学学报（理学版）》 (Journal of Zhengzhou University:Natural Science Edition)

年 卷 期：2024年第56卷第6期

页      码：32-38页

摘      要：死鱼对于水域生态和饮水安全存在巨大威胁,由于水面环境复杂,导致现有目标检测算法在死鱼检测任务中存在漏检、误检等情况。为此,以单次多边框检测(single shot multibox detector,SSD)为基础提出一种基于多尺度特征融合和注意力机制的水面死鱼检测方法FFA-SSD(SSD with feature fusion and attention)。首先,采用计算量和参数量更少且特征提取能力更强的残差网络ResNet50替换VGG16主干网络;其次,设计了多尺度特征融合模块,增强浅层特征和高层语义信息的融合;然后,引入通道注意力机制,抑制特征融合带来的冗余信息干扰,提升网络对目标的关注度;最后,设计了一种适用于小目标检测的数据增强算法,扩充训练数据中的小目标数量,丰富训练背景。实验结果表明,同现有目标检测算法相比,FFA-SSD算法可以更好地识别水面死鱼,检测精度达到93.5%。

主 题 词：SSD 目标检测 特征融合 注意力机制 数据增强 

学科分类：080904[080904] 0810[工学-土木类] 0809[工学-计算机类] 080202[080202] 08[工学] 080402[080402] 0804[工学-材料学] 081001[081001] 0802[工学-机械学] 

D　O　I：10.13705/j.issn.1671-6841.2023149

馆 藏 号：203131575...