-
公开(公告)号:CN118097494A
公开(公告)日:2024-05-28
申请号:CN202410098471.4
申请日:2024-01-24
Applicant: 东南大学
IPC: G06V20/40 , G06V20/54 , G06V10/82 , G06N3/0464 , G06N3/08
Abstract: 本发明公开了一种基于改进YOLOv5与DeepSORT的交通事件检测方法、装置及存储介质,所述方法包括:获取实际道路交通路况视频数据,并制作数据集;基于引入CAFM模块的改进YOLOv5模型对训练集和验证集进行目标检测模型训练;将得到的权重文件输入到改进的YOLOv5模型中,对测试集进行检测,提取车辆信息;将提取的车辆信息输入到嵌入了车辆类别融合模块的改进DeepSORT算法中,进行车辆跟踪;并引入增益值进行车辆逆行和违停判断。该方法能优化小尺寸目标检测效果,提升车辆分类和交通事件检测的准确率。