发明公开
- 专利标题: 基于对比学习的REIN-MoCoV3-OLA石油溢油检测方法
-
申请号: CN202410580598.X申请日: 2024-05-10
-
公开(公告)号: CN118429809A公开(公告)日: 2024-08-02
- 发明人: 顾函竹 , 于永涛 , 张蓝天 , 卢伟彬 , 黄龙 , 吴庆国 , 李银银 , 姜涛
- 申请人: 淮阴工学院
- 申请人地址: 江苏省淮安市经济技术开发区枚乘东路1号
- 专利权人: 淮阴工学院
- 当前专利权人: 淮阴工学院
- 当前专利权人地址: 江苏省淮安市经济技术开发区枚乘东路1号
- 代理机构: 淮安市科文知识产权事务所
- 代理商 张舒燕
- 主分类号: G06V20/10
- IPC分类号: G06V20/10 ; G06V10/44 ; G06V10/774 ; G06V10/74 ; G06V10/82 ; G06N3/0455 ; G06N3/0464 ; G06N3/048 ; G06N3/08 ; G06N3/0499
摘要:
本发明公开了一种基于对比学习的REIN‑MoCoV3‑OLA石油溢油检测方法,包括以下步骤:下载公开的ImageNet数据集和Oil Spill Detection Dataset数据集;搭建REIN‑MoCoV3‑OLA模型:在MoCoV3模型的主干层之间部署REIN模块,并根据微调数据集来训练参数。在模型中嵌入优化线性注意力模块OLA;使用公开的ImageNet数据集,采用对比学习的方式对REIN‑MoCoV3‑OLA模型进行预训练;4.使用Oil Spill Detection Dataset数据集对预训练后的模型进行微调。在微调阶段,模型将根据石油溢油检测任务的特定数据集来进一步优化模型参数,以适应特定任务的需求。与现有技术相比,本发明可利用对比学习和注意力机制来构建和训练适用于石油溢油检测任务的深度学习模型,并取得了良好的性能表现。