logo好方法网

一种船只检测模型训练方法、船只检测方法及装置


技术摘要:
本发明公开了一种船只检测模型训练方法、船只检测方法及装置,该船只检测模型训练方法包括:获取船只训练样本,所述船只训练样本包括多张海面环境图像;对所述船只训练样本进行提取,得到第一船只训练样本;将所述第一船只训练样本分割成m*n的网格,得到目标训练样本,  全部
背景技术:
海上船只的识别对于海上安防具有重要的战略意义,对于我国海洋强国战略实施 具有极大的帮助。海上船只检测是指对通过图像或视频等方式输入的船只进行特征信息提 取,并将提取的特征信息与数据库中的船只信息进行匹配,匹配好的船只信息和输入的船 只信号划分成同一类。海上船只检测应用及其广泛,例如,恶意船只的靠近、非我国船只进 入我国领海、智能监控系统、海洋大数据等。然而,在海面船只识别中,风浪、大雾、光照等都 会对海面船只识别造成重大的影响,其中,船只受海浪影响导致船只检测误报率较高。
技术实现要素:
因此,本发明要解决的技术问题在于克服现有技术中的海浪问题下船只检测率低 的缺陷,从而提供一种船只检测模型训练方法、船只检测方法及装置。 根据第一方面,本发明实施例公开了一种船只检测模型训练方法,包括如下步骤: 获取船只训练样本,所述船只训练样本包括多张海面环境图像;对所述船只训练样本进行 提取,得到第一船只训练样本;将所述第一船只训练样本分割成m*n的网格,得到目标训练 样本,m小于n;根据所述目标船只训练样本训练得到船只检测模型。 结合第一方面,在第一方面第一实施方式中,所述对所述船只训练样本进行提取, 得到第一船只训练样本,包括:提取所述船只训练样本中存在海浪影响的图像;对所述存在 海浪影响的图像进行标记,得到第一船只训练样本。 结合第一方面第一实施方式,在第一方面第二实施方式中,所述将所述第一船只 训练样本分割成m*n的网格,得到目标训练样本,m小于n,包括:根据第一预设的卷积核对所 述第一船只训练样本进行卷积操作,得到第一目标训练样本;根据第二预设的卷积核对所 述第一目标训练样本进行降维处理,得到目标训练样本。 结合第一方面,在第一方面第三实施方式中,在所述根据所述目标船只训练样本 训练得到船只检测模型之后,所述方法还包括:获取船只图像测试样本,所述船只图像测试 样本包括含船只的正测试样本和不含船只的负测试样本;根据所述船只图像测试样本及所 述船只检测模型得到测试结果;当所述测试结果满足预设条件时,将所述船只检测模型确 定为可用的船只检测模型。 根据第二方面,本发明实施例还公开了一种船只检测方法,包括如下步骤:获取船 只检测图像,所述船只检测图像包括船只和海浪;将所述船只检测图像输入到所述船只检 测模型中,得到船只检测结果;所述船只检测模型是通过如第一方面或第一方面任一实施 方式所述的船只检测模型训练方法训练生成的。 结合第二方面,在第二方面第一实施方式中,在所述获取船只检测图像之后,所述 4 CN 111553183 A 说 明 书 2/7 页 方法还包括:将所述船只检测图像分割成m*n的网格,m小于n。 根据第三方面,本发明实施例还公开了一种船只检测模型训练装置,包括:第一获 取模块,用于获取船只训练样本,所述船只训练样本包括多张海面环境图像;第一提取模 块,用于对所述船只训练样本进行提取,得到第一船只训练样本;分割模块,用于将所述第 一船只训练样本分割成m*n的网格,得到目标训练样本,m小于n;训练模块,用于根据所述目 标船只训练样本训练得到船只检测模型。 根据第四方面,本发明实施例还公开了一种船只检测装置,包括:第二获取模块, 用于获取船只检测图像;检测模块,用于将所述船只检测图像输入到所述船只检测模型中, 得到船只检测结果,所述船只检测模型是通过如第一方面或第一方面任一实施方式所述的 船只检测模型训练方法训练生成的。 根据第五方面,本发明实施例还公开了一种计算机设备,包括:至少一个处理器; 以及与所述至少一个处理器通信连接的存储器;其中,所述存储器存储有可被所述一个处 理器执行的指令,所述指令被所述至少一个处理器执行,以使所述至少一个处理器执行如 第一方面或第一方面任一实施方式所述船只检测模型训练方法或如第二方面或第二方面 任一实施方式所述船只检测方法。 根据第六方面,本发明实施例还公开了一种计算机可读存储介质,其上存储有计 算机指令,该指令被处理器执行时实现如第一方面或第一方面任一实施方式所述船只检测 模型训练方法或如第二方面或第二方面任一实施方式所述船只检测方法。 本发明技术方案,具有如下优点: 1 .本发明提供的船只检测模型训练方法及装置,通过获取船只训练样本,船只训 练样本包括多张海面环境图像,对船只训练样本进行提取,得到第一船只训练样本,将第一 船只训练样本分割成m*n的网格,得到目标训练样本,m小于n,根据目标船只训练样本训练 得到船只检测模型。本发明通过对船只训练样本进行非均匀分割,在保持横向检测密度不 变的情况下,增加纵向检测密度,使得切割后的训练样本与船只形状一致,对船只检测更准 确,减少海浪对船只的影响。 2.本发明提供的船只检测方法及装置,通过获取船只检测图像,船只检测图像包 括船只和海浪,对船只检测图像进行特征提取,得到船只特征,将船只特征输入到船只检测 模型中,得到船只检测结果,本发明通过将船只检测图像输入到船只检测模型,对船只检测 更准确,减少海浪对船只的影响。 附图说明 为了更清楚地说明本发明
分享到:
收藏