CN110210538B - 一种家居图像多目标识别方法及装置 - Google Patents
一种家居图像多目标识别方法及装置 Download PDFInfo
- Publication number
- CN110210538B CN110210538B CN201910428783.6A CN201910428783A CN110210538B CN 110210538 B CN110210538 B CN 110210538B CN 201910428783 A CN201910428783 A CN 201910428783A CN 110210538 B CN110210538 B CN 110210538B
- Authority
- CN
- China
- Prior art keywords
- feature map
- layer
- pooling
- feature
- image
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/25—Fusion techniques
- G06F18/253—Fusion techniques of extracted features
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Data Mining & Analysis (AREA)
- Physics & Mathematics (AREA)
- Evolutionary Computation (AREA)
- Artificial Intelligence (AREA)
- General Engineering & Computer Science (AREA)
- Life Sciences & Earth Sciences (AREA)
- General Physics & Mathematics (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Bioinformatics & Computational Biology (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Evolutionary Biology (AREA)
- General Health & Medical Sciences (AREA)
- Computing Systems (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Molecular Biology (AREA)
- Computational Linguistics (AREA)
- Biophysics (AREA)
- Biomedical Technology (AREA)
- Health & Medical Sciences (AREA)
- Image Analysis (AREA)
Abstract
本发明提供一种家居图像多目标识别方法及装置,能够为家居图像训练节省大量训练数据、计算资源和训练时间,且能够提高多目标识别的准确率。所述方法包括:获取家居图像数据集作为训练集;将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图;对池化特征图进行三次卷积操作,得到第二特征图;对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合;利用softmax分类器对融合后的特征图进行分类识别。本发明涉及深度学习领域。
Description
技术领域
本发明涉及深度学习领域,特别是指一种家居图像多目标识别方法及装置。
背景技术
近年来,随着人工智能的快速发展和互联网的普及,智慧时代迅速到来,智能家居成为当前的人工智能和物联网的一大热点,家庭中摄像设备数量急剧增加,海量家居图像数据需要处理,一张家居图像中通常含有多个识别目标,通常的图像识别方法只能针对单一目标的图像识别,多目标图像识别成为人工智能的研究重点,也成为制约智能家居发展的难点。
卷积神经网络(Convolutional Neural Networks,CNN)是近些年逐步兴起的一种人工神经网络结构,该技术被广泛的应用于图像识别领域,具有效率高、准确率高和能够处理海量数据等优点。CNN通过卷积、池化等一系列方法,将图像数据降维,方便其被训练,并高效地提取目标特征,因此能够处理庞大的图像数据识别问题。但是现有的CNN图像识别方法需要大量训练样本,并且每次训练需要大量的计算资源和训练时间,且存在多目标识别准确率低的问题。
发明内容
本发明要解决的技术问题是提供一种家居图像多目标识别方法及装置,以解决现有技术所存在的图像识别方法需要大量训练样本、计算资源、训练时间以及多目标识别准确率低的问题。
为解决上述技术问题,本发明实施例提供一种家居图像多目标识别方法,包括:
获取家居图像数据集作为训练集;
将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重,通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图;其中,所述第一特征图是由保留的预训练模型的最后一层卷积层输出,所述池化特征图是由保留的预训练模型的最后一层池化层输出;
对池化特征图进行三次卷积操作,得到第二特征图;
对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合;
利用softmax分类器对融合后的特征图进行分类识别。
进一步地,所述对池化特征图进行三次卷积操作,得到第二特征图包括:
将卷积神经网络模型的全连接层替换成三层全卷积层;
利用得到的三层全卷积层,对池化特征图进行三次卷积操作,得到第二特征图。
进一步地,全局均值池化操作是将第一特征图的所有像素值相加再除以第一特征图的大小,得到第一特征图的像素平均值;还用于将第二特征图的所有像素值相加再除以第二特征图的大小,得到第二特征图的像素平均值。
进一步地,所述将反卷积操作后的两组特征图进行融合包括:
将反卷积操作后的两组特征图按位相加进行融合。
进一步地,所述利用softmax分类器对融合后的特征图进行分类识别包括:
使用softmax分类器对融合特征图的每一个像素点计算分类损失,将每个像素点在所有第二特征图中该像素位置的概率最高的类别作为该像素点的类别。
本发明实施例还提供一种家居图像多目标识别装置,包括:获取模块和用于家居图像识别的多目标识别卷积神经网络模型;所述多目标识别卷积神经网络模型包括:特征迁移模块、全卷积模块、特征融合模块、分类模块;其中,
获取模块,用于获取家居图像数据集作为训练集;
特征迁移模块,用于将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重,通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图;其中,所述第一特征图是由保留的预训练模型的最后一层卷积层输出,所述池化特征图是由保留的预训练模型的最后一层池化层输出;
全卷积模块,用于对池化特征图进行三次卷积操作,得到第二特征图;
特征融合模块,用于对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合;
分类模块,用于利用softmax分类器对融合后的特征图进行分类识别。
进一步地,所述全卷积模块包括:
替换单元,用于将卷积神经网络模型的全连接层替换成三层全卷积层;
操作单元,用于利用得到的三层全卷积层,对池化特征图进行三次卷积操作,得到第二特征图。
进一步地,全局均值池化操作是将第一特征图的所有像素值相加再除以第一特征图的大小,得到第一特征图的像素平均值;还用于将第二特征图的所有像素值相加再除以第二特征图的大小,得到第二特征图的像素平均值。
进一步地,所述分类模块,用于使用softmax分类器对融合特征图的每一个像素点计算分类损失,将每个像素点在所有第二特征图中该像素位置的概率最高的类别作为该像素点的类别。
本发明的上述技术方案的有益效果如下:
上述方案中,获取家居图像数据集作为训练集;将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重,通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图;对池化特征图进行三次卷积操作,得到第二特征图;对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合;利用softmax分类器对融合后的特征图进行分类识别。这样,通过迁移学习的方法,将在目标领域图像数据集上学到的图像知识应用到家居图像分类问题上,完成目标领域数据集的特征迁移,能够为家居图像训练节省大量训练数据、计算资源和训练时间,且能够提高多目标识别的准确率。
附图说明
图1为本发明实施例提供的家居图像多目标识别方法的流程示意图;
图2为本发明实施例提供的特征融合模块的结构示意图;
图3为本发明实施例提供的MTR-CNN模型的具体结构示意图;
图4为本发明实施例提供的家居图像多目标识别装置的结构示意图。
具体实施方式
为使本发明要解决的技术问题、技术方案和优点更加清楚,下面将结合附图及具体实施例进行详细描述。
本发明针对现有的图像识别方法需要大量训练样本、计算资源、训练时间的问题,提供一种家居图像多目标识别方法及装置。
实施例一
如图1所示,本发明实施例提供的家居图像多目标识别方法,包括:
S101,获取家居图像数据集作为训练集;
S102,将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重,通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图;其中,所述第一特征图是由保留的预训练模型的最后一层卷积层输出,所述池化特征图是由保留的预训练模型的最后一层池化层输出;
S103,对池化特征图进行三次卷积操作,得到第二特征图;
S104,对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合;
S105,利用softmax分类器对融合后的特征图进行分类识别。
本发明实施例所述的家居图像多目标识别方法,获取家居图像数据集作为训练集;将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重,通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图;对池化特征图进行三次卷积操作,得到第二特征图;对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合;利用softmax分类器对融合后的特征图进行分类识别。这样,通过迁移学习的方法,将在目标领域图像数据集上学到的图像知识应用到家居图像分类问题上,完成目标领域数据集的特征迁移,能够为家居图像训练节省大量训练数据、计算资源和训练时间,且能够提高多目标识别的准确率。
本发明实施例所述的家居图像多目标识别方法可以应用于智能家居图像识别,主要用于解决家居图像多目标识别问题,以满足智能家居图像识别多目标、省资源、省时间的要求以及多目标识别准确率高。
为了实现本发明实施例所述的家居图像多目标识别方法,需要获取模块和用于家居图像识别的多目标识别卷积神经网络模型(MTR-CNN);所述多目标识别卷积神经网络模型包括:特征迁移模块、全卷积模块、特征融合模块、分类模块;其中,
获取模块,用于获取家居图像数据集作为训练集;
特征迁移模块,用于将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重,通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图;
全卷积模块,用于对池化特征图进行三次卷积操作,得到第二特征图;
特征融合模块,用于对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合;
分类模块,用于利用softmax分类器对融合后的特征图进行分类识别。
本实施例中,所述目标领域图像数据集为与家居领域相似的图像数据集。
本实施例中,通过特征迁移模块将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重,完成相似领域数据集的特征迁移,在此基础上输入训练集,通过保留的预训练模型中的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图。
在前述家居图像多目标识别方法的具体实施方式中,进一步地,所述对池化特征图进行三次卷积操作,得到第二特征图包括:
将卷积神经网络模型的全连接层替换成三层全卷积层;
利用得到的三层全卷积层,对池化特征图进行三次卷积操作,得到第二特征图。
本实施例中,全卷积模块将原卷积神经网络模型的全连接层替换成卷积核的大小分别为4096×1×1、4096×1×1、1000×1×1的三层全卷积层,全卷积层与特征迁移模块的最后一个池化层相连。
本实施例中,全卷积模块对特征迁移模块输出的池化特征图依次进行三次卷积操作,获得更高维度的第二特征图。
本实施例中,如图2所示,特征融合模块包括全局均值池化层、与全局均值池化层相连的归一化层、与归一化层相连的反池化层、与反池化层相连的反卷积层、与反卷积层相连的融合层五个网络层;其中,
全局均值池化层,用于对接收到的特征图内部取平均值,具体的:全局均值化层是将第一特征图的所有像素值相加再除以第一特征图的大小,得到第一特征图的像素平均值;还用于将第二特征图的所有像素值相加再除以第二特征图的大小,得到第二特征图的像素平均值;
归一化层,用于进行归一化操作将不同特征图的分布统一到一个预设的区间内,例如,可以使用L2归一化函数进行归一化操作,其中,L2归一化函数表示为:
反池化层,用于将归一化层输出的特征图恢复到全局均值池化前的尺寸,将全局均值池化层得到的平均值填入特征图的每个位置;
反卷积层,用于将反池化层输出的图像恢复到输入特征迁移模块时的尺寸(即:原图尺寸),具体的:对特征迁移模块输出的经全局均值池化、归一化、反池化处理得到的特征图进行16倍上采样,对全卷积模块输出的经全局均值池化、归一化、反池化处理得到的特征图进行32倍上采样;
融合层,用于将特征迁移模块输出的和全卷积模块输出的依次经全局均值池化、归一化、反池化、反卷积处理后得到的两组特征图,按位相加得到最后的融合结果。
本实施例中,特征融合模块与特征迁移模块的最后一个卷积层和全卷积模块的最后一个卷积层相连。
本实施例中,特征融合模块对特征迁移模块提取的第一特征图和全卷积模块提取的第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合,使得特征经上采样恢复后的细节更加完善,能够提高家居图像的识别精度。
在前述家居图像多目标识别方法的具体实施方式中,进一步地,所述利用softmax分类器对融合后的特征图进行分类识别包括:
使用softmax分类器对融合特征图的每一个像素点计算分类损失,将每个像素点在所有第二特征图中该像素位置的概率最高的类别作为该像素点的类别;
本实施例中,分类模块用于将恢复原图尺寸并经过特征融合模块融合后的特征图输入softmax分类器,利用softmax分类器对融合后的特征图的每一个像素点进行分类操作,将概率最高的类别作为该像素点的类别,得到最终的识别结果;这样,实现了对图像进行像素级的分类,解决了语义级别的图像分割问题,即多目标识别问题,从而能够提高家居图像多目标识别的准确率,并且可以接受任意尺寸的输入图像。
本实施例中,如图3所示,对本发明实施例所述的家居图像多目标识别方法进行详细说明,所述方法具体可以包括以下步骤:
A11,获取少量家居图像,对获取的家居图像进行图像预处理,包括通过旋转、平移、缩放、边缘填充等操作增大训练样本数,通过平滑、对比度增强等提高图像质量,将预处理后的家居图像作为训练集,并给训练集添加用于分类的标签;
本实施例中,标签是指给训练集中每一张图片加上一个类别识别标志,例如,餐桌、椅子等,标签在下文中用于图像分类任务,对于一个给定的待识别图像,计算其特征图与各分类标准特征图的损失距离,预测它属于的那个分类标签。
A12,将在图像识别数据库(ImageNet)2012数据集中的家居图像数据上预先训练好的卷积神经网络模型作为特征预训练模型,使用该模型训练学到的图像特征权重作为训练前的初始特征权重;
本实施例所述的家居图像多目标识别方法包括训练和验证两个过程,训练过程的初始特征权重为预训练模型的保留的特征权重,验证过程的初始特征权重为MTR-CNN模型经训练集训练后保留的特征权重。
A13,输入训练集,经卷积神经网络模型中的卷积层和池化层处理,将训练集中的图像数据不断降维,提取特征向量,构建第一特征图和池化特征图;A14,将预训练模型的全连接层替换为三层全卷积层,将池化特征图作为全卷积层的输入,通过三次卷积操作提取更深层次特征,得到更高维度的第二特征图;
A15,将卷积神经网络模型最后一个卷积层输出的第一特征图经过特征融合模块,依次作全局均值池化、归一化和反池化处理,再进行反卷积操作,作16倍上采样(放大16倍)恢复成原图尺寸;
A16,将全卷积层输出的第二特征图经过特征融合模块,依次作全局均值池化、归一化和反池化处理,再进行反卷积操作,作32倍上采样(放大32倍)恢复成原图尺寸;
A17,将A15、A16得到的两组特征图按位相加进行融合,获得更精细的特征图;
A18,将A17输出的融合特征图输入分类模块,使用softmax分类器对融合特征图的每一个像素点计算分类损失,将每个像素点在所有第二特征图中该像素位置的概率最高的类别作为该像素点的类别。
在A18中,通过对融合特征图的像素点分类可以将一张图像划分为多个不同标签类别的区域图像,从而实现对家居图像的多目标识别。
A19,通过反向传播更新卷积神经网络模型的特征权重,并判断当前迭代次数是否等于预设的最大迭代次数;若等于,则训练结束,得到训练好的MTR-CNN模型,否则,则返回步骤A13继续执行。
本实施例中,选用的预训练模型是在ImageNet数据集上预先训练好的卷积神经网络模型,具体的,所述预训练模型可以是可视化几何组卷积神经网络模型VGG-16,也可以是其他卷积神经网络模型。
本实施例中,采用验证集验证训练得到的MTR-CNN模型,所述验证集为PASCAL VOC2012数据集的一部分,该数据集包含了室内243种生活中常见的物品,例如沙发、电视、餐桌、盆栽、人等。数据集总共为1020张图像,每张图像大小不一。
将本申请所述的MTR-CNN与现有的多目标检测模型全卷积神经网络(FullyConvolutional Networks,简称FCN)分别进行多目标图像识别,得到的实验结果如表1所示:
表1MTR-CNN与FCN-8s、FCN-16s、FCN-32s的识别正确率对比
类型 | FCN-8s | FCN-16s | FCN-32s | MTR-CNN |
餐桌 | 63.7 | 62.1 | 59.8 | 68.3 |
沙发 | 58.4 | 55.7 | 54.1 | 63.8 |
椅子 | 70.6 | 67.0 | 62.4 | 77.3 |
床 | 68.2 | 66.5 | 63.1 | 73.6 |
瓶子 | 76.0 | 73.7 | 69.2 | 80.1 |
表1中的实验结果表明,本发明提供的家居图像多目标识别方法能够完成家居图像的多目标识别,并且本发明确定的MTR-CNN模型相比现有的多目标检测模型FCN-8s、FCN-16s和FCN-32s,分类性能要更好,正确率要高5%-15%。
综上,本发明实施例所述的家居图像多目标识别方法具有迁移学习省资源、省时间的优点,并对输入图像尺寸无要求,能够实现家居图像多目标识别且正确率较高。
实施例二
本发明还提供一种家居图像多目标识别装置的具体实施方式,由于本发明提供的家居图像多目标识别装置与前述家居图像多目标识别方法的具体实施方式相对应,该家居图像多目标识别装置可以通过执行上述方法具体实施方式中的流程步骤来实现本发明的目的,因此上述家居图像多目标识别方法具体实施方式中的解释说明,也适用于本发明提供的家居图像多目标识别装置的具体实施方式,在本发明以下的具体实施方式中将不再赘述。
如图4所示,本发明实施例还提供一种家居图像多目标识别装置,包括:获取模块1和用于家居图像识别的多目标识别卷积神经网络模型2;所述多目标识别卷积神经网络模型2包括:特征迁移模块21、全卷积模块22、特征融合模块23、分类模块24;其中,
获取模块1,用于获取家居图像数据集作为训练集;
特征迁移模块21,用于将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重,通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图;其中,所述第一特征图是由保留的预训练模型的最后一层卷积层输出,所述池化特征图是由保留的预训练模型的最后一层池化层输出;
全卷积模块22,用于对池化特征图进行三次卷积操作,得到第二特征图;
特征融合模块23,用于对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合;
分类模块24,用于利用softmax分类器对融合后的特征图进行分类识别。
本发明实施例所述的家居图像多目标识别装置,获取家居图像数据集作为训练集;将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重,通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图;对池化特征图进行三次卷积操作,得到第二特征图;对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合;利用softmax分类器对融合后的特征图进行分类识别。这样,通过迁移学习的方法,将在目标领域图像数据集上学到的图像知识应用到家居图像分类问题上,完成目标领域数据集的特征迁移,能够为家居图像训练节省大量训练数据、计算资源和训练时间,且能够提高多目标识别的准确率。
本实施例中,所述特征迁移模块包括:
迁移单元,用于实现目标领域特征权重的迁移,具体的:将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重;
构建单元,用于实现第一特征图和池化特征图的构建,具体的:通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图。
在前述家居图像多目标识别装置的具体实施方式中,进一步地,所述全卷积模块包括:
替换单元,用于将卷积神经网络模型的全连接层替换成三层全卷积层;
操作单元,用于利用得到的三层全卷积层,对池化特征图进行三次卷积操作,得到第二特征图。
在前述家居图像多目标识别装置的具体实施方式中,进一步地,全局均值池化操作是将第一特征图的所有像素值相加再除以第一特征图的大小,得到第一特征图的像素平均值;还用于将第二特征图的所有像素值相加再除以第二特征图的大小,得到第二特征图的像素平均值。
在前述家居图像多目标识别装置的具体实施方式中,进一步地,所述分类模块,用于使用softmax分类器对融合特征图的每一个像素点计算分类损失,将每个像素点在所有第二特征图中该像素位置的概率最高的类别作为该像素点的类别。
需要说明的是,在本文中,诸如第一和第二等之类的关系术语仅仅用来将一个实体或者操作与另一个实体或操作区分开来,而不一定要求或者暗示这些实体或操作之间存在任何这种实际的关系或者顺序。
以上所述是本发明的优选实施方式,应当指出,对于本技术领域的普通技术人员来说,在不脱离本发明所述原理的前提下,还可以做出若干改进和润饰,这些改进和润饰也应视为本发明的保护范围。
Claims (4)
1.一种家居图像多目标识别方法,其特征在于,包括:
获取家居图像数据集作为训练集;
将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重,通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图;其中,所述第一特征图是由保留的预训练模型的最后一层卷积层输出,所述池化特征图是由保留的预训练模型的最后一层池化层输出;
对池化特征图进行三次卷积操作,得到第二特征图,包括:
将卷积神经网络模型的全连接层替换成三层全卷积层;
利用得到的三层全卷积层,对池化特征图进行三次卷积操作,得到第二特征图;
对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合;
其中,全局均值池化操作包括:将第一特征图的所有像素值相加再除以第一特征图的大小,得到第一特征图的像素平均值;还将第二特征图的所有像素值相加再除以第二特征图的大小,得到第二特征图的像素平均值;
归一化步骤包括:进行归一化操作将不同特征图的分布统一到一个预设的区间内,使用L2归一化函数进行归一化操作,其中,L2归一化函数表示为:
反池化步骤包括:将归一化输出的特征图恢复到全局均值池化前的尺寸,将全局均值池化层得到的平均值填入特征图的每个位置;
反卷积操作包括:将反池化层输出的图像恢复到输入特征迁移模块时的尺寸,即原图尺寸,具体的:对特征迁移步骤输出的经全局均值池化、归一化、反池化处理得到的特征图进行16倍上采样,对全卷积步骤输出的经全局均值池化、归一化、反池化处理得到的特征图进行32倍上采样;
融合步骤包括:将反卷积处理后得到的两组特征图,按位相加得到最后的融合结果;
利用softmax分类器对融合后的特征图进行分类识别。
2.根据权利要求1所述的家居图像多目标识别方法,其特征在于,所述利用softmax分类器对融合后的特征图进行分类识别包括:
使用softmax分类器对融合特征图的每一个像素点计算分类损失,将每个像素点在所有第二特征图中该像素位置的概率最高的类别作为该像素点的类别。
3.一种家居图像多目标识别装置,其特征在于,包括:获取模块和用于家居图像识别的多目标识别卷积神经网络模型;所述多目标识别卷积神经网络模型包括:特征迁移模块、全卷积模块、特征融合模块、分类模块;其中,
获取模块,用于获取家居图像数据集作为训练集;
特征迁移模块,用于将在目标领域图像数据集上训练好的卷积神经网络模型作为预训练模型,采用迁移学习的方法,保留预训练模型的卷积层和池化层,并将预训练模型中卷积层和池化层学习到的特征权重作为训练集训练的初始特征权重,通过保留的卷积层和池化层提取训练集的图像特征,构建第一特征图和池化特征图;其中,所述第一特征图是由保留的预训练模型的最后一层卷积层输出,所述池化特征图是由保留的预训练模型的最后一层池化层输出;
全卷积模块,用于对池化特征图进行三次卷积操作,得到第二特征图;
所述全卷积模块包括:
替换单元,用于将卷积神经网络模型的全连接层替换成三层全卷积层;
操作单元,用于利用得到的三层全卷积层,对池化特征图进行三次卷积操作,得到第二特征图;
特征融合模块,用于对第一特征图和第二特征图依次作全局均值池化、归一化、反池化、反卷积操作,将反卷积操作后的两组特征图进行融合;
特征融合模块包括全局均值池化层、与全局均值池化层相连的归一化层、与归一化层相连的反池化层、与反池化层相连的反卷积层、与反卷积层相连的融合层五个网络层;其中,
全局均值池化层,用于对接收到的特征图内部取平均值,具体的:全局均值化层是将第一特征图的所有像素值相加再除以第一特征图的大小,得到第一特征图的像素平均值;还用于将第二特征图的所有像素值相加再除以第二特征图的大小,得到第二特征图的像素平均值;
归一化层,用于进行归一化操作将不同特征图的分布统一到一个预设的区间内,使用L2归一化函数进行归一化操作,其中,L2归一化函数表示为:
反池化层,用于将归一化层输出的特征图恢复到全局均值池化前的尺寸,将全局均值池化层得到的平均值填入特征图的每个位置;
反卷积层,用于将反池化层输出的图像恢复到输入特征迁移模块时的尺寸,即原图尺寸,具体的:对特征迁移模块输出的经全局均值池化、归一化、反池化处理得到的特征图进行16倍上采样,对全卷积模块输出的经全局均值池化、归一化、反池化处理得到的特征图进行32倍上采样;
融合层,用于将特征迁移模块输出的和全卷积模块输出的依次经全局均值池化、归一化、反池化、反卷积处理后得到的两组特征图,按位相加得到最后的融合结果;
分类模块,用于利用softmax分类器对融合后的特征图进行分类识别。
4.根据权利要求3所述的家居图像多目标识别装置,其特征在于,所述分类模块,用于使用softmax分类器对融合特征图的每一个像素点计算分类损失,将每个像素点在所有第二特征图中该像素位置的概率最高的类别作为该像素点的类别。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201910428783.6A CN110210538B (zh) | 2019-05-22 | 2019-05-22 | 一种家居图像多目标识别方法及装置 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201910428783.6A CN110210538B (zh) | 2019-05-22 | 2019-05-22 | 一种家居图像多目标识别方法及装置 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN110210538A CN110210538A (zh) | 2019-09-06 |
CN110210538B true CN110210538B (zh) | 2021-10-19 |
Family
ID=67788303
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN201910428783.6A Active CN110210538B (zh) | 2019-05-22 | 2019-05-22 | 一种家居图像多目标识别方法及装置 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN110210538B (zh) |
Families Citing this family (8)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN110689077A (zh) * | 2019-09-29 | 2020-01-14 | 福建师范大学 | 一种新型数字图像分类方法 |
CN111401526A (zh) * | 2020-03-20 | 2020-07-10 | 厦门渊亭信息科技有限公司 | 一种模型通用的深度神经网络表征可视化方法和装置 |
CN111612855B (zh) * | 2020-04-09 | 2024-06-14 | 北京旷视科技有限公司 | 物体颜色识别方法、装置及电子设备 |
CN111539325A (zh) * | 2020-04-23 | 2020-08-14 | 四川旅游学院 | 基于深度学习的森林火灾探测方法 |
CN112465053A (zh) * | 2020-12-07 | 2021-03-09 | 深圳市彬讯科技有限公司 | 基于家具图像的物体识别方法、装置、设备及存储介质 |
CN112560625B (zh) * | 2020-12-09 | 2022-09-23 | 河海大学 | 基于时空联合响应网络的智能家居***及其控制方法 |
CN112926441B (zh) * | 2021-02-24 | 2021-11-30 | 广州极点三维信息科技有限公司 | 基于机器视觉的家居识别方法、***、装置及介质 |
CN113283308A (zh) * | 2021-04-30 | 2021-08-20 | 杭州品茗安控信息技术股份有限公司 | 一种工程图纸的图像识别方法、***及相关装置 |
Citations (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108281183A (zh) * | 2018-01-30 | 2018-07-13 | 重庆大学 | 基于卷积神经网络和迁移学习的宫颈涂片图像诊断*** |
CN108509978A (zh) * | 2018-02-28 | 2018-09-07 | 中南大学 | 基于cnn的多级特征融合的多类目标检测方法及模型 |
CN108520219A (zh) * | 2018-03-30 | 2018-09-11 | 台州智必安科技有限责任公司 | 一种卷积神经网络特征融合的多尺度快速人脸检测方法 |
CN109615016A (zh) * | 2018-12-20 | 2019-04-12 | 北京理工大学 | 一种基于金字塔输入增益的卷积神经网络的目标检测方法 |
Family Cites Families (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108734211B (zh) * | 2018-05-17 | 2019-12-24 | 腾讯科技(深圳)有限公司 | 图像处理的方法和装置 |
CN108898078A (zh) * | 2018-06-15 | 2018-11-27 | 上海理工大学 | 一种多尺度反卷积神经网络的交通标志实时检测识别方法 |
-
2019
- 2019-05-22 CN CN201910428783.6A patent/CN110210538B/zh active Active
Patent Citations (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108281183A (zh) * | 2018-01-30 | 2018-07-13 | 重庆大学 | 基于卷积神经网络和迁移学习的宫颈涂片图像诊断*** |
CN108509978A (zh) * | 2018-02-28 | 2018-09-07 | 中南大学 | 基于cnn的多级特征融合的多类目标检测方法及模型 |
CN108520219A (zh) * | 2018-03-30 | 2018-09-11 | 台州智必安科技有限责任公司 | 一种卷积神经网络特征融合的多尺度快速人脸检测方法 |
CN109615016A (zh) * | 2018-12-20 | 2019-04-12 | 北京理工大学 | 一种基于金字塔输入增益的卷积神经网络的目标检测方法 |
Non-Patent Citations (3)
Title |
---|
Fully Convolutional Networks for Semantic Segmentation;Jonathan Long et al.;《arXiv:1411.4038v2》;20150308;第1-10页 * |
基于全卷积神经网络的林区航拍图像虫害区域识别方法;刘文定 等;《农业机械学报》;20190331;第50卷(第3期);第1-2节,图4 * |
结合全卷积神经网络的室内场景分割;黄龙 等;《中国图象图形学报》;20190131;第24卷(第1期);摘要 * |
Also Published As
Publication number | Publication date |
---|---|
CN110210538A (zh) | 2019-09-06 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN110210538B (zh) | 一种家居图像多目标识别方法及装置 | |
CN108960141B (zh) | 基于增强型深度卷积神经网络的行人再识别方法 | |
CN111753828B (zh) | 一种基于深度卷积神经网络的自然场景水平文字检测方法 | |
CN113468227B (zh) | 基于图神经网络的信息推荐方法、***、设备和存储介质 | |
CN110287777B (zh) | 一种自然场景下的金丝猴躯体分割算法 | |
CN112668579A (zh) | 基于自适应亲和力和类别分配的弱监督语义分割方法 | |
CN112651940B (zh) | 基于双编码器生成式对抗网络的协同视觉显著性检测方法 | |
CN110675421B (zh) | 基于少量标注框的深度图像协同分割方法 | |
CN110555461A (zh) | 基于多结构卷积神经网络特征融合的场景分类方法及*** | |
CN116503676A (zh) | 一种基于知识蒸馏小样本增量学习的图片分类方法及*** | |
CN104881852A (zh) | 基于免疫克隆和模糊核聚类的图像分割方法 | |
CN116152554A (zh) | 基于知识引导的小样本图像识别*** | |
CN111079930B (zh) | 数据集质量参数的确定方法、装置及电子设备 | |
CN114818963A (zh) | 一种基于跨图像特征融合的小样本检测算法 | |
CN112906800B (zh) | 基于图像组自适应的协同显著性检测方法 | |
CN114693923A (zh) | 一种基于上下文和注意力的三维点云语义分割方法 | |
CN114333062A (zh) | 基于异构双网络和特征一致性的行人重识别模型训练方法 | |
CN110991247B (zh) | 一种基于深度学习与nca融合的电子元器件识别方法 | |
CN116883746A (zh) | 一种基于分区池化超图神经网络的图节点分类方法 | |
CN111783688A (zh) | 一种基于卷积神经网络的遥感图像场景分类方法 | |
CN116433909A (zh) | 基于相似度加权多教师网络模型的半监督图像语义分割方法 | |
CN101504723A (zh) | 一种建立投影空间的方法和装置 | |
CN109977777B (zh) | 基于新型RF-Net模型的手势识别方法 | |
CN114332715A (zh) | 气象自动观测积雪识别方法、装置、设备及存储介质 | |
CN114842330B (zh) | 一种多尺度背景感知池化弱监督建筑物提取方法 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |