[发明专利]一种家居图像多目标识别方法及装置

说明书

本发明提供一种家居图像多目标识别方法及装置，能够为家居图像训练节省大量训练数据、计算资源和训练时间，且能够提高多目标识别的准确率。所述方法包括：获取家居图像数据集作为训练集；将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型，采用迁移学习的方法，保留预训练模型的卷积层和池化层，通过保留的卷积层和池化层提取训练集的图像特征，构建第一特征图和池化特征图；对池化特征图进行三次卷积操作，得到第二特征图；对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作，将反卷积操作后的两组特征图进行融合；利用softmax分类器对融合后的特征图进行分类识别。本发明涉及深度学习领域。

技术领域

本发明涉及深度学习领域，特别是指一种家居图像多目标识别方法及装置。

背景技术

近年来，随着人工智能的快速发展和互联网的普及，智慧时代迅速到来，智能家居成为当前的人工智能和物联网的一大热点，家庭中摄像设备数量急剧增加，海量家居图像数据需要处理，一张家居图像中通常含有多个识别目标，通常的图像识别方法只能针对单一目标的图像识别，多目标图像识别成为人工智能的研究重点，也成为制约智能家居发展的难点。

卷积神经网络(Convolutional Neural Networks,CNN)是近些年逐步兴起的一种人工神经网络结构，该技术被广泛的应用于图像识别领域，具有效率高、准确率高和能够处理海量数据等优点。CNN通过卷积、池化等一系列方法，将图像数据降维，方便其被训练，并高效地提取目标特征，因此能够处理庞大的图像数据识别问题。但是现有的CNN图像识别方法需要大量训练样本，并且每次训练需要大量的计算资源和训练时间，且存在多目标识别准确率低的问题。

发明内容

本发明要解决的技术问题是提供一种家居图像多目标识别方法及装置，以解决现有技术所存在的图像识别方法需要大量训练样本、计算资源、训练时间以及多目标识别准确率低的问题。

为解决上述技术问题，本发明实施例提供一种家居图像多目标识别方法，包括：

获取家居图像数据集作为训练集；

将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型，采用迁移学习的方法，保留预训练模型的卷积层和池化层，并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重，通过保留的卷积层和池化层提取训练集的图像特征，构建第一特征图和池化特征图；其中，所述第一特征图是由保留的预训练模型的最后一层卷积层输出，所述池化特征图是由保留的预训练模型的最后一层池化层输出；

对池化特征图进行三次卷积操作，得到第二特征图；

对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作，将反卷积操作后的两组特征图进行融合；

利用softmax分类器对融合后的特征图进行分类识别。

进一步地，所述对池化特征图进行三次卷积操作，得到第二特征图包括：

将卷积神经网络模型的全连接层替换成三层全卷积层；

利用得到的三层全卷积层，对池化特征图进行三次卷积操作，得到第二特征图。

进一步地，全局均值池化操作是将第一特征图的所有像素值相加再除以第一特征图的大小，得到第一特征图的像素平均值；还用于将第二特征图的所有像素值相加再除以第二特征图的大小，得到第二特征图的像素平均值。

进一步地，所述将反卷积操作后的两组特征图进行融合包括：

将反卷积操作后的两组特征图按位相加进行融合。

进一步地，所述利用softmax分类器对融合后的特征图进行分类识别包括：