[发明专利]基于迁移学习的内河船舶重识别方法、设备及存储介质

说明书

本发明提供了一种基于迁移学习的内河船舶重识别方法、设备及存储介质，重识别方法利用风格迁移后的行人重识别数据集，初始化一个CNN模型得到第一重识别模型，并利用第一重识别模型对无标签的内河船舶数据集进行特征提取；利用无监督K‑Means聚类算法，对第一重识别模型提取的特征进行聚类，得到具有伪标签的第一内河船舶数据集；利用第一内河船舶数据集对第一重识别模型微调，得到内河船舶重识别模型。利用内河船舶重识别模型，得到每一张待识别内河船舶图像的候选图像列表。本发明提出的基于迁移学习的内河船舶重识别方法不需要对内河船舶数据集进行标注，很好的解决了现有技术中在同域场景下内河船舶数据集有限、标注成本大以及精度不高的问题。

技术领域

本发明涉及计算机视觉技术领域，特别涉及基于迁移学习的内河船舶重识别方法、设备及存储介质。

背景技术

随着计算机视觉技术的不断发展，大规模的摄像头布置在城市的各个角落，行人重识别技术结合目前大量的图像和视频数据，在智能视频监控以及智能安保等领域，发挥着越来越重要的作用。得益于深度学习算法的快速发展、硬件计算能力的迅速提升以及大规模数据集如CUHK03、Market-1501、MSMT17等的公布，行人重识别技术在一些标准公开数据集上的测试准确率已经得到了显著提高。

现有技术中，应用最为广泛的重识别方法的基本思想是把重识别问题转换为分类问题，方法缺点是对于给定一个目标域，需要大量有标签的数据集来初始化CNN模型。具体地，以内河船舶重识别方法为例，其基本方法为，首先，利用一个特定场景的大量有标签的数据集预训练一个CNN(卷积神经网络)模型，比如采用ResNet-50模型；然后，提取输入图像的全局特征，之后接一个全连接层和softmax分类器，把每一类内河船舶视为一类，通过优化分类损失函数来训练这个模型。尽管基于深度神经卷积网络的方法取得了显著的进展，而现实中，针对一个特定的场景，往往拿不到这么多的数据集从而导致模型的测试性能不佳。

进一步地，不仅如此，人为对数据集进行标注往往又花费较大的人力和物力。这其中，较为典型的就是内河船舶重识别方法，就普遍面临这样的问题，目前针对内河船舶重识别的数据集非常有限，同时给定一个特定的内河船舶重识别场景，为了达到令人满意的识别精度，往往需要花费大量的人力和物力去标注内河船舶样本来训练重识别模型。

因此，如何解决内河船舶标注样本不足，从而进一步提高内河船舶重识别方法的精度，成为本领域技术人员亟待解决的技术问题之一。

需要说明的是，公开于该发明背景技术部分的信息仅仅旨在加深对本发明一般背景技术的理解，而不应当被视为承认或以任何形式暗示该信息构成已为本领域技术人员所公知的现有技术。

发明内容

本发明所要解决的技术问题为，现有技术中内河船舶重识别方法在同域场景下数据集有限，标注成本大以及识别精度不高的问题。本发明的其中一个目的是提供一种基于迁移学习的内河船舶重识别方法，另外一个目的是提供一种计算机设备和提供一种计算机存储介质。

为了实现上述第一个目的，本发明通过以下技术方案予以实现：一种基于迁移学习的内河船舶重识别方法，包括如下步骤：

S100：利用风格迁移后的行人重识别数据集，初始化一个CNN模型得到第一重识别模型，并利用所述第一重识别模型对无标签的内河船舶数据集进行特征提取；

S200:利用无监督K-Means聚类算法，对所述第一重识别模型提取的特征进行聚类，得到具有伪标签的第一内河船舶数据集；

S300：利用所述第一内河船舶数据集对所述第一重识别模型微调，得到内河船舶重识别模型；

S400：利用所述内河船舶重识别模型，提取待识别内河船舶数据集中每一张待识别内河船舶图像的深度特征得到特征向量，计算所述特征向量之间的欧式距离，得到每一张待识别内河船舶图像的候选图像列表；