基于注意力的Water-Lite-HRNet遥感图像水体分割方法

    公开(公告)号:CN117058674A

    公开(公告)日:2023-11-14

    申请号:CN202310992432.4

    申请日:2023-08-08

    申请人: 淮阴工学院

    摘要: 本发明公开了一种基于注意力的Water‑Lite‑HRNet遥感图像水体分割方法,包括以下步骤:首先预处理LandCover.ai原始高分辨率遥感图像,得到本发明使用的水体分割数据集,包括训练集、测试集和验证集;其次对原始HRNet网络进行结构化剪枝,得到Water‑Lite‑HRNet网络,同时嵌入设计的注意力模块和相邻分支注意力模块;然后对Water‑Lite‑HRNet采用水体分割数据集中的训练集及验证集对网络模型进行微调与训练,得到分割模型Y;最后将水体分割数据集中的测试集输入到Y中,得到初始分割结果,对初始分割结果进行后处理,修正错误或不确定性,得到最终的水体分割结果。与现有技术相比,本发明提出的基于注意力的Water‑Lite‑HRNet模型,在大规模减少HRNet参数量提高模型训练速度的同时提高了模型分割精度。

    一种基于微调大模型的溢油检测方法、系统及设备

    公开(公告)号:CN118506150A

    公开(公告)日:2024-08-16

    申请号:CN202410554630.7

    申请日:2024-05-07

    申请人: 淮阴工学院

    摘要: 本发明公开了一种基于微调大模型的溢油检测方法、系统及设备,其种方法包括:采集并保存溢油图像数据;对溢油图像数据根据溢油级别进行人工标注,并制作成数据集;将数据集按照一定比例分割为训练集和测试集并且对溢油图像进行预处理;构建基于Vision Transformer和LoRA的溢油检测模型;包括:构建基于Vision Transformers的模型,通过Vision Transformers的多头自注意力机制,捕捉图像中的特征;通过LoRA技术,冻结Vision Transformers的权重,将可训练的秩分解矩阵注入到Vision Transformers架构的每一层,在每个Vision Transformers块中注入可训练层,减少需要训练的参数数量;通过调整参数在训练集上迭代训练溢油检测模型,得到最优溢油检测模型,对溢油进行检测;本发明通过LoRA进行微调的效果与全模型微调相同,速度更快,计算需求更少,同时保持或提升模型在溢油检测任务上的性能。

    一种基于Conv-LSTM-RA网络的空气质量指数预测方法

    公开(公告)号:CN116485015A

    公开(公告)日:2023-07-25

    申请号:CN202310354751.2

    申请日:2023-04-04

    申请人: 淮阴工学院

    摘要: 本专利公开了一种基于Conv‑LSTM‑RA网络的空气质量指数预测方法,首先,对获取到的历史空气质量数据进行预处理;其次,利用卷积神经网络(CNN)中的一维卷积结构(Conv1D)进行特征提取,并在卷积神经网络中加入残差模块,以获取更多的特征信息;然后,使用长短期记忆(LSTM)在时间维度上根据设定的窗口进行未来空气质量指数的预测;最后,在LSTM中使用注意力机制,利用注意力机制学习输入序列的权重分布,得出最终预测结果。与现有的单一网络模型相比,本发明提出的混合模型方法能够更加有效的预测数据,提高预测模型的准确率。

    一种基于改进注意力的Water-Res-Swin遥感图像水体分割方法

    公开(公告)号:CN118608786A

    公开(公告)日:2024-09-06

    申请号:CN202410776988.4

    申请日:2024-06-17

    申请人: 淮阴工学院

    摘要: 一种基于改进注意力的Water‑Res‑Swin遥感图像水体分割方法,包括以下步骤:1.预处理自制数据集,得到水体分割数据集;2.搭建Water‑Res‑Swin模型:分别使用ResNet50网络和Swin Transformer网络进行特征提取,得到多尺度特征图,将相同维度的特征图融合,提高模型的特征表现能力。同时,在模型中嵌入优化代理注意力(OAG)模块,以降低计算复杂度并生成增强的特征表示;3.采用自制水体分割数据集中的训练集及验证集对Water‑Res‑Swin网络模型进行训练与微调,生成分割模型Y;4.将水体分割数据集中的测试集输入到Y中,得到初始分割结果,对初始分割结果进行后处理,修正错误或不确定性,得到最终的水体分割结果。本发明提出的基于改进注意力的Water‑Res‑Swin模型,在减少计算复杂度的同时提高了模型分割精度。

    基于对比学习的REIN-MoCoV3-OLA石油溢油检测方法

    公开(公告)号:CN118429809A

    公开(公告)日:2024-08-02

    申请号:CN202410580598.X

    申请日:2024-05-10

    申请人: 淮阴工学院

    摘要: 本发明公开了一种基于对比学习的REIN‑MoCoV3‑OLA石油溢油检测方法,包括以下步骤:下载公开的ImageNet数据集和Oil Spill Detection Dataset数据集;搭建REIN‑MoCoV3‑OLA模型:在MoCoV3模型的主干层之间部署REIN模块,并根据微调数据集来训练参数。在模型中嵌入优化线性注意力模块OLA;使用公开的ImageNet数据集,采用对比学习的方式对REIN‑MoCoV3‑OLA模型进行预训练;4.使用Oil Spill Detection Dataset数据集对预训练后的模型进行微调。在微调阶段,模型将根据石油溢油检测任务的特定数据集来进一步优化模型参数,以适应特定任务的需求。与现有技术相比,本发明可利用对比学习和注意力机制来构建和训练适用于石油溢油检测任务的深度学习模型,并取得了良好的性能表现。