一种遥感船舶检测模型及方法
摘要:
本发明公开了一种遥感船舶检测模型,包括Res2Net50、FPN网络、空间上下文聚合模块和RPN;Res2Net50用于提取遥感图像的多尺度特征,将多尺度特征输出至FPN网络;FPN用于将多尺度特征中的深层特征与浅层特征进行融合,得到融合特征,将融合特征输出至上下文聚合模块;上下文聚合模块基于融合特征,获取目标船舶的上下文信息,将包含目标船舶的上下文信息的特征输出至RPN;RPN生成目标船舶的候选框。本发明还公开了一种遥感船舶检测方法,包括:获取包含遥感图像和标注文件的数据集;构建遥感船舶检测模型;利用数据集对遥感船舶检测模型进行训练;利用训练好的遥感船舶检测模型进行目标船舶检测。本发明能够有效抑制背景中相似地物的误检问题,降低了小型舰船的漏检率。
0/0