本发明涉及图像处理,尤其涉及一种安全带识别方法、装置、设备及介质。
背景技术:
1、随着我国科技的迅速发展,车辆的数量越来越多,车辆的使用日益频繁,通过驾驶人员系安全带可保障驾驶人员安全。
2、目前,通过图像处理来监察驾驶人员是否佩戴安全带,进行驾驶安全的保障。
3、但是,驾驶室情况复杂,简单的图像处理导致图像处理过程中识别安全带准确率低。
技术实现思路
1、本发明提供了一种安全带识别方法、装置、设备及介质,以提高安全带识别的准确性。
2、第一方面,本发明实施例提供了一种安全带识别方法,该方法包括:
3、获取识别图像,识别图像包括主副驾驶用户的身体部位图像;
4、将识别图像输入到多任务模型中;
5、通过多任务模型中至少一个卷积层对识别图像进行处理,得到目标特征图;
6、对目标特征图进行多个任务类型的分类处理,得到各任务类型对应的识别结果,任务类型包括:安全带识别任务类型、反光衣识别任务类型和遮挡识别任务类型;
7、根据各任务类型对应的识别结果,检测安全带的佩戴识别结果。
8、第二方面,本发明实施例提供了一种安全带识别装置,装置包括:
9、图像获取模块,用于获取识别图像,识别图像包括主副驾驶用户的身体部位图像;
10、图像输入模块,用于将识别图像输入到多任务模型中;
11、特征图获取模块,用于通过多任务模型中至少一个卷积层对识别图像进行处理,得到目标特征图;
12、任务识别模块,用于对目标特征图进行多个任务类型的分类处理,得到各任务类型对应的识别结果,任务类型包括:安全带识别任务类型、反光衣识别任务类型和遮挡识别任务类型;
13、结果获取模块,用于根据各任务类型对应的识别结果,检测安全带的佩戴识别结果。
14、第三方面,本发明实施例提供了一种安全带识别设备,安全带识别设备包括:
15、至少一个处理器;以及
16、与至少一个处理器通信连接的存储器;其中,
17、存储器存储有可被至少一个处理器执行的计算机程序,计算机程序被至少一个处理器执行,以使至少一个处理器能够执行本发明任一实施例的安全带识别方法。
18、第四方面,本发明实施例提供了一种计算机可读存储介质,计算机可读存储介质存储有计算机指令,计算机指令用于使处理器执行时实现本发明任一实施例的安全带识别方法。
19、本发明实施例的技术方案,通过将识别图像输入到多任务模型中至少一个卷积层对识别图像进行处理,得到目标特征图;对目标特征图进行多个任务类型的分类处理,通过安全带识别任务类型、反光衣识别任务类型和遮挡识别任务类型处理,得到各任务类型对应的识别结果,检测安全带的佩戴识别结果,通过卷积层的处理,获得具有关键特征的目标特征图,针对该特征进行不同任务类型的分类处理,通过多维数据监测安全带的佩戴结果,提升了安全带识别的准确性。
20、应当理解,本部分所描述的内容并非旨在标识本发明的实施例的关键或重要特征,也不用于限制本发明的范围。本发明的其它特征将通过以下的说明书而变得容易理解。
1.一种安全带识别方法,其特征在于,所述方法包括:
2.根据权利要求1所述的方法,其特征在于,所述获取识别图像,包括:
3.根据权利要求1所述的方法,其特征在于,通过所述多任务模型中至少一个卷积层对所述识别图像进行处理,得到目标特征图,包括:
4.根据权利要求1所述的方法,其特征在于,所述根据各所述任务类型对应的识别结果,检测安全带的佩戴识别结果,包括:
5.根据权利要求4所述的方法,其特征在于,所述根据未系安全带时长、反光衣识别任务类型的识别结果和遮挡识别任务类型的识别结果,检测安全带的佩戴识别结果,包括:
6.根据权利要求4所述的方法,其特征在于,所述根据反光衣识别任务类型的识别结果和遮挡识别任务类型的识别结果,检测安全带的佩戴识别结果,包括:
7.根据权利要求2所述的方法,其特征在于,所述目标检测模型以第一帧率在芯片中运行,所述多任务模型以第二帧率在所述芯片中运行。
8.一种安全带识别装置,其特征在于,所述装置包括:
9.一种安全带识别设备,其特征在于,所述安全带识别设备包括:
10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质存储有计算机指令,所述计算机指令用于使处理器执行时实现权利要求1-7任一项所述的安全带识别方法。
