一种基于注意力模型的遥感图像船舶检测方法和装置
基本信息
| 申请号 | CN202210579628.6 | 申请日 | - |
| 公开（公告）号 | CN114677596A | 公开（公告）日 | 2022-06-28 |
| 申请公布号 | CN114677596A | 申请公布日 | 2022-06-28 |
| 分类号 | G06V20/10（2022.01）I;G06V20/54（2022.01）I;G06V10/774（2022.01）I;G06V10/82（2022.01）I;G06V10/26（2022.01）I;G06V10/44（2022.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I;G06K9/62（2022.01）I | 分类 | 计算；推算；计数; |
| 发明人 | 黎晨阳;王军;徐晓刚;徐冠雷;朱亚光 | 申请（专利权）人 | 浙江工商大学 |
| 代理机构 | 杭州浙科专利事务所（普通合伙） | 代理人 | - |
| 地址 | 311100浙江省杭州市余杭区中泰街道之江实验室南湖总部 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本发明公开一种基于注意力模型的遥感图像船舶检测方法和装置，该方法包括：步骤一，收集船舶遥感图像，使用标注信息对图像数据集进行数据扩充；步骤二，对收集的遥感图像进行预处理，得到船舶检测模型训练使用的数据集；步骤三，将训练数据集里的图像输入到针对小尺寸目标改进过的YOLOV5注意力模型中，进行训练，得到训练好的遥感船舶检测模型；步骤四，将待检测的遥感图像裁剪，后输入到训练好的遥感船舶检测模型中，输出船舶的边界框和置信度；将所有裁剪图的边界框映射回原始遥感图像，经过置信度阈值过滤和非极大值抑制过滤重复目标后，获得最终的检测结果。本发明解决了遥感图像中船舶分布稀疏、尺寸过小导致的检测困难的问题。 |
