【技术实现步骤摘要】
目标检测方法、装置、存储介质及终端
[0001]本专利技术涉及信息
,尤其涉及一种目标检测方法、装置、存储介质及终端。
技术介绍
[0002]随着人工智能技术的快速发展,涌现一大批基于深度学习的目标检测算法,并被广泛应用于辅助驾驶、视频监控、机器人视觉、工业检测等领域的目标检测任务中。视觉感知是辅助驾驶中道路环境感知的重要组成部分,可自动对摄像机所拍摄图像进行分析,主动预测车辆周围存在的潜在危险状况,如行人是否不按交通规则横穿马路、前方车辆是否突然刹车等。
[0003]现有技术在针对驾驶场景下摄像机所拍摄图像进行目标检测时,以YOLOv3算法为基础框架,通过嵌入SENet结构以增强特征映射图的感受野,使网络学习到的特征信息更全面。然而此方法存在以下缺点:(1)SENet只是在通道维度上对特征进行了筛选加权,无法较好获取的位置关系信息,检测精度欠佳。
[0004](2)YOLOv3算法存在召回率不足,定位不够准确的缺点。与YOLOv1、YOLOv2等之前的版本相比,YOLOv3的精度有所提升,但是检测速度 ...
【技术保护点】
【技术特征摘要】
1.一种目标检测方法,其特征在于,包括:获取驾驶场景下摄像机所拍摄的图像;将所述图像输入至已训练好的目标检测网络,通过所述目标检测网络对所述图像进行判断和预测,得到目标分类和位置信息;其中,所述目标检测网络采用轻量化的YOLOv5s网络结构作为基础框架,在YOLOv5s主干网络的跨阶段局部网络中嵌入瓶颈注意力机制模块,以及在YOLOv5s主干网络的指定卷积层中采用深度可分离卷积运算。2.如权利要求1所述的目标检测方法,其特征在于,所述目标检测网络在YOLOv5s主干网络的BottleneckCSP1_x层之中嵌入瓶颈注意力机制模块,得到基于瓶颈注意力机制模块的跨阶段局部网络。3.如权利要求2所述的目标检测方法,其特征在于,所述目标检测网络将YOLOv5s主干网络中的指定CBH模块替换为MBH模块,所述CBH模块由卷积运算、归一化处理以及激活函数组成,所述MBH模块由基于深度可分离卷积运算的倒置残差模块、归一化处理以及激活函数组成。4.如权利要求3所述的目标检测方法,其特征在于,所述基于深度可分离卷积运算的倒置残差模块包括第一单点卷积层、深度卷积层、第二单点卷积层和融合层,其中所述第一单点卷积层和深度卷积层采用BatchNorm操作和非线性ReLU6激活函数,所述第二单点卷积层采用BatchNorm操作而不采用非线性ReLU6激活函数;所述第一单点卷积层用于将第一低维度特征表示扩展到第一高维度特征表示,所述深度卷积层用于基于深度可分离卷积运算对所述第一高维度特征表示进行特征提取,得到第二高维度特征表示;所述第二单点卷积层用于将第二高维度特征表示进行压缩,得到第二低维度特征表示;所述...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。