摘要
本发明公开了一种基于目标检测的道路逆行与交通事故事件检测方法,包括以下步骤:S1、获取原始数据;S2、在原始数据中获取样本,标注样本中车辆在帧图片中的位置及车辆类别;S3、经过数据处理得到训练集和验证集;S4、对原始YOLOv5的数据增强方法、激活函数进行改进,得到YOLOv5-better模型;S5、将训练集和验证集输入到YOLOv5-better模型,经过训练得到改进模型的权重文件;S6、将得到的权重文件输入到YOLOv5-better模型,进行测试集测试得到车辆信息,并输入到deepsort,得到车辆的编号id和类别;S7、将每个id在视频帧中对应的位置信息输入到逻辑判断算法,判断是否逆行或者发生事故。本发明采用上述基于目标检测的道路逆行与交通事故事件检测方法,能够适用智能视频分析工作。
- 单位