- 专利标题: 基于注意力的Water-Lite-HRNet遥感图像水体分割方法
-
申请号: CN202310992432.4申请日: 2023-08-08
-
公开(公告)号: CN117058674A公开(公告)日: 2023-11-14
- 发明人: 于永涛 , 姜涛 , 江佳敏 , 李银银 , 黄龙 , 顾函竹 , 卢伟彬 , 吴庆国 , 刘超 , 王俊
- 申请人: 淮阴工学院
- 申请人地址: 江苏省淮安市经济技术开发区枚乘东路1号
- 专利权人: 淮阴工学院
- 当前专利权人: 淮阴工学院
- 当前专利权人地址: 江苏省淮安市经济技术开发区枚乘东路1号
- 代理机构: 淮安市科文知识产权事务所
- 代理商 吴晶晶
- 主分类号: G06V20/70
- IPC分类号: G06V20/70 ; G06V20/10 ; G06V10/26 ; G06V10/77 ; G06V10/80 ; G06V10/82 ; G06N3/0464 ; G06N3/08
摘要:
本发明公开了一种基于注意力的Water‑Lite‑HRNet遥感图像水体分割方法,包括以下步骤:首先预处理LandCover.ai原始高分辨率遥感图像,得到本发明使用的水体分割数据集,包括训练集、测试集和验证集;其次对原始HRNet网络进行结构化剪枝,得到Water‑Lite‑HRNet网络,同时嵌入设计的注意力模块和相邻分支注意力模块;然后对Water‑Lite‑HRNet采用水体分割数据集中的训练集及验证集对网络模型进行微调与训练,得到分割模型Y;最后将水体分割数据集中的测试集输入到Y中,得到初始分割结果,对初始分割结果进行后处理,修正错误或不确定性,得到最终的水体分割结果。与现有技术相比,本发明提出的基于注意力的Water‑Lite‑HRNet模型,在大规模减少HRNet参数量提高模型训练速度的同时提高了模型分割精度。