一种基于YOLOv5的城市街道场景目标检测方法

    公开(公告)号:CN116385860A

    公开(公告)日:2023-07-04

    申请号:CN202310384232.0

    申请日:2023-04-11

    Inventor: 张元 陈斌 杨浩

    Abstract: 本发明涉及目标检测技术领域,公开了一种基于YOLOv5的城市街道场景目标检测方法,包括以下步骤:步骤1:构建基于改进YOLO‑v5的目标检测模型;所述目标检测模型包括主干特征提取网络、特征融合模块和输出模块;且主干提取网络中的下采样模块由可变形卷积单元和Focus单元组成;特征融合模块的下采样模块由预设步距的卷积单元和Focus单元组成;步骤2:训练目标检测模型;步骤3:由训练后的目标检测模型进行场景目标检测;其中,由主干特征提取网络提取待检测场景图像的特征;由特征融合模块输出不同尺度的特征图;并由输出模块基于特征图输出目标检测特征图。本发明能够在达到较高的检测精度的同时,保证达到较高的检测效率。

Patent Agency Ranking