[发明专利]一种船只检索方法、装置、电子设备及存储介质

说明书

本发明实施例提供了一种船只检索方法、装置、电子设备及存储介质。该方法包括：获取待检索船只影像；利用预先构建的神经网络模型对待检索船只影像进行船只位置识别，提取船只特征向量；基于特征向量将待检索船只与预先构建的特征数据库进行比对，计算待检索船只与特征数据库中船只的第一相似度；选择相似度大于第一预设阈值的船只，得到预检索结果；从识别的船只位置处截取局部影像；将局部影像遍历预检索结果中的每一张船只影像，计算局部影像与预检索结果中的每一张船只影像中与局部影像相同大小区域的第二相似度；当存在第二相似度达到第二预设阈值的目标船只时，确定待检索船只为目标船只。通过实施该方法，提高了船只检索的准确性。

技术领域

本发明涉及图像处理领域，具体涉及一种船只检索方法、装置、电子设备及存储介质。

背景技术

随着观测技术的进步，可获取的影像数据量正在以惊人的速度增长。同时，数据日益多元化，数据获取数度更快，获取周期更短，时效性也更强。然而，与数据获取方面的快速发展不同，现有的数据处理技术还远远不能达到要求，从而导致了很多数据的浪费，不能充分利用数据的价值。研究如何从海量的数据中准确地获取有效的目标信息，检索出需要的图像，有利于数据库的高效利用。然而，当前许多图像中地物复杂且覆盖范围广，船只作为小型地物，具有特征信息相对较少，检索十分困难。

现有技术中，船只图像检索通常是计算待检索船只与数据库中的船只的相似度，以该相似度来确定出待检索船只身份，然而，在实际处理中，由于船只影像只能拍摄到船只的局部影像，在进行特征比对和相似度计算时，往往为了综合和局部影像的影响，会将相似度阈值相应地下调，这使得船只影响检索时，检索结果中会存在较大的噪音，导致检索结果不准确。

因此研究如何从海量船只图像数据中准确的对船只的身份进行识别，仍然是一个亟待解决的问题。

发明内容

有鉴于此，本发明实施例提供了一种船只检索方法、装置、电子设备及存储介质，以解决现有技术中船只检索准确性低的技术问题。

根据第一方面，本发明实施例提供了一种船只检索方法，包括如下步骤：获取待检索船只影像；利用预先构建的神经网络模型从所述待检索船只影像中识别出船只位置，提取所述待检索船只的特征向量；基于所述特征向量将所述待检索船只与预先构建的特征数据库进行比对，计算所述待检索船只与所述特征数据库中船只的第一相似度，其中，所述特征数据库中存有船只特征向量及其与船只的对应关系；选择相似度大于第一预设阈值的船只，得到预检索结果；从所述待检索船只影像上所述船只位置处截取所述待检索船只上的局部影像；将所述局部影像遍历所述预检索结果中的每一张船只影像，计算所述局部影像与所述预检索结果中的每一张船只影像中与所述局部影像相同大小的区域的第二相似度；当所述预检索结果存在所述第二相似度达到第二预设阈值的目标船只时，确定所述待检索船只为所述目标船只。

本发明实施例提供的船只检索方法，先基于船只影像对船只位置进行识别，并提取船只特征向量，利用提取的特征向量与预先构建的特征数据库进行比对，根据第一预设阈值与第一相似度比对结果得到初步检索结果；然后截取确定的船只位置处的局部影像，用截取的局部影像历遍初步检索结果中的所有船只影像，根据第二预设阈值与第二相似度比对结果得到最终检索结果的目标船只，确定待检索船只为最终检索结果中的目标船只，完成船只检索。初步结果是现有技术中常用的特征检索步骤，但是只能粗略的得到检索结果，不排其中包含其他不是待检索船只的船只影像，因此，本发明实施例在现有技术上进一步对初步检索结果中利用局部影像的特征比对，并根据局部影像的图像熵的不同设定了不同的第二预设阈值，对初步检索结果进一步筛选得到最终的检索结果，由此得到的检索结果是经过了两次筛选的结果，提高了船只检索的准确性，基本上排除了其它干扰选项，便可以将该最终结果的目标船只确定为待检索船只。