【技术实现步骤摘要】
一种基于显著性先验的交通注视目标检测方法
[0001]本专利技术涉及计算机视觉
,尤其涉及一种基于显著性先验的交通注视目标检测方法。
技术介绍
[0002]随着汽车电子化、智能化的发展,智能驾驶已经成为未来汽车发展的主流趋势之一。视觉显著性和目标检测作为智能驾驶系统的重要组成部分,是安全驾驶的重要保障。
[0003]当前研究人员通常把交通场景中的视觉显著性和目标检测当作两个独立的任务。即视觉显著性能预测驾驶员关注的区域或目标,但没有提供目标类别;而目标检测尽可能检测场景中所有目标,但无法预测场景中哪些物体是吸引驾驶员关注的,这些吸引驾驶员关注的物体是和当前驾驶任务以及驾驶安全密切相关的。因此目前亟需一种将两者之间建立联系的方法从而提高检测的准确度和检测性能。
技术实现思路
[0004]针对现有技术中所存在的不足,本专利技术提供了一种基于显著性先验的交通注视目标检测方法,能够模拟驾驶员的选择性注意,结合视觉显著性和目标检测,聚焦驾驶员注视区域内的注视目标检测,提高模型检测准确度,为智能驾驶或高级辅 ...
【技术保护点】
【技术特征摘要】
1.一种基于显著性先验的交通注视目标检测方法,其特征在于:包括以下步骤:S1,基于驾驶员的眼动,建立注视目标检测数据集,所述注视目标检测数据集包括用于在交通场景中吸引驾驶员注意的显著性信息,以及吸引驾驶员注意的显著区域内的物体类别;S2,基于YOLOv5
‑
L模型和轻量的注意力模块CBAM构建一个强基线检测模块;S3,共享S2中建立的强基线检测模块的backbone和top
‑
down路径,建立显著区域引导模块用于预测交通场景中吸引驾驶员注意的显著区域,所述显著区域用于引导强基线检测模块关注驾驶员注意区域内的注视目标;S4,通过模拟交通场景中驾驶员的选择性注意,提出显著性引导策略;S5,基于S2中建立的强基线检测模块、S3中构建的显著区域引导模块以及S4中提出的显著性引导策略,建立注视目标检测模型;S6,将训练集的图片缩放到给定的输入分辨率后输入S5中构建的注视目标检测模型,并对注视目标检测模型进行训练,从而提高注视目标检测模型的检测性能。2.如权利要求1所述的一种基于显著性先验的交通注视目标检测方法,其特征在于:在S1中,基于驾驶员的眼动,建立注视目标检测数据集时,包括以下步骤:步骤a,利用行车记录仪采集不同天气和不同驾驶场景下车辆行驶状态中的交通视频;步骤b,让驾驶员在模拟驾驶任务时观看采集的交通视频,同时利用眼动仪记录驾驶员的眼动信息;步骤c,将记录到的所有驾驶员的注视点映射到交通视频上;步骤d,对注视目标进行标注,最终得到驾驶员注视目标检测数据集;步骤e,基于已标注的注视目标检测训练集生成显著区域掩膜。3.如权利要求2所述的一种基于显著性先验的交通注视目标检测方法,其特征在于:在步骤d中,对注视目标进行标注时,从交通视频的每五帧中提取一帧,用来标注注视目标。4.如权利要求3所述的一种基于显著性先验的交通注视目标检测方法,其特征在于:在S2中,将一个由卷积操作,convolution(k=3,s=1),批规范化处理和SILU激活构建的基本模块添加到backbone的第一层,并将轻量的注意力模块CBAM添加到第一次和第二次下采样层中,从而建立强基线检测模块以检测注视目标的具体位置和类别;利用损失函数对强基线检测模块进行优...
【专利技术属性】
技术研发人员:石一,颜红梅,
申请(专利权)人:电子科技大学长三角研究院湖州,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。