发明公开
- 专利标题: 一种基于卷积神经网络和注意力机制的水下图像分类方法
-
申请号: CN201911310684.4申请日: 2019-12-18
-
公开(公告)号: CN111046967A公开(公告)日: 2020-04-21
- 发明人: 杨淼 , 汤雁冰 , 胡珂 , 卢道华 , 徐启华 , 杨慧珍 , 纪林海 , 姚潇 , 康家银
- 申请人: 江苏科技大学 , 江苏科技大学海洋装备研究院
- 申请人地址: 江苏省镇江市京口区梦溪路2号
- 专利权人: 江苏科技大学,江苏科技大学海洋装备研究院
- 当前专利权人: 江苏科技大学,江苏科技大学海洋装备研究院
- 当前专利权人地址: 江苏省镇江市京口区梦溪路2号
- 代理机构: 北京一格知识产权代理事务所
- 代理商 万小侠
- 主分类号: G06K9/62
- IPC分类号: G06K9/62 ; G06K9/46 ; G06N3/04
摘要:
本发明涉及一种基于卷积神经网络和注意力机制的水下图像分类方法,以Inception-Attention结构为基础构建卷积神经网络,实现对图像的背景特征进行提取并加权,使用训练集中的图像作为原始输入图像,输入到所提网络中进行训练,分别在网络的主干分类分支上和辅助分类分支上得到预测结果,通过分别计算预测结果与图像的真实标签之间的交叉熵以及网络的L2正则化项得到损失函数,并利用随机梯度下降对网络进行优化,再将待预测的图像输入到已训练的网络中,可得到对应的预测结果。本发明的优点在于:本发明通过收集大量的水下图像和非水下图像对所提网络进行训练,大规模图像数据集上进行识别水下图像任务时,可以快速、准确地进行图像分类。