一种基于卫星影像船舶数据的协同检测分类模型

本发明公开了一种基于卫星影像船舶数据的协同检测分类模型，包括以下步骤：(1)输入遥感影像，对影像进行数据预处理，而后裁剪样本图片并划分为训练集、验证集；(2)使用训练集图片训练模型，模型由分类分支和检测分支构成；分类分支由重剪切模块、特征提取模块和分类输出模块构成；(3)使用模型进行推理测试时，待测图片先经过检测分支，获得多个目标检测结果；再将每个目标分别以其中心为准向四周扩充裁剪出图片输入分类分支，获得分类结果；设置分类阈值，用分类结果对检测结果进行过滤，获得模型的输出结果。本发明的模型通过结合检测分支强大的目标提取能力和分类分支良好的误检抑制能力，有效提高影像中船舶目标的检测精度。