CN111401253B - 一种基于深度学习的目标检测方法 - Google Patents
一种基于深度学习的目标检测方法 Download PDFInfo
- Publication number
- CN111401253B CN111401253B CN202010187584.3A CN202010187584A CN111401253B CN 111401253 B CN111401253 B CN 111401253B CN 202010187584 A CN202010187584 A CN 202010187584A CN 111401253 B CN111401253 B CN 111401253B
- Authority
- CN
- China
- Prior art keywords
- target detection
- background
- deep learning
- target
- network model
- 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
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V20/00—Scenes; Scene-specific elements
- G06V20/20—Scenes; Scene-specific elements in augmented reality scenes
-
- 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
-
- 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/08—Learning methods
- G06N3/084—Backpropagation, e.g. using gradient descent
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V2201/00—Indexing scheme relating to image or video recognition or understanding
- G06V2201/07—Target detection
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- General Health & Medical Sciences (AREA)
- General Engineering & Computer Science (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Artificial Intelligence (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- Biomedical Technology (AREA)
- Life Sciences & Earth Sciences (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Health & Medical Sciences (AREA)
- Multimedia (AREA)
- Image Analysis (AREA)
- Image Processing (AREA)
Abstract
本发明实施例公开了一种基于深度学习的目标检测方法,涉及深度学习领域,所述方法包括:创建目标检测网络模型;利用所述目标检测网络模型对输入图像进行背景筛除;基于所述背景去除的图像进行目标识别。本发明能够解决现有深度学习技术中需要消耗大量的计算资源和时间来进行参数训练,导致目标检测效率低的问题。
Description
技术领域
本发明实施例涉及深度学习领域,具体涉及一种基于深度学习的目标检测方法。
背景技术
近年来,我们见证了深度学习的巨大进步和成功,它在图像分类和目标检测等计算机视觉领域带来了巨大的飞跃。工业界已经开始使用深度学习来开发诸如自动驾驶、监控系统、虚拟助手和医疗保健等人工智能软件和服务,但是训练和运行深度学习模型需要大量的计算能力,它限制了深度学习的可扩展性和适用性,特别是在资源受限的环境中。
为了克服这个问题,已经投入了大量的精力来提高深度学习模型在硬件和软件方面的效率。为了提高深部神经网络的性能,提出了处理零值的硬件设计、处理稀疏性和高精度异常值的硬件设计、片上卷积神经网络(CNN)模型和专用加速器。从软件加速的角度看,它包括轻量化网络设计、网络剪枝和网络量化,目标检测(OD)是计算机视觉领域最流行的应用之一。虽然在开发高效的OD网络方面已经投入了大量的努力,但是最先进的模型仍然需要巨大的计算量,当目标环境中消耗的资源量很大时,实际工作者往往不得不选择一个精度更高或速度更快的特征抽取器,但同时也增加了计算成本。
发明内容
本发明实施例的目的在于提供一种基于深度学习的目标检测方法,用以解决现有深度学习技术中需要消耗大量的计算资源和时间来进行参数训练,导致目标检测效率低的问题。
为实现上述目的,本发明实施例主要提供如下技术方案:
本发明实施例提供了一种基于深度学习的目标检测方法,
所述方法包括:创建目标检测网络模型;利用所述目标检测网络模型对输入图像进行背景筛除;基于所述背景去除的图像进行目标识别。
优选地,所述目标检测网络模型包括背景筛除模块和目标检测模块,所述背景筛除模块用去除图像背景中的像素;所述目标检测模块包括特征提取网络和锚头网络,用于提取目标特征。
优选地,所述背景筛除模块采用目标掩码生成网络,所述目标检测模块采用单点多盒探测器作为基础架构。
优选地,对所述输入图像进行背景筛除的方法,具体包括:利用背景筛除模块获取输入图像并生成对象掩码;利用所述对象掩码对所有图像的输入特征进行遮罩,生成像素为零的背景和像素为1的前景。
优选地,所述包括:对输入特征进行遮罩时,当特征映射的形状由于合并和跨步卷积操作而发生变化时,在遮罩时应用最大池化层,使其形状始终与相应的特征映射形状相匹配。
优选地,对所有图像的输入特征进行遮罩,具体包括:将对象掩码与目标检测模块中的各层进行元素相乘来筛除背景区域。
优选地,所述方法包括:通过端到端的方式集成所述目标检测网络模型。
优选地,所述集成方式包括利用代理梯度的方法训练端到端网络模型。
本发明实施例提供的技术方案至少具有如下优点:
本发明通过将通过绕行提高检测速度的方法,将目标所在的背景区域像素做清除处理,创建一种新的目标检测模型,该模型通过对稀疏卷积背景区域的跳过运算来提高检测性能,并以端到端的方式集成它们。该方法在保持与原模型相当的检测精度的同时,大大降低了计算量。
附图说明
图1为本发明实施例提供的一种基于深度学习的目标检测方法流程图。
图2为本发明实施例提供的目标检测网络模型框架图。
具体实施方式
以下由特定的具体实施例说明本发明的实施方式,熟悉此技术的人士可由本说明书所揭露的内容轻易地了解本发明的其他优点及功效。
以下描述中,为了说明而不是为了限定,提出了诸如特定系统结构、接口、技术之类的具体细节,以便透彻理解本发明。然而,本领域的技术人员应当清楚,在没有这些具体细节的其它实施例中也可以实现本发明。在其它情况中,省略对众所周知的系统、电路以及方法的详细说明,以免不必要的细节妨碍本发明的描述。
在SSD、YOLO等一级检测器中,将卷积骨干网的特征输入到子网中进行目标分类和边界盒回归,这些一级检测器旨在通过直接对预定义锚进行分类并使用CNNs对其进行细化,而无需生成建议步骤,从而提高效率,然而,它仍然需要计算所有的锚。
基于深度卷积神经网络模型拥有大量的参数,对于数据有强大的拟合能力。但因此也需要消耗大量的计算资源和时间来进行参数训练,而且这种消耗跟输入图像的尺寸成正相关,即图像输入像素信息越多,消耗越大。由此可知减少像素的处理量可以实现特征识别的加速。
基于此,本发明实施例提供一种基于深度学习的目标检测方法,参考图1,该方法包括:
S1、创建目标检测网络模型;
目标检测网络模型包括背景筛除模块和目标检测模块,背景筛除模块用去除图像背景中的像素,背景筛除模块采用目标掩码生成网络Objectness Mask GenerationNetwork,即OMGN。
目标检测模块采用单点多盒探测器SSD作为基础架构,包括特征提取网络和锚头网络两个子模块,用于提取目标特征。特征提取网络即Feature Extraction Network,下文称为FEN。锚头网络即Anchor Head Network,下文称为AHN。每个锚定位置的特征映射被输入AHN以确定对象类并定位其边界框。
S2、利用目标检测网络模型对输入图像进行背景筛除;
目标检测是图像处理里最基本也最重要的任务,在检测任务中模型需要将目标从图像中提取出来,目标区域被称为前景,而剩余的非目标区域被称为背景。本实施例主动对背景区域进行筛除,缩小目标检测的观察区域,进而降低计算量并提升效率。
具体地,参考图2,在本实施例一个可行的方法中,利用背景筛除模块获取输入图像并生成对象掩码,其中背景像素的值为零,前景像素的值为1。这项任务将图像中的每个像素分为两类,可以看作是语义分割的简化版本。对象遮罩的形状可以是任意形状,如分段遮罩,也可以是框,具体取决于批注数据的可用性。快速SCNN是一种快速、轻量级的语义分割模型,且是一个完全卷积的网络可以通过改变输入图像的大小来轻松控制其计算成本。因此采用SCNN作为OMG网络。
利用对象掩码对所有图像的输入特征进行遮罩,生成像素为零的背景和像素为1的前景。具体地,输入图像和OD网络卷积层的所有输入特征现在都用一个二值对象遮罩,该遮罩将背景区域中的像素值归零。如图2所示,该遮罩操作为上图中表示为符号的按元素的乘法层。需要说明的是,当特征映射的形状由于合并和跨步卷积等操作而发生变化时,会对遮罩应用最大池化层max pooling,为使其形状始终与相应的特征映射的形状匹配,图2中具有阴影的方块表示最大池化。
S3、基于背景去除的图像进行目标识别。
通过端到端的方式集成目标检测网络模型,进行目标识别。其中,由于OMG网络通过argmax函数为每个输入像素生成零和一两个离散值中的一个,又由于argmax函数是不可微的,梯度几乎总是零,用标准的反向传播方法训练端到端模型是不可能的。
因此,本实施例采用代理梯度的方法训练端到端网络模型。
该方法使用了一种近似于原始激活函数梯度方向的后向传递的代理求导函数,表示为:
具有适于反向传播的可微和非平凡梯度值的作用,当选择合适的代理导数时,训练过程收敛于原始的局部极小值附近。
由此可见,本实施例通过将通过绕行提高检测速度的方法,将目标所在的背景区域像素做清除处理,创建一种新的目标检测模型,该模型通过对稀疏卷积背景区域的跳过运算来提高检测性能,并以端到端的方式集成它们。该方法在保持与原模型相当的检测精度的同时,大大降低了计算量。
本发明所公开的实施例提供一种计算机可读存储介质,所述计算机可读存储介质中存储有计算机程序指令,当所述计算机程序指令在计算机上运行时,使得计算机执行上述的方法。
以上所述的具体实施方式,对本发明的目的、技术方案和有益效果进行了进一步详细说明,所应理解的是,以上所述仅为本发明的具体实施方式而已,并不用于限定本发明的保护范围,凡在本发明的技术方案的基础之上,所做的任何修改、等同替换、改进等,均应包括在本发明的保护范围之内。
Claims (4)
1.一种基于深度学习的目标检测方法,其特征在于,所述方法包括:
创建目标检测网络模型;通过端到端的方式集成所述目标检测网络模型,并利用代理梯度的方法训练网络模型;
利用所述目标检测网络模型对输入图像进行背景筛除;
基于所述背景去除的图像进行目标识别;
所述目标检测网络模型包括背景筛除模块和目标检测模块,其中,背景筛除模块用去除图像背景中的像素;目标检测模块包括特征提取网络和锚头网络,用于提取目标特征;
所述背景筛除模块采用目标掩码生成网络,具体利用快速SCNN模型;所述目标检测模块采用单点多盒探测器作为基础架构。
2.如权利要求1所述的一种基于深度学习的目标检测方法,其特征在于,对所述输入图像进行背景筛除的方法,具体包括:
利用背景筛除模块获取输入图像并生成对象掩码;
利用所述对象掩码对所有图像的输入特征进行遮罩,生成像素为零的背景和像素为1的前景。
3.如权利要求2所述的一种基于深度学习的目标检测方法,其特征在于,所述包括:对输入特征进行遮罩时,当特征映射的形状由于合并和跨步卷积操作而发生变化时,在遮罩时应用最大池化层,使其形状始终与相应的特征映射形状相匹配。
4.如权利要求2所述的一种基于深度学习的目标检测方法,其特征在于,对所有图像的输入特征进行遮罩,具体包括:
将对象掩码与目标检测模块中的各层进行元素相乘来筛除背景区域。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010187584.3A CN111401253B (zh) | 2020-03-17 | 2020-03-17 | 一种基于深度学习的目标检测方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010187584.3A CN111401253B (zh) | 2020-03-17 | 2020-03-17 | 一种基于深度学习的目标检测方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN111401253A CN111401253A (zh) | 2020-07-10 |
CN111401253B true CN111401253B (zh) | 2022-09-13 |
Family
ID=71432580
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202010187584.3A Active CN111401253B (zh) | 2020-03-17 | 2020-03-17 | 一种基于深度学习的目标检测方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN111401253B (zh) |
Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN106529565A (zh) * | 2016-09-23 | 2017-03-22 | 北京市商汤科技开发有限公司 | 目标识别模型训练和目标识别方法及装置、计算设备 |
CN106815579A (zh) * | 2017-01-22 | 2017-06-09 | 深圳市唯特视科技有限公司 | 一种基于多区域双流卷积神经网络模型的动作检测方法 |
CN110188835A (zh) * | 2019-06-05 | 2019-08-30 | 国家广播电视总局广播电视科学研究院 | 基于生成式对抗网络模型的数据增强行人再识别方法 |
Family Cites Families (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US9008359B2 (en) * | 2012-06-28 | 2015-04-14 | International Business Machines Corporation | Detection of static object on thoroughfare crossings |
CN106780536A (zh) * | 2017-01-13 | 2017-05-31 | 深圳市唯特视科技有限公司 | 一种基于对象掩码网络的形状感知实例分割方法 |
CN108961293B (zh) * | 2018-06-04 | 2021-01-19 | 国光电器股份有限公司 | 一种背景减除的方法、装置、设备及存储介质 |
CN109173263B (zh) * | 2018-08-31 | 2021-08-24 | 腾讯科技(深圳)有限公司 | 一种图像数据处理方法和装置 |
CN109325490B (zh) * | 2018-09-30 | 2021-04-27 | 西安电子科技大学 | 基于深度学习和rpca的太赫兹图像目标识别方法 |
CN110490073A (zh) * | 2019-07-15 | 2019-11-22 | 浙江省北大信息技术高等研究院 | 目标检测方法、装置、设备及存储介质 |
-
2020
- 2020-03-17 CN CN202010187584.3A patent/CN111401253B/zh active Active
Patent Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN106529565A (zh) * | 2016-09-23 | 2017-03-22 | 北京市商汤科技开发有限公司 | 目标识别模型训练和目标识别方法及装置、计算设备 |
CN106815579A (zh) * | 2017-01-22 | 2017-06-09 | 深圳市唯特视科技有限公司 | 一种基于多区域双流卷积神经网络模型的动作检测方法 |
CN110188835A (zh) * | 2019-06-05 | 2019-08-30 | 国家广播电视总局广播电视科学研究院 | 基于生成式对抗网络模型的数据增强行人再识别方法 |
Also Published As
Publication number | Publication date |
---|---|
CN111401253A (zh) | 2020-07-10 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
Chen et al. | Weakly supervised semantic segmentation with boundary exploration | |
CN110322495B (zh) | 一种基于弱监督深度学习的场景文本分割方法 | |
CN113221905B (zh) | 基于均匀聚类的语义分割的无监督域适应方法、装置、系统和存储介质 | |
CN111461212A (zh) | 一种用于点云目标检测模型的压缩方法 | |
CN112200131A (zh) | 一种基于视觉的车辆碰撞检测方法、智能终端及存储介质 | |
Guillermo et al. | Implementation of automated annotation through mask rcnn object detection model in cvat using aws ec2 instance | |
Liunanda et al. | Implementasi Algoritma YOLO pada Aplikasi Pendeteksi Senjata Tajam di Android. | |
CN115393598A (zh) | 一种基于非显著区域对象挖掘的弱监督语义分割方法 | |
CN114495041A (zh) | 车辆与目标物之间距离的测量方法、装置、设备及介质 | |
CN111401253B (zh) | 一种基于深度学习的目标检测方法 | |
Wang et al. | Semantic segmentation of sewer pipe defects using deep dilated convolutional neural network | |
Nguyen et al. | Smart solution to detect images in limited visibility conditions based convolutional neural networks | |
Ding et al. | Object as distribution | |
WO2021237727A1 (en) | Method and apparatus of image processing | |
CN114821190A (zh) | 图像分类模型训练方法、图像分类方法、装置及设备 | |
Kim et al. | Location-based saliency maps from a fully connected layer using multi-shapes | |
Das et al. | Object Detection on Scene Images: A Novel Approach | |
Xu et al. | Deep Neural Network‐Based Sports Marketing Video Detection Research | |
Cao et al. | Semantic-aware context modeling for road extraction in remote sensing images | |
Vimala et al. | Convolutional Neural Network-Based Automatic Object Detection on Aerial Images | |
He et al. | Are Deep Point Cloud Classifiers Suffer From Out-of-distribution Overconfidence Issue? | |
CN112446266B (zh) | 一种适合前端的人脸识别网络结构 | |
CN112446267B (zh) | 一种适合前端的人脸识别网络的设置方法 | |
EP4131178A1 (en) | Image classification method and apparatus, and method and apparatus for improving training of an image classifier | |
CN111401319B (zh) | PSENet网络渐进式扩展后处理出现文本粘连的解决方法 |
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 |