一种基于注意力机制的轻量级遮挡船舶检测方法

本发明属于计算机视觉与目标检测技术领域，具体涉及一种基于注意力机制的轻量级遮挡船舶检测方法，包括：获取待处理遮挡条件下船舶目标的数据集；对所述数据集中的图片进行预处理；将预处理后的数据集按照比例划分为训练集和测试集；构建用于特征提取、特征融合和预测目标所在的位置和所属类别的eAodeMar模型，所述eAodeMar模型包括位置增强模块和轻量级多尺度特征语义关联模块；基于训练集对eAodeMar模型进行训练，得到训练好的eAodeMar模型；将测试集输入到训练好的eAodeMar模型中，实现遮挡条件下无人艇海上船舶目标的检测。本发明提出的eAodeMar模型能够完成对场景中船只的正确定位和快速识别，检测速度得到了大幅提升。