交通信号灯识别方法及装置与流程

文档序号:22388836发布日期:2020-09-29 17:52阅读:324来源:国知局
交通信号灯识别方法及装置与流程

本发明涉及电子技术领域,其中,本发明尤其涉及一种交通信号灯识别方法及装置。



背景技术:

随着电子技术和汽车技术的发展,汽车的自动驾驶逐渐成为电子技术的应用领域中主要的研究方向。在汽车自动驾驶时,需要实时获取前方道路驾驶信息,尤其是汽车前方交通信号灯的显示信息的识别,以使汽车根据路况信息调整汽车的自动驾驶策略,使汽车根据在满足路况以及交通规则的情况下实现启动、停止、加速或减速。

现有技术中,汽车自动驾驶时用于识别交通信号灯的电子设备获取拍摄设备采集的汽车前方道路的图像后,通过深度学习模型,以机器学习的方式提取图像的特征并对图像中所包括的交通信号灯进行识别。其中,深度学习模型通过若干包括交通信号灯的图像集合进行训练,深度学习模型提取并存储图像集合中交通信号灯的特征集合,以在识别时将待处理的图像中交通信号灯的特征与特征集合进行比对从而实现交通信号灯的显示信息的识别。

但是采用现有技术,深度学习模型在训练过程中具体包括图像的交通信号灯检测和交通信号灯识别的两部分内容,因此在深度学习模型计算损失率(loss)时,最终计算的是检测和识别两部分的loss的和,而深度学习神经网络在反向传播计算进行优化时,其交通信号灯检测的参数会影响信号灯识别的参数loss的计算,进而会降低深度学习模型识别交通信号灯的准确率。因此,如何提高对交通信号灯的识别准确率,是本领域亟待解决的技术问题。



技术实现要素:

本发明提供一种交通信号灯识别方法及装置,使用不同的第一深度学习模型和第二深度学习模型分别进行图像的交通信号灯检测和交通信号灯识别,从而能够避免两个深度学习模型的互相干扰,提高深度学习模型的检测效率和识别效率,进而提高对交通信号灯的识别准确率。

本发明第一方面提供一种交通信号灯识别方法,包括:

确定待处理图像、第一深度学习模型和第二深度学习模型;

根据所述第一深度学习模型确定所述待处理图像中包含交通信号灯的目标区域;

根据所述第二深度学习模型识别所述目标区域中所述交通信号灯的显示信息。

本发明第二方面提供一种交通信号灯识别装置,包括:

确定模块,用于确定待处理图像、第一深度学习模型和第二深度学习模型;

检测模块,用于根据第一深度学习模型确定待处理图像中包含交通信号灯的目标区域;

识别模块,用于根据第二深度学习模型识别所述目标区域中所述交通信号灯的显示信息。

综上,本发明提供一种交通信号灯识别方法及装置,其中方法包括:确定待处理图像、第一深度学习模型和第二深度学习模型;根据第一深度学习模型确定待处理图像中包含交通信号灯的目标区域;根据第二深度学习模型识别目标区域中交通信号灯的显示信息。本发明提供的交通信号灯识别方法及装置,通过使用不同的第一深度学习模型和第二深度学习模型分别进行图像的交通信号灯检测和交通信号灯识别,从而能够避免两个深度学习模型的互相干扰,提高深度学习模型的检测效率和识别效率,进而提高对交通信号灯的识别准确率。

附图说明

为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动性的前提下,还可以根据这些附图获得其他的附图。

图1为本发明提供的交通信号灯识别方法实施例一的流程示意图;

图2为本发明提供的确定待识别图像中目标区域的应用示意图;

图3为本发明提供的第一深度学习模型一实施例的结构示意图;

图4为本发明提供的第二深度学习模型一实施例的结构示意图;

图5为本发明提供的交通信号灯识别方法实施例二的流程示意图;

图6为本发明提供的跟踪目标区域的应用示意图;

图7为本发明提供的交通信号灯识别装置实施例一的结构示意图;

图8为本发明提供的交通信号灯识别装置实施例二的结构示意图。

具体实施方式

下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。

本发明的说明书和权利要求书及上述附图中的术语“第一”、“第二”、“第三”、“第四”等(如果存在)是用于区别类似的对象,而不必用于描述特定的顺序或先后次序。应该理解这样使用的数据在适当情况下可以互换,以便这里描述的本发明的实施例例如能够以除了在这里图示或描述的那些以外的顺序实施。此外,术语“包括”和“具有”以及他们的任何变形,意图在于覆盖不排他的包含,例如,包含了一系列步骤或单元的过程、方法、系统、产品或设备不必限于清楚地列出的那些步骤或单元,而是可包括没有清楚地列出的或对于这些过程、方法、产品或设备固有的其它步骤或单元。

下面以具体地实施例对本发明的技术方案进行详细说明。下面这几个具体的实施例可以相互结合,对于相同或相似的概念或过程可能在某些实施例不再赘述。

图1为本发明提供的交通信号灯识别方法实施例一的流程示意图。如图1所示,本实施例的执行主体可以是任何具备相关数据处理功能的电子设备,例如:手机、平板电脑、笔记本电脑、台式电脑和服务器等。或者,本实施例的执行主体还可以是电子设备的芯片,例如:cpu和gpu等。本发明后续各实施例中以电子设备为执行主体为示例进行说明,并且在本实施例一种可能的应用方式中,电子设备可以设置在自动驾驶车辆上,通过对设置在车辆上的拍摄设备得到的车辆前方的图像进行交通信号灯的识别,以使自动驾驶车辆根据交通信号灯的识别结果确定自动驾驶策略。

则在s101中,电子设备首先确定待处理图像、第一深度学习模型和第二深度学习模型。其中,电子设备通过第一深度学习模型和第二深度学习模型对待处理图像中的交通信号灯进行识别,第一深度学习模型与第二深度学习模型独立存在。在本实施例中,第一深度学习模型用于检测并提取待处理图像中包含交通信号灯的目标区域,第二深度学习模型用于识别目标区域中交通信号灯的显示信息。

可选地,第一深度学习模型和第二深度学习模型可以是电子设备根据包括交通信号灯的多个训练图像通过机器学习算法提取特征后,得到并存储在存储设备内的包含交通信号灯图像特征的深度学习模型。或者,第一深度学习模型和第二深度学习模型还可以是电子设备连接服务器后,通过有线通信或者无线通信方式从服务器所获取的,本实施例对此不做限定。

可选地,若电子设备应用在自动驾驶场景下,则s101中所获取的待处理图像可以是自动驾驶车辆的拍摄设备所拍摄的车辆前方的实时图像,该图像可能包括待识别的交通信号灯,电子设备需要对该实时图像中的交通信号灯进行识别。或者,在其他可能的应用场景下,s101中的电子设备可以通过获取的存储设备中存储的图像作为待处理图像,s101中的电子设备还可以根据电子设备的用户的指示的图像作为待处理图像,并对待处理图像中的交通信号灯进行识别。需要说明的是,本实施例对于具体得到待识别图像、第一深度学习模型和第二深度学习模型的方式与主体均不做具体限定,s101中强调用于识别交通信号灯的电子设备对待识别图像以及两个深度学习模型的获取和确定。

随后在s102中,电子设备对s101中所获取的待处理图像进行识别,根据s101中所确定的第一深度学习模型确定待处理图像中包含交通信号灯的目标区域。

具体地,在s102中电子设备通过第一深度学习模型对待处理图像中包含的交通信号灯的目标区域进行检测,得到待处理图像中目标区域的坐标。下面结合图2对本发明所述的目标区域进行进一步说明,其中,图2为本发明提供的确定待识别图像中目标区域的应用示意图,如图2所示,待识别图像10内包括交通信号灯101。当电子设备获取待识别图像10后,通过第一深度学习模型提取待识别图像10中包括交通信号灯101的目标区域20。并且该矩形的目标区域可以通过矩形的两个对角顶点a和b在待识别图像10的坐标来表示。例如一种可能的坐标可以使用待识别图像10左下角的像素点作为原点(0,0)建立坐标系,并根据a点和b点在待识别图像10与原点的相对像素点位置得到a点的坐标例如(80,50)和b点的坐标例如(100,40)等。可选地,在如图2所示的示例中,仅以目标区域的形状为矩形作为示例性的说明,目标区域还可以是圆形或者不规则形状等形状,在此不做限定。

进一步地,s102一种可能的实现方式中,根据第一深度学习模型通过多个卷积特征尺度对待处理图像进行处理后,得到待处理图像的多个特征;随后根据多个不同卷积特征尺度得到的特征,确定待处理图像中包含交通信号灯的目标区域。即,本实施例通过所获取的第一深度学习模型对待处理图像使用多个不同卷积特征尺度进行特征提取操作,并根据多个不同卷积特征尺度得到的特征共同确定待处理图像中包含交通信号灯的目标区域。

具体地,下面结合附图3对本实施例中第一深度学习模型确定待处理图像中目标区域的流程进行说,其中,图3为本发明提供的第一深度学习模型一实施例的结构示意图。在图3中以第一深度学习模型为深度卷积神经网络中的深度残差网络(deepresidualnetwork,简称:resnet)为例进行说明。

由于在现有技术中的resnet50内只存在一个特定大小尺度的卷积特征,通过该resnet50内仅有的一个卷积特征尺度能够识别出待处理图像中大部分目标区域。但是,针对在交通信号灯检测过程中,若待处理图像中包括的交通信号灯距离较远而在待处理图像中显示占比较小,如果还以整张图片为主体并使用resnet50内特定卷积特征尺度进行检测时,可能无法检测到图像中较小的交通信号灯。因此,本实施例提供的卷积神经网络resnet50中,通过不同大小的卷积特征尺度对待处理图像进行处理,以满足对待处理图像中显示占比较小的交通信号灯的检测,在图3所示的示例中,卷积神经网络resnet50包括三个不同大小的卷积特征尺度,记为第一卷积特征尺度、第二卷积特征尺度和第三卷积特征尺度。

例如在图3中,若待处理图像的尺度为512*512,则将待处理图像送入卷积神经网络resnet50后,resnet50使用首先最大尺度例如128*128的第一卷积特征尺度对待处理图像中的交通信号灯位置进行检测。随后,resnet50再依次使用第二卷积特征尺度和第三卷积特征尺度再次对待处理图像中的交通信号灯位置进行检测,其中,不同的卷积特征尺度的按照由从大到小的顺序前后排列,每个较小卷积特征尺度对之前的较大卷积特征尺度所输出的特征进行处理。第二卷积特征尺度可以是例如64*64,第三卷积特征尺度可以是例如32*32。最后,resnet50可以根据非极大值抑制(nonmaximumsuppression,简称:nms)算法根据三个卷积特征尺度的输出结果确定最终的目标区域。其中,由于根据最小尺度的第三卷积特征尺度可能无法检测到待处理图像中占比较小的交通信号灯,而尺度较大的第一卷积特征尺度能够对占比较小的交通信号灯进行更加精细地识别。因此对于包含较小交通信号灯的待处理图像,第一卷积特征尺度能够检测到待处理图像中的交通信号灯而第二卷积特征尺度和第三卷积特征尺度可能无输出结果,则可以直接将第一卷积特征尺度的输出结果作为目标区域。而若对于包含较大交通信号灯的待处理图像,resnet50中的三个卷积特征尺度都可能会检测出包含交通信号灯的目标区域,则nms算法会遍历三个卷积特征尺度的输出结果,选取具有最高评分结果的输出结果作为目标区域。其中,所述评分为每个卷积特征尺度在检测待处理图像中的交通信号灯时,根据机器学习评价标准对于最有可能存在交通信号灯的区域的评分,每个卷积特征尺度都会将该卷积特征尺度下评分最高的区域进行输出。最终resnet50结合三个具有不同卷积特征尺度的检测结果确定待处理图像中包括交通信号灯的目标区域,输出的bbox信息中可以包括目标区域的矩形两个对角顶点a和b在待识别图像10的坐标(x,y)和(w,h)。

可选地,在本发明如图3所示实施例中,基于resnet50使用batchnorm层进行训练可以加速训练收敛并对卷积层的数据进行归一化的处理。因此,在本实施例中,将batchnorm层和scale层合并融合至卷积层中,使得卷积层在不影响准确度情况下加速预测计算时间,能够尽快输出检测结果。即,在如图3所示的卷积神经网络resnet50中,每个卷积特征尺度的计算时,都会经过batchnorm层和scale层的计算,从而提高卷积神经网络resnet50的计算速度并减少对于电子设备内存和显存空间的占用,加速卷积神经网络resnet50网络传播的速度。

可选地,由于在卷积神经网络的应用中,使用更小的卷积核是在保证当前网络精度的情况下,减少神经网络计算参数的趋势之一。则在本发明如图3所示的实施例中,可以使用2个1*1的卷积核代替现有的卷积神经网络resnet50中3*3的卷积核,从而在保证具有相同感知视野的条件下,在一定程度上提升网络的深度,还能够提升卷积神经网络的计算效率和性能。

进一步地,在本实施例中,为了提高卷积神经网络resnet50的性能,在如图3所示的实施例中,卷积神经网络resnet50将神经网络中采用32bit浮点数表示的模型参数采用int8量化压缩后进行计算,从而加快卷积神经网络resnet50的计算速度,提升卷积神经网络的计算效率和性能。

进一步地,基于在自动驾驶应用场景下,电子设备所采集的实时路况图像最为待处理图像时,交通信号灯多位于待处理图像的上部分,例如在图2所示的示例中,待处理图像10中的交通信号灯101位于整个待处理图像上半部分。则在本实施例s102中,第一深度学习模型在对待处理图像进行检测时,从所述待处理图像的特定区域确定包含交通信号灯的目标区域。其中,特定区域可以是最可能包括交通信号灯的区域,例如待处理图像的上部分的区域。例如:第一深度学习模型可以通过预设比例提取部分待处理图像进行检测计算,预设比例可以设置为60%,则第一深度学习模型在s102中根据待处理图像进行裁剪后保留的上60%的部分图像进行目标区域的检测,以提高第一深度学习模型的检测计算效率。

可以理解的是,若s102中第一深度学习模型在待处理图像中未检测出包括交通信号灯的目标区域,则不会继续通过后续对交通信号灯具体显示信息进行识别。

随后在s103中,电子设备根据s102中所确定的包括交通信号灯的目标区域,根据s101中所获取的第二深度学习模型,识别目标区域中交通信号灯的显示信息。其中,显示信息包括但不限于:交通信号灯的颜色。交通信号灯的颜色可以分为4类:红、黄、绿和交通信号灯关闭(off)状态的颜色。

图4为本发明提供的第二深度学习模型一实施例的结构示意图。如图4示出了一种第二深度学习模型为深度卷积神经网络为例,其中,该深度卷积神经网络包括:3个卷积层(convolutions)、2个池化(max-pooling)层、2个全连接层、1个空间金字塔池化(spatialpyramidpooling,简称:spp)层和多个logistic分类器。

如图4所示,深度卷积神经网络将输入的待处理图像的目标区域依次经过3个卷积层和2个池化层的处理后,将卷积特征图送入spp层。其中,深度卷积神经网络通过卷积层和池化层对待处理图像进行的提取卷积特征图的处理可参照现有技术,不再赘述。可选地,如图4所示的深度卷积神经网络获取到输入的待处理图像后,可以首先根据s102中第一深度学习模型所确定的目标区域的坐标在待处理图像中提取目标区域。

而由于s102中,第一深度学习模型使用不同大小的卷积特征尺度对待处理图像中包含交通信号灯的目标区域进行检测,而不同卷积特征尺度得到的目标区域的尺度不固定;又因为深度卷积神经网络在训练过程中,需要使用统一尺度的图像进行信号的属性的分类学习。因此,为了统一第一深度学习模型输出的目标区域的尺度以及深度卷积神经网络进行目标区域中信号灯识别的尺度,本实施例在现有的深度卷积神经网络基础上,加入了spp层,用于将第一深度学习模型输出的目标区域调整为目标尺度后,再通过深度卷积神经网络进行识别。其中,所述的目标尺度是深度卷积神经网络训练过程中使用的包含交通信号灯的训练图像的尺度。例如,深度卷积神经网络在训练过程中,使用目标尺度为128*128的包含交通信号灯的图像进行特征的识别与模型的训练。而在待处理图像的识别过程中,若第一深度学习模型所提取的待处理图像中目标区域的尺度为64*64,此时深度卷积神经网络需要通过spp层对目标区域进行尺度变换,将目标区域的尺度由64*64变换为目标尺度128*128后,再通过全连接层的处理并送入logistic分类器进行交通信号灯显示信息的识别,从而能够统一深度卷积神经网络中的分类器所处理的图像的尺度,允许第一深度学习模型在检测过程中对待处理图像进行处理后得到的图像的尺度与深度卷积神经网络训练时尺度不同,进而提高分类器的检测准确程度。

如图4所示的实施例中,针对现有技术中深度卷积神经网络使用的softmax分类器由于其单输出的特性而只能对目标区域中出现的得分最高的交通信号灯进行识别,而交通信号灯在道路上设置较为复杂,在自动驾驶的场景下,拍摄设备所采集的待处理图像中可能不只包括一个交通信号灯,而是包括多个交通信号灯,并且同一个目标区域内还可能会包括距离较近的两个交通信号灯,softmax分类器无法对目标区域中出现的多个交通信号灯进行识别。则本实施例图4提供的s103一种可能的实现方式中,电子设备根据第二深度学习模型识别s102中所获取的目标区域中至少一个交通信号灯的显示信息。

具体地,所述的深度卷积神经网络中包括逻辑(logistic)分类器,利用logistic分类器的多输出特性,对同一目标区域内至少一个交通信号灯进行识别,所述至少一个交通信号灯包括:一个或者多个交通信号灯,所述多个指两个或两个以上。可选地,本实施例中深度卷积神经网络包括的logistic分类器可以是多个。其中,s102中若第一深度学习模型检测出目标区域中包括至少一个交通信号灯,则在所确定的目标区域中进行标记。s103中深度卷积神经网络根据标记的个数,通过多个logistic分类器识别出与标记个数相同的交通信号灯的显示信息。例如:若深度卷积神经网络所获取的目标区域中标记有2个交通信号灯,则在如图4所示的logistic分类步骤中,由多个不同的logistic分类器同时对经过前述处理后的卷积图数据进行识别,并取多个不同的logistic分类器的输出中得分最高的2个交通信号灯作为2个交通信号灯的显示信息的输出。

进一步地,在本实施例中,为了提高深度卷积神经网络的性能,在如图4所示的实施例中,深度卷积神经网络将神经网络中采用32bit浮点数表示的模型参数采用int8量化压缩后进行计算,从而加快深度卷积神经网络的计算速度,提升卷积神经网络的计算效率和性能。

可选地,在本发明各实施例的基础上,本发明还提供一种训练得到第一深度学习模型和第二深度学习模型的方式。其中,使用第一训练图像集合训练得到第一深度学习模型,第一训练图像集合包括多个包含交通信号灯的图像;使用第二训练图像集合训练得到第一深度学习模型,第二训练图像集合包括多个包含交通信号灯的目标区域。即,本发明将用于检测的第一深度学习模型和用于识别的第二深度学习模型分开进行深度学习以得到不同的深度学习模型。

而现有的交通信号灯识别方法中,深度学习模型中检测和识别部分是一并处理,这样在求深度学习模型的损失率(loss)时,最终计算的是检测和识别两部分的loss的和。其中,loss为神经网络在图像的检测与识别过程中用于指示网络输出结果与实际图像之差的参数,神经网络可以根据loss的回馈机制对神经网络内的参数进行调整。例如:若神经网络在图像检测时所检测的包含交通信号灯的目标区域顶点的坐标为a而实际坐标为b,则此时loss=b-a,而神经网络在图像识别时识别出交通信号灯的属性为c而实际属性为d,此时loss=d-c,因此对于该神经网络对于图像既进行检测又进行识别,该神经网络的loss=(b-a)+(d-c)。由于神经网络的训练最终目标是通过loss来衡量,loss越小时说明神经网络能够输出的内容和实际的内容更加接近。明显地,在现有技术中神经网络的loss是检测和识别两部分之和,对于该神经网络最终需要输出的交通信号灯识别结果,由于识别结果在训练时loss有检测结果的loss的加入,检测的loss影响识别的loss,导致神经网络的loss较大而不能进一步降低,而loss越大说明神经网络能够输出的内容与实际的内容越不相似,从而会由于检测部分的引入会降低神经网络对交通信号灯的识别准确率。因此,本发明提供的将用于检测的第一深度学习模型和用于识别的第二深度学习模型分开进行深度学习,能够避免两个深度学习模型的互相干扰,使得专门用于识别的第二深度学习模型在训练时无检测部分loss的干扰,能够获得尽可能小的loss,进而提高训练得到的深度学习模型的检测效率和识别效率。

综上,本实施例提供的交通信号灯识别方法中,通过使用不同的第一深度学习模型和第二深度学习模型分别进行图像的交通信号灯检测和交通信号灯识别,从而能够避免两个深度学习模型的互相干扰,提高深度学习模型的检测效率和识别效率,进而提高对交通信号灯的识别准确率。

图5为本发明提供的交通信号灯识别方法实施例二的流程示意图。如图5所示的实施例二在如图1所示基础上,还包括:s104:根据目标区域对所述交通信号灯在待处理图像中的位置信息进行跟踪。

其中,在s104一种可能的实现方式中,电子设备具体根据medianflow跟踪算法来实现多张连续的图像中特定交通信号灯位置信息的跟踪。例如,图6为本发明提供的跟踪目标区域的应用示意图,在如图6所示的示例中,在自动驾驶的应用场景下,电子设备对拍摄设备在时间轴t上所采集的连续n帧图像进行检测时,由于连续采集的图像之间交通信号灯可能位置变化较小。因此在第一帧图像①作为待处理图像进行如前述实施例中s102中第一深度学习模型检测得到交通信号灯的位置信息后,通过medianflow跟踪算法建立该交通信号灯的位置信息与交通信号灯的对应关系,并对下一帧图像②中交通信号灯的位置信息进行预测,从而在电子设备通过s102检测包括交通信号灯的目标区域时,第一深度学习模型能够以图像①中交通信号灯的位置信息为参考,对图像②中包括交通信号灯的目标区域进行检测,从而实现对连续n张图像中交通信号灯的位置信息进行跟踪,进而实现对交通信号灯的识别效率。

图7为本发明提供的交通信号灯识别装置实施例一的结构示意图。如图7所示,本实施例提供的交通信号灯识别装置包括:确定模块701,检测模块702和识别模块703。其中,确定模块701用于确定待处理图像、第一深度学习模型和第二深度学习模型;检测模块702用于根据第一深度学习模型确定待处理图像中包含交通信号灯的目标区域;识别模块703用于根据第二深度学习模型识别所述目标区域中所述交通信号灯的显示信息。

本实施例提供的交通信号灯识别装置可用于执行如图1所示的交通信号灯识别方法,其实现方式与原理相同,不再赘述。

可选地,在上述实施例中,检测模块702具体用于,根据第一深度学习模型通过多个卷积特征尺度对待处理图像处理,得到待处理图像的多个特征;根据第一深度学习模型根据多个特征确定待处理图像中包含交通信号灯的目标区域。

可选地,在上述实施例中,检测模块702具体用于,根据第一深度学习模型依次使用多个卷积特征尺度对待处理图像进行处理,使得多个卷积特征尺度中的每个卷积特征尺度输出一个待处理图像的特征;其中,多个卷积特征尺度按照从大到小的顺序前后排列,每个卷积特征尺度对之前的卷积特征尺度所输出的特征进行处理得到卷积特征尺度的特征。

可选地,在上述实施例中,检测模块702具体用于,根据第一深度学习模型通过非极大值抑制nms算法,从多个特征中确定待处理图像中包含交通信号灯的目标区域。

可选地,在上述实施例中,第一深度学习模型为卷积神经网络resnet50;resnet50中的每个卷积特征尺度均包括batchnorm层和scale层;其中,batchnorm层和scale层用于提高卷积特征尺度的计算速度。

可选地,在上述实施例中,识别模块703具体用于,根据第二深度学习模型识别目标区域中至少一个交通信号灯的显示信息。

可选地,在上述实施例中,识别模块703具体用于,根据第二深度学习模型提取目标区域的卷积特征;根据第二深度学习模型通过空间金字塔池化spp层调整卷积特征的尺寸;根据第二深度学习模型通过多个logistic分类器对调整尺寸后的卷积特征进行分类识别,确定目标区域中至少两个交通信号灯的显示信息。

可选地,在上述各实施例中,第一深度学习模型使用int8量化压缩;和/或,第二深度模型使用int8量化压缩。

可选地,在上述实施例中,检测模块702具体用于,根据所述第一深度学习模型在所述待处理图像的特定区域确定包含交通信号灯的目标区域。

图8为本发明提供的交通信号灯识别装置实施例二的结构示意图。如图8所示的交通信号灯识别装置在如图7所示实施例一的基础上,还包括:跟踪模块801,用于根据所述目标区域对所述交通信号灯在所述待处理图像中的位置信息进行跟踪。

本实施例提供的交通信号灯识别装置可用于执行如图5所示的交通信号灯识别方法,其实现方式与原理相同,不再赘述。

本发明还提供一种电子设备,包括:处理器,存储器以及计算机程序;其中,所述计算机程序被存储在所述存储器中,并且被配置为由所述处理器执行,所述计算机程序包括用于执行如前述实施例中任一项所述的交通信号灯识别方法的指令。

本发明还提供一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,所述计算机程序使得服务器执行如前述实施例中任一项所述的交通信号灯识别方法。

本领域普通技术人员可以理解:实现上述各方法实施例的全部或部分步骤可以通过程序指令相关的硬件来完成。前述的程序可以存储于一计算机可读取存储介质中。该程序在执行时,执行包括上述各方法实施例的步骤;而前述的存储介质包括:rom、ram、磁碟或者光盘等各种可以存储程序代码的介质。

最后应说明的是:以上各实施例仅用以说明本发明的技术方案,而非对其限制;尽管参照前述各实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分或者全部技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本发明各实施例技术方案的范围。

当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1