[发明专利]一种交通违法行为检测方法、装置及计算机可读存储介质

说明书

本发明公开了一种交通违法行为检测方法、装置及计算机可读存储介质，将道路环境图像输入至二维图像目标检测模型，输出道路环境图像中目标的二维位置信息以及类别信息；基于二维位置信息获取包含目标的截锥体点云；将截锥体点云输入至三维图像目标检测模型，输出目标的三维位置信息；参考三维位置信息对目标进行违法行为检测，并将违法行为检测结果上报至后台服务器。通过本发明的实施，采用深度学习的方式，结合两阶段的检测网络实现三维复杂场景中的目标检测，并且结合点云数据可以实现极端光照环境下的目标检测，有效保证了违法行为检测的精度和效率。

技术领域

本发明涉及图像识别领域，尤其涉及一种交通违法行为检测方法、装置及计算机可读存储介质。

背景技术

目前，交通违法行为严重影响了我国交通安全和道路畅通，如何对交通违法行为进行有效检测对于交通管理有着重要意义。

相关技术中在进行交通违法行为检测时，通常所采用的方式是由交通管理人员对所采集的视频图像进行人工识别，一方面，人工识别较为费时费力，另一方面，在复杂交通环境下人工识别的准确率较为有限。

发明内容

本发明实施例的主要目的在于提供一种交通违法行为检测方法、装置及计算机可读存储介质，至少能够解决相关技术中进行交通违法行为检测的效率较低以及准确率较为有限的问题。

为实现上述目的，本发明实施例第一方面提供了一种交通违法行为检测方法，该方法包括：

将道路环境图像输入至二维图像目标检测模型，输出所述道路环境图像中目标的二维位置信息以及类别信息；其中，所述目标的类别信息包括：车辆、行人、交通指示灯、交通标志以及道路指示线；

基于所述二维位置信息获取包含所述目标的截锥体点云；

将所述截锥体点云输入至三维图像目标检测模型，输出所述目标的三维位置信息；

参考所述三维位置信息对所述目标进行违法行为检测；

将违法行为检测结果上报至后台服务器。

为实现上述目的，本发明实施例第二方面提供了一种交通违法行为检测装置，该装置包括：

第一检测模块，用于将道路环境图像输入至二维图像目标检测模型，输出所述道路环境图像中目标的二维位置信息以及类别信息；其中，所述目标的类别信息包括：车辆、行人、交通指示灯、交通标志以及道路指示线；

获取模块，用于基于所述二维位置信息获取包含所述目标的截锥体点云；

第二检测模块，用于将所述截锥体点云输入至三维图像目标检测模型，输出所述目标的三维位置信息；

第三检测模块，用于参考所述三维位置信息对所述目标进行违法行为检测；

上报模块，用于将违法行为检测结果上报至后台服务器。

为实现上述目的，本发明实施例第三方面提供了一种电子装置，该电子装置包括：处理器、存储器和通信总线；

所述通信总线用于实现所述处理器和存储器之间的连接通信；

所述处理器用于执行所述存储器中存储的一个或者多个程序，以实现上述任意一种交通违法行为检测方法的步骤。

为实现上述目的，本发明实施例第四方面提供了一种计算机可读存储介质，该计算机可读存储介质存储有一个或者多个程序，所述一个或者多个程序可被一个或者多个处理器执行，以实现上述任意一种交通违法行为检测方法的步骤。