CN109902556A - 行人检测方法、系统、计算机设备及计算机可存储介质 - Google Patents
行人检测方法、系统、计算机设备及计算机可存储介质 Download PDFInfo
- Publication number
- CN109902556A CN109902556A CN201910031726.4A CN201910031726A CN109902556A CN 109902556 A CN109902556 A CN 109902556A CN 201910031726 A CN201910031726 A CN 201910031726A CN 109902556 A CN109902556 A CN 109902556A
- Authority
- CN
- China
- Prior art keywords
- convolutional layer
- pedestrian
- convolution
- layer
- pond
- 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.)
- Pending
Links
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
Abstract
本发明实施例提供了一种行人检测方法,所述方法包括:将待检测图片输入到所述YOLO神经网络;通过所述YOLO神经网络对所述待检测图片执行卷积操作和池化操作;通过预设的B个行人预测框在所述2A个x*y卷积特征图上做卷积操作,输出x*y*B*(5+C)的输出张量,C表示C个条件类别概率;及根据所述输出张量获取所述待检测图片中的行人及行人所在位置。本实施例可以有效提高小物体的预测准确性。
Description
技术领域
本发明实施例涉及计算机数据处理领域,尤其涉及一种行人检测方法、系统、计算机设备及计算机可读存储介质。
背景技术
近年来,深度卷积网络推动目标分类、目标检测、语义分割等领域取得了巨大的发展。其中,行人检测技术成为了计算机视觉领域的一个研究热点问题。行人检测技术可以应用在视频监控、智能辅助驾驶、行人分析以及智能机器人等领域,对人们的生活产生了深远的影响。
现有技术中,行人检测技术一类解决方案是基于梯度方向直方图(Histogram OfGradient,HOG)进行改进的。但是这种行人检测技术的检测精度主要依赖于研究人员的特征的设计,但设计出优秀的特征即使对于有扎实的专业基础的研究者来说也是十分困难的。另一类解决方案是,随着深度学习技术的兴起,本领域技术人员开始使用深度卷积网络自动的在数据集中提取特征,从而有效地避免了传统方法中依靠研究人员手工设计特征,有效地提高了检测的精度。
例如,基于YOLO(You Only Look Once)v1、YOLOv2等神经网络,其凭借检测速度快、准确度高等优点开始被人们重视并应用于行人检测技术中。然而,发明人目前所了解的基于上述神经网络的检测技术,对小目标的检测能力较差。
发明内容
有鉴于此,本发明实施例的目的是提供一种行人检测方法、系统、计算机设备及计算机可读存储介质,可以有效解决对小目标的检测能力较差的技术问题。
为实现上述目的,本发明实施例提供了一种基于YOLO神经网络的行人检测方法,所述方法包括:
将待检测图片输入到所述YOLO神经网络,所述YOLO神经网络包括主线和支线,所述主线包括依顺序耦合的N个卷积层,所述N个卷积层之间配置有M个池化层,第M-1个池化层耦合所述支线,所述支线包括依序耦合的第M+1个池化层和第N+1个卷积层;
通过所述YOLO神经网络对所述待检测图片执行卷积操作和池化操作,以得到2A个x*y卷积特征图,所述2A个x*y卷积特征图包括所述第N个卷积层输出A个x*y卷积特征图以及所述第N+1个卷积层输出A个x*y卷积特征图;
通过预设的B个行人预测框在所述2A个x*y卷积特征图上做卷积操作,输出x*y*B*(5+C)的输出张量,C表示C个条件类别概率;及
根据所述输出张量获取所述待检测图片中的行人及行人所在位置。
优选地,将待检测图片输入到所述YOLO神经网络的步骤之前,还包括:配置所述YOLO神经网络;其中,所述主线包括依顺序耦合的第一组卷积层、第一池化层、第二组卷积层、第二池化层、第三组卷积层、第三池化层、第四组卷积层、第四池化层、第五组卷积层、第五池化层、第六组卷积层,所述第四池化层之后还耦合有所述支线,所述支线包括依序耦合的第六池化层和第七组卷积层,其中,依序耦合的所述第五组卷积层、第五池化层和第六组卷积层与依序耦合的第六池化层和第七组卷积层之间为并联关系。
优选地,通过预设的B个行人预测框在所述2A个x*y卷积特征图上做卷积操作,输出x*y*B*(5+C)的输出张量的步骤,包括:
为所述2A个x*y卷积特征图中的每个网格配置B个行人预测框,获取每个网格上对应的B个行人预测框的(5+C)的向量;
其中,“5”表示单个网格对应其中一个行人预测框所预测得到的4个坐标值(tx、ty、tw、th)和1个置信度to,to表示该单个网格对应的行人预测框中是否存在物体,C表示该单个网格对应的行人类别的条件类别概率。
优选地,根据所述输出张量获取所述待检测图片中的行人及行人所在位置的步骤,包括:
根据每个网格上对应的B个行人预测框的(5+C)的向量,预测每个网格上对应的B个边界框的边界框数据,所述边界框数据包括每个边界框在所述待检测图片中的位置、尺寸和行人类别的行人类别概率;其中,预测每个网格上对应的B个边界框的边界框数据的计算公式包括:
bx=σ(tx)+cx
by=σ(ty)+cy
其中,σ为sigmoid函数;tx:预测边界框中心点距所在网格左边的距离;ty:预测边界框中心点距所在网格上边的距离;cx、cy为该网格距离所述待检测图片左上角的位置偏移,每个网格的长和宽为1;pw、ph为行人预测框的宽度和长度;tw:预测边界框的宽度,除以所述待检测图片宽度归一化后的值;th:预测边界框的高度,除以所述待检测图片高度归一化后的值;Pr(object):行人类别的条件类别概率;为边界框和实际边界框的交并比,所述交并比为交集部分面积和并集部分面积之比;bx、by为边界框的坐标信息;bw、bh为边界框的尺寸信息;σ(to)为边界框对行人类别的行人类别概率。
优选地,根据所述输出张量获取所述待检测图片中的行人及行人所在位置的步骤,包括:
通过每个网格对应的B个边界框对应的B个行人类别概率,得到x*y*B个边界框对应的x*y*B个行人类别概率,选择大于预设阈值的一个或多个目标边界框。
优选地:
所述第一组卷积层包括:由32个3X3卷积核组成的第一卷积层;
所述第一池化层包括:步长为2的池化层;
所述第二组卷积层包括:由64个3X3卷积核组成的第二卷积层;
所述第二池化层包括:步长为2的池化层;
所述第三组卷积层包括依序耦合的:由128个3X3卷积核组成的第三卷积层、由64个1X1卷积核组成的第四卷积层、由128个3X3卷积核组成的第五卷积层;
所述第三池化层包括:步长为2的池化层;
所述第四组卷积层包括依序耦合的:由256个3X3卷积核组成的第六卷积层、由128个1X1卷积核组成的第七卷积层、由256个3X3卷积核组成的第八卷积层;
所述第四池化层包括:步长为2的池化层;
所述第五组卷积层包括依序耦合的:由512个3X3卷积核组成的第九卷积层、由256个1X1卷积核组成的第十卷积层、由512个3X3卷积核组成的第十一卷积层、由256个1X1卷积核组成的第十二卷积层、由512个3X3卷积核组成的第十三卷积层;
所述第五池化层包括:步长为2的池化层;
所述第六组卷积层包括依序耦合的:由1024个3X3卷积核组成的第十四卷积层、由512个1X1卷积核组成的第十五卷积层、由1024个1X1卷积核组成的第十六卷积层、由512个1X1卷积核组成的第十七卷积层、由1024个3X3卷积核组成的第十八卷积层、由1024个3X3卷积核组成的第十九卷积层、由1024个3X3卷积核组成的第二十卷积层、由1024个3X3卷积核组成的第二十一卷积层、由125个1X1卷积核组成的第二十二卷积层;
所述第六池化层:步长为2的池化层;
所述第七组卷积层包括依序耦合的:由125个1X1卷积核组成的第二十三卷积层。
优选地,还包括:
通过损失函数得到所述YOLO神经网络实际输出和样本标签值之间的偏差,用于训练所述YOLO神经网络。
为实现上述目的,本发明实施例还提供了行人检测系统,包括:
输入模块,用于将待检测图片输入到所述YOLO神经网络,所述YOLO神经网络包括主线和支线,所述主线包括依顺序耦合的N个卷积层,所述N个卷积层之间配置有M个池化层,第M-1个池化层耦合所述支线,所述支线包括依序耦合的第M+1个池化层和第N+1个卷积层;
特征提取模块,用于通过所述YOLO神经网络对所述待检测图片执行卷积操作和池化操作,以得到2A个x*y卷积特征图,所述2A个x*y卷积特征图包括所述第N个卷积层输出A个x*y卷积特征图以及所述第N+1个卷积层输出A个x*y卷积特征图;
行人预测框处理模块,用于通过预设的B个行人预测框在所述2A个x*y卷积特征图上做卷积操作,输出x*y*B*(5+C)的输出张量,C表示C个条件类别概率;及
获取模块,用于根据所述输出张量获取所述待检测图片中的行人及行人所在位置。
为实现上述目的,本发明实施例还提供了一种计算机设备,所述计算机设备存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序,所述计算机程序被处理器执行时实现如上所述的行人检测方法的步骤。
为实现上述目的,本发明实施例还提供了一种计算机可读存储介质,所述计算机可读存储介质内存储有计算机程序,所述计算机程序可被至少一个处理器所执行,以使所述至少一个处理器执行如上所述的行人检测方法的步骤。
本发明实施例提供的行人检测方法、系统、计算机设备及计算机可读存储介质,通过新增一个包括第M+1个池化层和第N+1的支线,该支线在YOLO神经网络的卷积中间过程中提取待检测图片的特征,用于保留更多小目标的信息,从而有效提高了小物体的预测准确性。
附图说明
图1为本发明行人检测方法实施例一的流程图。
图2为本发明行人检测方法实施例二的流程图。
图3为本发明行人检测方法实施例二中YOLO神经网络的结构示意图。
图4为本发明行人检测系统实施例三的程序模块示意图。
图5为本发明计算机设备实施例四的硬件结构示意图。
具体实施方式
为了使本发明的目的、技术方案及优点更加清楚明白,以下结合附图及实施例,对本发明进行进一步详细说明。应当理解,此处所描述的具体实施例仅用以解释本发明,并不用于限定本发明。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
需要说明的是,在本发明中涉及“第一”、“第二”等的描述仅用于描述目的,而不能理解为指示或暗示其相对重要性或者隐含指明所指示的技术特征的数量。由此,限定有“第一”、“第二”的特征可以明示或者隐含地包括至少一个该特征。另外,各个实施例之间的技术方案可以相互结合,但是必须是以本领域普通技术人员能够实现为基础,当技术方案的结合出现相互矛盾或无法实现时应当认为这种技术方案的结合不存在,也不在本发明要求的保护范围之内。
以下实施例将以计算机设备2为执行主体进行示例性描述。
实施例一
参阅图1,示出了本发明实施例一之基于YOLO神经网络的行人检测方法的步骤流程图。可以理解,本方法实施例中的流程图不用于对执行步骤的顺序进行限定。具体如下。
步骤S100,将待检测图片输入到所述YOLO神经网络。
所述YOLO神经网络包括主线和支线,所述主线包括依顺序耦合的N个卷积层,所述N个卷积层之间配置有M个池化层,第M-1个池化层耦合所述支线,所述支线包括依序耦合的第M+1个池化层和第N+1个卷积层。
进一步的,预先配置所述YOLO神经网络:所述主线包括依顺序耦合的第一组卷积层、第一池化层、第二组卷积层、第二池化层、第三组卷积层、第三池化层、第四组卷积层、第四池化层、第五组卷积层、第五池化层、第六组卷积层,所述第四池化层之后还耦合有所述支线,所述支线包括依序耦合的第六池化层和第七组卷积层,其中,依序耦合的所述第五组卷积层、第五池化层和第六组卷积层与依序耦合的第六池化层和第七组卷积层之间为并联关系。
步骤102,通过所述YOLO神经网络对所述待检测图片执行卷积操作和池化操作,以得到2A个x*y卷积特征图。
所述2A个x*y卷积特征图包括所述第N个卷积层输出A个x*y卷积特征图以及所述第N+1个卷积层输出A个x*y卷积特征图。
步骤104,通过预设的B个行人预测框(anchor box)在所述2A个x*y卷积特征图上做卷积操作,输出x*y*B*(5+C)的输出张量,C表示C个条件类别概率。
为所述2A个x*y卷积特征图中的每个网格配置B个行人预测框,获取每个网格上对应的B个行人预测框的(5+C)的向量;
其中,“5”表示单个网格对应其中一个行人预测框所预测得到的4个坐标值(tx、ty、tw、th)和1个置信度to,to表示该单个网格对应的行人预测框中是否存在物体,C表示该单个网格对应的行人类别的条件类别概率。
步骤106,根据所述输出张量获取所述待检测图片中的行人及行人所在位置。
根据每个网格上对应的B个行人预测框的(5+C)的向量,预测每个网格上对应的B个边界框(bounding box)的边界框数据,所述边界框数据包括每个边界框在所述待检测图片中的位置、尺寸和行人类别的行人类别概率;其中,预测每个网格上对应的B个边界框的边界框数据的计算公式包括:
bx=σ(tx)+cx
by=σ(ty)+cy
其中,σ为sigmoid函数,用于将边界框的偏移值维持在(0,1)范围内,使得边界框的中心位置被约束在相应的网格内,防止偏移过多;tx:预测边界框中心点距所在网格左边的距离;ty:预测边界框中心点距所在网格上边的距离;cx、cy为该网格距离所述待检测图片左上角的位置偏移,每个网格的长和宽为1;pw、ph为行人预测框的宽度和长度;tw:预测边界框的宽度,除以所述待检测图片宽度归一化后的值;th:预测边界框的高度,除以所述待检测图片高度归一化后的值;Pr(object):行人类别的条件类别概率;bx、by为边界框的坐标信息;bw、bh为边界框的尺寸信息;为边界框和实际边界框的交并比,所述交并比为交集部分面积和并集部分面积之比;σ(to)为边界框对行人类别的行人类别概率。
IOU=交集部分面积/并集部分面积,边界框和实际边界框完全重合时IOU=1,不相交时IOU=0,即用于体现了预测的边界框与真实边界框的接近程度。
通过每个网格对应的B个边界框对应的B个行人类别概率,得到x*y*B个边界框对应的x*y*B个行人类别概率,选择大于预设阈值的一个或多个目标边界框。
在一实施例中,还包括步骤:通过损失函数得到所述YOLO神经网络实际输出和样本标签值之间的偏差,用于训练所述YOLO神经网络。
为判断网格i的第j个边界框是否负责这个物体,是则取1,否则取0。为判断有物体的中心落在网格i中,网格中包含有物体的中心,则负责预测该物体的行人类别概率。为判断网格i的第j个边界框中不存在物体。
λcoord为坐标损失权重,取5,λnoobj为置信度损失权重,取0.5。
在式(1)和式(2)为坐标预测部分,式(3)为含物体的边界框的置信度预测部分,式(4)为不含物体的边界框的置信度预测部分,式(5)为行人类别预测部分。
实施例二
参阅图2,示出了本发明实施例二之基于YOLO神经网络的行人检测方法的步骤流程图。
步骤S200,配置YOLO神经网络。
所述YOLO神经网络包括主线和支线,所述主线包括依顺序耦合的第一组卷积层、第一池化层、第二组卷积层、第二池化层、第三组卷积层、第三池化层、第四组卷积层、第四池化层、第五组卷积层、第五池化层、第六组卷积层。所述第四池化层之后还耦合所述支线,所述支线包括依序耦合的第六池化层和第七组卷积层。因此,依序耦合的所述第五组卷积层、第五池化层和第六组卷积层与依序耦合的第六池化层和第七组卷积层为并联关系。
具体的,如图3所示:
所述第一组卷积层包括:由32个3X3卷积核组成的第一卷积层;
所述第一池化层包括:步长为2的池化层;
所述第二组卷积层包括:由64个3X3卷积核组成的第二卷积层;
所述第二池化层包括:步长为2的池化层;
所述第三组卷积层包括依序耦合的:由128个3X3卷积核组成的第三卷积层、由64个1X1卷积核组成的第四卷积层、由128个3X3卷积核组成的第五卷积层;
所述第三池化层包括:步长为2的池化层;
所述第四组卷积层包括依序耦合的:由256个3X3卷积核组成的第六卷积层、由128个1X1卷积核组成的第七卷积层、由256个3X3卷积核组成的第八卷积层;
所述第四池化层包括:步长为2的池化层;
所述第五组卷积层包括依序耦合的:由512个3X3卷积核组成的第九卷积层、由256个1X1卷积核组成的第十卷积层、由512个3X3卷积核组成的第十一卷积层、由256个1X1卷积核组成的第十二卷积层、由512个3X3卷积核组成的第十三卷积层;
所述第五池化层包括:步长为2的池化层;
所述第六组卷积层包括依序耦合的:由1024个3X3卷积核组成的第十四卷积层、由512个1X1卷积核组成的第十五卷积层、由1024个1X1卷积核组成的第十六卷积层、由512个1X1卷积核组成的第十七卷积层、由1024个3X3卷积核组成的第十八卷积层、由1024个3X3卷积核组成的第十九卷积层、由1024个3X3卷积核组成的第二十卷积层、由1024个3X3卷积核组成的第二十一卷积层、由125个1X1卷积核组成的第二十二卷积层;
所述第六池化层:步长为2的池化层;
所述第七组卷积层包括依序耦合的:由125个1X1卷积核组成的第二十三卷积层。
步骤S202,将416*416像素规格的待检测图片输入到所述YOLO神经网络。
步骤204,通过所述YOLO神经网络对所述待检测图片执行卷积操作和池化操作,以得到250个13*13卷积特征图。
所述250个卷积特征图包括所述第二十二卷积层输出125个13*13卷积特征图以及所述第二十三卷积层输出125个13*13卷积特征图。
步骤206,通过5个行人预测框在所述250个13*13卷积特征图上做卷积操作,输出13*13*5*(5+C)的输出张量,C表示C个条件类别概率。
5个行人预测框为根据不同的面积比例和长宽比例的5个先验框。
每个13*13卷积特征图由13*13个网格组成。
为所述250个13*13卷积特征图中的每个网格配置5个行人预测框,获取每个网格上对应的5个行人预测框的(5+C)的向量。
步骤208,根据所述输出张量获取所述待检测图片中的行人及行人所在位置。
根据每个网格上对应的5个行人预测框的(5+C)的向量,预测每个网格上对应的5个边界框(bounding box)的边界框数据,所述边界框数据包括每个边界框在所述待检测图片中的位置、尺寸和行人类别的行人类别概率。
通过每个网格对应的5个边界框对应的5个行人类别概率,得到13*13*5个边界框对应的13*13*5个行人类别概率,选择大于预设阈值的一个或多个目标边界框。
实施例三
请继续参阅图4,示出了本发明行人检测系统实施例三的程序模块示意图。在本实施例中,行人检测系统20可以包括或被分割成一个或多个程序模块,一个或者多个程序模块被存储于存储介质中,并由一个或多个处理器所执行,以完成本发明,并可实现上述行人检测方法。本发明实施例所称的程序模块是指能够完成特定功能的一系列计算机程序指令段,比程序本身更适合于描述行人检测系统20在存储介质中的执行过程。以下描述将具体介绍本实施例各程序模块的功能:
输入模块200,用于将待检测图片输入到所述YOLO神经网络。
所述YOLO神经网络包括主线和支线,所述主线包括依顺序耦合的N个卷积层,所述N个卷积层之间配置有M个池化层,第M-1个池化层耦合所述支线,所述支线包括依序耦合的第M+1个池化层和第N+1个卷积层。
特征提取模块202,用于通过所述YOLO神经网络对所述待检测图片执行卷积操作和池化操作,以得到2A个x*y卷积特征图。
所述2A个x*y卷积特征图包括所述第N个卷积层输出A个x*y卷积特征图以及所述第N+1个卷积层输出A个x*y卷积特征图。
行人预测框处理模块204,用于通过预设的B个行人预测框在所述2A个x*y卷积特征图上做卷积操作,输出x*y*B*(5+C)的输出张量,C表示C个条件类别概率。
具体的,行人预测框处理模块204,用于:为所述2A个x*y卷积特征图中的每个网格配置B个行人预测框,获取每个网格上对应的B个行人预测框的(5+C)的向量。也即,将行人预测框在x*y卷积特征图上做滑窗操作,每个滑到一对应网格处,得到相应的向量。
“5”表示单个网格对应其中一个行人预测框所预测得到的4个坐标值(tx、ty、tw、th)和1个置信度to,to表示该单个网格对应的行人预测框中是否存在物体,C表示该单个网格对应的行人类别的条件类别概率。
获取模块206,用于根据所述输出张量获取所述待检测图片中的行人及行人所在位置。
具体的,获取模块206用于:
根据每个网格上对应的B个行人预测框的(5+C)的向量,预测每个网格上对应的B个边界框的边界框数据,所述边界框数据包括每个边界框在所述待检测图片中的位置、尺寸和行人类别的行人类别概率;其中,预测每个网格上对应的B个边界框的边界框数据的计算公式包括:
bx=σ(tx)+cx
by=σ(ty)+cy
其中,σ为sigmoid函数;tx:预测边界框中心点距所在网格左边的距离;ty:预测边界框中心点距所在网格上边的距离;cx、cy为该网格距离所述待检测图片左上角的位置偏移,每个网格的长和宽为1;pw、ph为行人预测框的宽度和长度;tw:预测边界框的宽度,除以所述待检测图片宽度归一化后的值;th:预测边界框的高度,除以所述待检测图片高度归一化后的值;Pr(object):行人类别的条件类别概率;为边界框和实际边界框的交并比,所述交并比为交集部分面积和并集部分面积之比;σ(to)为边界框对行人类别的行人类别概率。
通过每个网格对应的B个边界框对应的B个行人类别概率,得到x*y*B个边界框对应的x*y*B个行人类别概率,选择大于预设阈值的一个或多个目标边界框。
在一实施例中:
还包括配置模块208,用于配置所述YOLO神经网络;
所述YOLO神经网络包括主线和支线,所述主线包括依顺序耦合的第一组卷积层、第一池化层、第二组卷积层、第二池化层、第三组卷积层、第三池化层、第四组卷积层、第四池化层、第五组卷积层、第五池化层、第六组卷积层。所述第四池化层之后还耦合所述支线,所述支线包括依序耦合的第六池化层和第七组卷积层。因此,依序耦合的所述第五组卷积层、第五池化层和第六组卷积层与依序耦合的第六池化层和第七组卷积层为并联关系。。
进一步:
所述第一组卷积层包括:由32个3X3卷积核组成的第一卷积层;
所述第一池化层包括:步长为2的池化层;
所述第二组卷积层包括:由64个3X3卷积核组成的第二卷积层;
所述第二池化层包括:步长为2的池化层;
所述第三组卷积层包括依序耦合的:由128个3X3卷积核组成的第三卷积层、由64个1X1卷积核组成的第四卷积层、由128个3X3卷积核组成的第五卷积层;
所述第三池化层包括:步长为2的池化层;
所述第四组卷积层包括依序耦合的:由256个3X3卷积核组成的第六卷积层、由256个1X1卷积核组成的第七卷积层、由256个3X3卷积核组成的第八卷积层;
所述第四池化层包括:步长为2的池化层;
所述第五组卷积层包括依序耦合的:由512个3X3卷积核组成的第九卷积层、由128个1X1卷积核组成的第十卷积层、由512个3X3卷积核组成的第十一卷积层、由256个1X1卷积核组成的第十二卷积层、由512个3X3卷积核组成的第十三卷积层;
所述第五池化层包括:步长为2的池化层;
所述第六组卷积层包括依序耦合的:由1024个3X3卷积核组成的第十四卷积层、由512个1X1卷积核组成的第十五卷积层、由1024个1X1卷积核组成的第十六卷积层、由512个1X1卷积核组成的第十七卷积层、由1024个3X3卷积核组成的第十八卷积层、由1024个3X3卷积核组成的第十九卷积层、由1024个3X3卷积核组成的第二十卷积层、由1024个3X3卷积核组成的第二十一卷积层、由125个1X1卷积核组成的第二十二卷积层;
所述第六池化层:步长为2的池化层;
所述第七组卷积层包括依序耦合的:由125个1X1卷积核组成的第二十三卷积层。
在一实施例中:
还包括训练模块210,用于:
通过损失函数得到所述YOLO神经网络实际输出和样本标签值之间的偏差,用于训练所述YOLO神经网络。
实施例四
参阅图5,是本发明实施例四之计算机设备的硬件架构示意图。本实施例中,所述计算机设备2是一种能够按照事先设定或者存储的指令,自动进行数值计算和/或信息处理的设备。该计算机设备2可以是机架式服务器、刀片式服务器、塔式服务器或机柜式服务器(包括独立的服务器,或者多个服务器所组成的服务器集群)等。如图所示,所述计算机设备2至少包括,但不限于,可通过系统总线相互通信连接存储器21、处理器22、网络接口23、以及行人检测系统20。其中:
本实施例中,存储器21至少包括一种类型的计算机可读存储介质,所述可读存储介质包括闪存、硬盘、多媒体卡、卡型存储器(例如,SD或DX存储器等)、随机访问存储器(RAM)、静态随机访问存储器(SRAM)、只读存储器(ROM)、电可擦除可编程只读存储器(EEPROM)、可编程只读存储器(PROM)、磁性存储器、磁盘、光盘等。在一些实施例中,存储器21可以是计算机设备2的内部存储单元,例如该计算机设备2的硬盘或内存。在另一些实施例中,存储器21也可以是计算机设备2的外部存储设备,例如该计算机设备20上配备的插接式硬盘,智能存储卡(Smart Media Card,SMC),安全数字(Secure Digital,SD)卡,闪存卡(Flash Card)等。当然,存储器21还可以既包括计算机设备2的内部存储单元也包括其外部存储设备。本实施例中,存储器21通常用于存储安装于计算机设备2的操作系统和各类应用软件,例如实施例五的行人检测系统20的程序代码等。此外,存储器21还可以用于暂时地存储已经输出或者将要输出的各类数据。
处理器22在一些实施例中可以是中央处理器(Central Processing Unit,CPU)、控制器、微控制器、微处理器、或其他数据处理芯片。该处理器22通常用于控制计算机设备2的总体操作。本实施例中,处理器22用于运行存储器21中存储的程序代码或者处理数据,例如运行行人检测系统20,以实现实施例一或二的行人检测方法。
所述网络接口23可包括无线网络接口或有线网络接口,该网络接口23通常用于在所述计算机设备2与其他电子装置之间建立通信连接。例如,所述网络接口23用于通过网络将所述计算机设备2与外部终端相连,在所述计算机设备2与外部终端之间的建立数据传输通道和通信连接等。所述网络可以是企业内部网(Intranet)、互联网(Internet)、全球移动通讯系统(Global System of Mobile communication,GSM)、宽带码分多址(WidebandCode Division Multiple Access,WCDMA)、4G网络、5G网络、蓝牙(Bluetooth)、Wi-Fi等无线或有线网络。
需要指出的是,图5仅示出了具有部件20-23的计算机设备2,但是应理解的是,并不要求实施所有示出的部件,可以替代的实施更多或者更少的部件。
在本实施例中,存储于存储器21中的所述行人检测系统20还可以被分割为一个或者多个程序模块,所述一个或者多个程序模块被存储于存储器21中,并由一个或多个处理器(本实施例为处理器22)所执行,以完成本发明。
例如,图4示出了所述实现行人检测系统20实施例四的程序模块示意图,该实施例中,所述基于行人检测系统20可以被划分为输入模块200、特征提取模块202、行人预测框处理模块204、获取模块206、配置模块208和训练模块210。其中,本发明所称的程序模块是指能够完成特定功能的一系列计算机程序指令段,比程序更适合于描述所述行人检测系统20在所述计算机设备2中的执行过程。所述程序模块200-210的具体功能在实施例三中已有详细描述,在此不再赘述。
实施例五
本实施例还提供一种计算机可读存储介质,如闪存、硬盘、多媒体卡、卡型存储器(例如,SD或DX存储器等)、随机访问存储器(RAM)、静态随机访问存储器(SRAM)、只读存储器(ROM)、电可擦除可编程只读存储器(EEPROM)、可编程只读存储器(PROM)、磁性存储器、磁盘、光盘、服务器、App应用商城等等,其上存储有计算机程序,程序被处理器执行时实现相应功能。本实施例的计算机可读存储介质用于存储行人检测系统20,被处理器执行时实现实施例一或二的行人检测方法。
上述本发明实施例序号仅仅为了描述,不代表实施例的优劣。
通过以上的实施方式的描述,本领域的技术人员可以清楚地了解到上述实施例方法可借助软件加必需的通用硬件平台的方式来实现,当然也可以通过硬件,但很多情况下前者是更佳的实施方式。
以上仅为本发明的优选实施例,并非因此限制本发明的专利范围,凡是利用本发明说明书及附图内容所作的等效结构或等效流程变换,或直接或间接运用在其他相关的技术领域,均同理包括在本发明的专利保护范围内。
Claims (10)
1.一种基于YOLO神经网络的行人检测方法,其特征在于,所述方法包括:
将待检测图片输入到所述YOLO神经网络,所述YOLO神经网络包括主线和支线,所述主线包括依顺序耦合的N个卷积层,所述N个卷积层之间配置有M个池化层,第M-1个池化层还耦合所述支线,所述支线包括依序耦合的第M+1个池化层和第N+1个卷积层;
通过所述YOLO神经网络对所述待检测图片执行卷积操作和池化操作,以得到2A个x*y卷积特征图,所述2A个x*y卷积特征图包括所述第N个卷积层输出A个x*y卷积特征图以及所述第N+1个卷积层输出A个x*y卷积特征图;
通过预设的B个行人预测框在所述2A个x*y卷积特征图上做卷积操作,输出x*y*B*(5+C)的输出张量,C表示C个条件类别概率;及
根据所述输出张量获取所述待检测图片中的行人及行人所在位置。
2.根据权利要求1所述的行人检测方法,其特征在于,将待检测图片输入到所述YOLO神经网络的步骤之前,还包括:
配置所述YOLO神经网络;
其中,所述主线包括依顺序耦合的第一组卷积层、第一池化层、第二组卷积层、第二池化层、第三组卷积层、第三池化层、第四组卷积层、第四池化层、第五组卷积层、第五池化层、第六组卷积层,所述第四池化层之后还耦合有所述支线,所述支线包括依序耦合的第六池化层和第七组卷积层,其中,依序耦合的所述第五组卷积层、第五池化层和第六组卷积层与依序耦合的第六池化层和第七组卷积层之间为并联关系。
3.根据权利要求2所述的行人检测方法,其特征在于,通过预设的B个行人预测框在所述2A个x*y卷积特征图上做卷积操作,输出x*y*B*(5+C)的输出张量的步骤,包括:
为所述2A个x*y卷积特征图中的每个网格配置B个行人预测框,获取每个网格上对应的B个行人预测框的(5+C)的向量;
其中,“5”表示单个网格对应其中一个行人预测框所预测得到的4个坐标值(tx、ty、tw、th)和1个置信度to,to表示该单个网格对应的行人预测框中是否存在物体,C表示该单个网格对应的行人类别的条件类别概率。
4.根据权利要求3所述的行人检测方法,其特征在于,根据所述输出张量获取所述待检测图片中的行人及行人所在位置的步骤,包括:
根据每个网格上对应的B个行人预测框的(5+C)的向量,预测每个网格上对应的B个边界框的边界框数据,所述边界框数据包括每个边界框在所述待检测图片中的位置、尺寸和行人类别的行人类别概率;其中,预测每个网格上对应的B个边界框的边界框数据的计算公式包括:
bx=σ(tx)+cx
by=σ(ty)+cy
其中,σ为sigmoid函数;tx:预测边界框中心点距所在网格左边的距离;ty:预测边界框中心点距所在网格上边的距离;cx、cy为该网格距离所述待检测图片左上角的位置偏移,每个网格的长和宽为1;pw、ph为行人预测框的宽度和长度;tw:预测边界框的宽度,除以所述待检测图片宽度归一化后的值;th:预测边界框的高度,除以所述待检测图片高度归一化后的值;Pr(object):行人类别的条件类别概率;为边界框和实际边界框的交并比,所述交并比为交集部分面积和并集部分面积之比;bx、by为边界框的坐标信息;bw、bh为边界框的尺寸信息;σ(to)为边界框对行人类别的行人类别概率。
5.根据权利要求4所述的行人检测方法,其特征在于,根据所述输出张量获取所述待检测图片中的行人及行人所在位置的步骤,包括:
通过每个网格对应的B个边界框对应的B个行人类别概率,得到x*y*B个边界框对应的x*y*B个行人类别概率,选择大于预设阈值的一个或多个目标边界框。
6.根据权利要求2~5任意一项所述的行人检测方法,其特征在于:
所述第一组卷积层包括:由32个3X3卷积核组成的第一卷积层;
所述第一池化层包括:步长为2的池化层;
所述第二组卷积层包括:由64个3X3卷积核组成的第二卷积层;
所述第二池化层包括:步长为2的池化层;
所述第三组卷积层包括依序耦合的:由128个3X3卷积核组成的第三卷积层、由64个1X1卷积核组成的第四卷积层、由128个3X3卷积核组成的第五卷积层;
所述第三池化层包括:步长为2的池化层;
所述第四组卷积层包括依序耦合的:由256个3X3卷积核组成的第六卷积层、由128个1X1卷积核组成的第七卷积层、由256个3X3卷积核组成的第八卷积层;
所述第四池化层包括:步长为2的池化层;
所述第五组卷积层包括依序耦合的:由512个3X3卷积核组成的第九卷积层、由256个1X1卷积核组成的第十卷积层、由512个3X3卷积核组成的第十一卷积层、由256个1X1卷积核组成的第十二卷积层、由512个3X3卷积核组成的第十三卷积层;
所述第五池化层包括:步长为2的池化层;
所述第六组卷积层包括依序耦合的:由1024个3X3卷积核组成的第十四卷积层、由512个1X1卷积核组成的第十五卷积层、由1024个1X1卷积核组成的第十六卷积层、由512个1X1卷积核组成的第十七卷积层、由1024个3X3卷积核组成的第十八卷积层、由1024个3X3卷积核组成的第十九卷积层、由1024个3X3卷积核组成的第二十卷积层、由1024个3X3卷积核组成的第二十一卷积层、由125个1X1卷积核组成的第二十二卷积层;
所述第六池化层:步长为2的池化层;
所述第七组卷积层包括依序耦合的:由125个1X1卷积核组成的第二十三卷积层。
7.根据权利要求1~5任意一项所述的行人检测方法,其特征在于,还包括:
通过损失函数得到所述YOLO神经网络实际输出和样本标签值之间的偏差,用于训练所述YOLO神经网络。
8.一种行人检测系统,其特征在于,包括:
输入模块,用于将待检测图片输入到所述YOLO神经网络,所述YOLO神经网络包括主线和支线,所述主线包括依顺序耦合的N个卷积层,所述N个卷积层之间配置有M个池化层,第M-1个池化层耦合所述支线,所述支线包括依序耦合的第M+1个池化层和第N+1个卷积层;
特征提取模块,用于通过所述YOLO神经网络对所述待检测图片执行卷积操作和池化操作,以得到2A个x*y卷积特征图,所述2A个x*y卷积特征图包括所述第N个卷积层输出A个x*y卷积特征图以及所述第N+1个卷积层输出A个x*y卷积特征图;
行人预测框处理模块,用于通过预设的B个行人预测框在所述2A个x*y卷积特征图上做卷积操作,输出x*y*B*(5+C)的输出张量,C表示C个条件类别概率;及
获取模块,用于根据所述输出张量获取所述待检测图片中的行人及行人所在位置。
9.一种计算机设备,所述计算机设备存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至7中任一项所述的行人检测方法的步骤。
10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质内存储有计算机程序,所述计算机程序可被至少一个处理器所执行,以使所述至少一个处理器执行如权利要求1至7中任一项所述的行人检测方法的步骤。
Priority Applications (2)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201910031726.4A CN109902556A (zh) | 2019-01-14 | 2019-01-14 | 行人检测方法、系统、计算机设备及计算机可存储介质 |
PCT/CN2019/118471 WO2020147410A1 (zh) | 2019-01-14 | 2019-11-14 | 行人检测方法、系统、计算机设备及计算机可存储介质 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201910031726.4A CN109902556A (zh) | 2019-01-14 | 2019-01-14 | 行人检测方法、系统、计算机设备及计算机可存储介质 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN109902556A true CN109902556A (zh) | 2019-06-18 |
Family
ID=66943667
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN201910031726.4A Pending CN109902556A (zh) | 2019-01-14 | 2019-01-14 | 行人检测方法、系统、计算机设备及计算机可存储介质 |
Country Status (2)
Country | Link |
---|---|
CN (1) | CN109902556A (zh) |
WO (1) | WO2020147410A1 (zh) |
Cited By (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111179247A (zh) * | 2019-12-27 | 2020-05-19 | 上海商汤智能科技有限公司 | 三维目标检测方法及其模型的训练方法及相关装置、设备 |
CN111209822A (zh) * | 2019-12-30 | 2020-05-29 | 南京华图信息技术有限公司 | 一种热红外图像的人脸检测方法 |
WO2020147410A1 (zh) * | 2019-01-14 | 2020-07-23 | 平安科技(深圳)有限公司 | 行人检测方法、系统、计算机设备及计算机可存储介质 |
Families Citing this family (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN112308045B (zh) * | 2020-11-30 | 2023-11-24 | 深圳集智数字科技有限公司 | 一种密集人群的检测方法、装置及电子设备 |
CN112700472A (zh) * | 2020-12-30 | 2021-04-23 | 深圳云天励飞技术股份有限公司 | 目标跟踪方法及相关设备 |
CN114118124B (zh) * | 2021-09-29 | 2023-09-12 | 北京百度网讯科技有限公司 | 图像检测方法和装置 |
CN115147642A (zh) * | 2022-06-02 | 2022-10-04 | 盛视科技股份有限公司 | 基于视觉的渣土车检测方法、装置、计算机及存储介质 |
Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107862287A (zh) * | 2017-11-08 | 2018-03-30 | 吉林大学 | 一种前方小区域物体识别及车辆预警方法 |
CN108509859A (zh) * | 2018-03-09 | 2018-09-07 | 南京邮电大学 | 一种基于深度神经网络的无重叠区域行人跟踪方法 |
CN108805070A (zh) * | 2018-06-05 | 2018-11-13 | 合肥湛达智能科技有限公司 | 一种基于嵌入式终端的深度学习行人检测方法 |
Family Cites Families (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN108537117B (zh) * | 2018-03-06 | 2022-03-11 | 哈尔滨思派科技有限公司 | 一种基于深度学习的乘客检测方法和系统 |
CN108875600A (zh) * | 2018-05-31 | 2018-11-23 | 银江股份有限公司 | 一种基于yolo的车辆信息检测和跟踪方法、装置及计算机存储介质 |
CN108921875B (zh) * | 2018-07-09 | 2021-08-17 | 哈尔滨工业大学(深圳) | 一种基于航拍数据的实时车流检测与追踪方法 |
CN109902556A (zh) * | 2019-01-14 | 2019-06-18 | 平安科技(深圳)有限公司 | 行人检测方法、系统、计算机设备及计算机可存储介质 |
-
2019
- 2019-01-14 CN CN201910031726.4A patent/CN109902556A/zh active Pending
- 2019-11-14 WO PCT/CN2019/118471 patent/WO2020147410A1/zh active Application Filing
Patent Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107862287A (zh) * | 2017-11-08 | 2018-03-30 | 吉林大学 | 一种前方小区域物体识别及车辆预警方法 |
CN108509859A (zh) * | 2018-03-09 | 2018-09-07 | 南京邮电大学 | 一种基于深度神经网络的无重叠区域行人跟踪方法 |
CN108805070A (zh) * | 2018-06-05 | 2018-11-13 | 合肥湛达智能科技有限公司 | 一种基于嵌入式终端的深度学习行人检测方法 |
Non-Patent Citations (1)
Title |
---|
殷帅 等: "基于YOLO网络的数据采集与标注", 仪表技术, no. 12, pages 22 - 25 * |
Cited By (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
WO2020147410A1 (zh) * | 2019-01-14 | 2020-07-23 | 平安科技(深圳)有限公司 | 行人检测方法、系统、计算机设备及计算机可存储介质 |
CN111179247A (zh) * | 2019-12-27 | 2020-05-19 | 上海商汤智能科技有限公司 | 三维目标检测方法及其模型的训练方法及相关装置、设备 |
CN111209822A (zh) * | 2019-12-30 | 2020-05-29 | 南京华图信息技术有限公司 | 一种热红外图像的人脸检测方法 |
Also Published As
Publication number | Publication date |
---|---|
WO2020147410A1 (zh) | 2020-07-23 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN109902556A (zh) | 行人检测方法、系统、计算机设备及计算机可存储介质 | |
CN112685565B (zh) | 基于多模态信息融合的文本分类方法、及其相关设备 | |
Wu et al. | Object detection based on RGC mask R‐CNN | |
US11861919B2 (en) | Text recognition method and device, and electronic device | |
WO2022105125A1 (zh) | 图像分割方法、装置、计算机设备及存储介质 | |
CN113379718A (zh) | 一种目标检测方法、装置、电子设备以及可读存储介质 | |
CN109522963A (zh) | 一种单机运行的特征建筑目标检测方法和系统 | |
CN112016510B (zh) | 基于深度学习的信号灯识别方法和装置、设备及存储介质 | |
CN110349138B (zh) | 基于实例分割框架的目标物体的检测方法及装置 | |
CN110457677A (zh) | 实体关系识别方法及装置、存储介质、计算机设备 | |
CN108229658A (zh) | 基于有限样本的物体检测器的实现方法及装置 | |
CN111444807A (zh) | 目标检测方法、装置、电子设备和计算机可读介质 | |
CN113378712A (zh) | 物体检测模型的训练方法、图像检测方法及其装置 | |
KR102138747B1 (ko) | 가로 글자와 세로 글자를 동시에 처리하기 위한 방법 및 시스템 | |
CN110019952B (zh) | 视频描述方法、系统及装置 | |
CN114565803A (zh) | 用于提取难样本的方法、装置及机械设备 | |
CN115331048B (zh) | 图像分类方法、装置、设备以及存储介质 | |
CN116343169A (zh) | 路径规划方法、目标对象运动控制方法、装置及电子设备 | |
CN109190467A (zh) | 一种基于关键点回归的多物体检测方法、系统、终端和存储介质 | |
CN112241736A (zh) | 一种文本检测的方法及装置 | |
CN112016503B (zh) | 人行道检测方法、装置、计算机设备及存储介质 | |
CN115116080A (zh) | 表格解析方法、装置、电子设备和存储介质 | |
CN113989618A (zh) | 可回收物品分类识别方法 | |
CN114529719A (zh) | 一种地面地图要素语义分割方法、系统、介质及设备 | |
CN111814865A (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 |