智能交通系统

2023-03-11 15:01| 来源: 网络整理| 查看: 265

目录前言一、环境配置二、车辆检测、跟踪、计数算法及代码解读1、主函数各参数含义2、算法实现3、效果展示三、车辆测速算法及代码解读1、算法流程2、核心代码3、效果展示四、车辆碰撞检测算法及代码解读1、算法流程2、核心代码3、效果展示五、违规进入专用车道检测算法及代码解读1、算法流程2、核心代码3、效果展示六、总结及源码获取1、总结2、项目资源获取

前言

1、本项目通过yolov5-5.0和deepsort实现了一个多功能智能交通监控系统，可为一些同学的课设、大作业等提供参考。分别实现了不同车辆的跟踪，统计不同车型“上行”和“下行”的数量，实时检测车辆速度，检测两车是否发生碰撞或者距离过近时进行碰撞预警，检测车辆是否违规进入专用车道（例如非公交车进入公交车专用车道，非法占用高速应急车道等行为）。最终检测效果如下，红色框内区域即为自定义的公交车专用车道。 2、可训练自己的数据集，可以换成yolov5各种版本的权重。在这里插入图片描述

一、环境配置 pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html pip install -r requirements.txt 二、车辆检测、跟踪、计数算法及代码解读 1、主函数各参数含义

如下代码所示，可根据自己需求更改。使用yolov5s.pt、yolov5m.pt、yolov5l.pt、yolov5x.pt预训练权重均可，也可以使用自己训练好的权重，本项目中调用的是训练好的可以检测car、truck、bus三个类别的权重。

if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument('--weights', type=str, default='best.pt', help='model.pt path') # 加载的v5权重，可以是公开数据集的预训练权重，也可以是自己数据集训练的权重 parser.add_argument('--source', type=str, default='IMG_6894.mp4', help='source') # 待检测的视频路径 parser.add_argument('--output', type=str, default='test_out', help='output folder') # output folder parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)') parser.add_argument('--conf-thres', type=float, default=0.6, help='object confidence threshold') parser.add_argument('--iou-thres', type=float, default=0.5, help='IOU threshold for NMS') parser.add_argument('--fourcc', type=str, default='mp4v', help='output video codec (verify ffmpeg support)') parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu') parser.add_argument('--view-img', action='store_true', help='display results') parser.add_argument('--save-txt', action='store_true', help='save results to *.txt') parser.add_argument('--classes', nargs='+', type=int, default=[0, 1, 2], help='filter by class') # car、truck、bus parser.add_argument('--agnostic-nms', action='store_true', help='class-agnostic NMS') parser.add_argument('--augment', action='store_true', help='augmented inference') parser.add_argument("--config_deepsort", type=str, default="deep_sort/configs/deep_sort.yaml") 2、算法实现

使用yolov5和deepsort分别实现车辆的目标检测、跟踪、计数，算法流程如下，计数原理是将整个视频图像区域划分为四个象限，把第一象限和第三象限的车辆分别认为是下行和上行的车辆。计数具体代码实现时使用了一系列if-eif条件语句，对目标框的中心坐标以及四个象限边界的位置关系进行判断。在这里插入图片描述

3、效果展示

以视频中红色横线为基准，统计car、bus、truck三种车型“上行”和“下行”的数量和总共的数量，并显示在图像左上角，同时将车型、车辆ID打印在车辆检测框的正上方。可以拓展到行人跟踪等多个领域，只要将训练好的所需类别的yolov5权重替换掉，代码里面做一些修改就行了。在这里插入图片描述

三、车辆测速算法及代码解读 1、算法流程

首先使用提前设定好的车辆真实宽度和检测出来的车辆像素宽度求出真实距离和像素距离的比值，再使用每辆车的前后两帧框的中心坐标计算出两帧之间移动的像素距离。利用这个比值和像素距离做映射，就可以求出两帧之间车辆移动的真实距离。然后距离除以两帧之间的时间，就是速度了。本测速算法中将车辆真实移动距离与像素移动距离看成是线性关系，仅在监控相机轴线与车辆移动方向垂直时才能成立，并且检测出来的车辆框在空间上会产生一定形变，使得真实距离和像素距离的映射关系不准确。有兴趣的同学可以在代码中加入透视变换，将图像变成类似于遥感数据的俯瞰图，实现测速后再将图像变换为原始图像视角，就能实现比较准确的车辆测速了。

2、核心代码

我的项目将测速代码封装到了Estimated_speed()函数里面，有详细注释，调用即可。

def Estimated_speed(locations, fps, width): present_IDs = [] prev_IDs = [] work_IDs = [] work_IDs_index = [] work_IDs_prev_index = [] work_locations = [] # 当前帧数据：中心点x坐标、中心点y坐标、目标序号、车辆类别、车辆像素宽度 work_prev_locations = [] # 上一帧数据，数据格式相同 speed = [] for i in range(len(locations[1])): present_IDs.append(locations[1][i][2]) # 获得当前帧中跟踪到车辆的ID for i in range(len(locations[0])): prev_IDs.append(locations[0][i][2]) # 获得前一帧中跟踪到车辆的ID for m, n in enumerate(present_IDs): if n in prev_IDs: # 进行筛选，找到在两帧图像中均被检测到的有效车辆ID，存入work_IDs中 work_IDs.append(n) work_IDs_index.append(m) for x in work_IDs_index: # 将当前帧有效检测车辆的信息存入work_locations中 work_locations.append(locations[1][x]) for y, z in enumerate(prev_IDs): if z in work_IDs: # 将前一帧有效检测车辆的ID索引存入work_IDs_prev_index中 work_IDs_prev_index.append(y) for x in work_IDs_prev_index: # 将前一帧有效检测车辆的信息存入work_prev_locations中 work_prev_locations.append(locations[0][x]) for i in range(len(work_IDs)): speed.append( math.sqrt((work_locations[i][0] - work_prev_locations[i][0]) ** 2 + # 计算有效检测车辆的速度，采用线性的从像素距离到真实空间距离的映射 (work_locations[i][1] - work_prev_locations[i][1]) ** 2) * # 当视频拍摄视角并不垂直于车辆移动轨迹时，测算出来的速度将比实际速度低 width[work_locations[i][3]] / (work_locations[i][4]) * fps / 5 * 3.6 * 2) for i in range(len(speed)): speed[i] = [round(speed[i], 1), work_locations[i][2]] # 将保留一位小数的单位为km/h的车辆速度及其ID存入speed二维列表中 return speed 3、效果展示

将计算出来的速度实时显示在车辆正上方。可以针对不同的道路环境设定一个速度阈值，例如高速上通常限速为120km/h，将每一帧中各车辆的速度和ID存到一个列表里面，读取列表，和设定的速度阈值进行比较。若超速，则发出警告。需要注意的是，每判断完一帧后，要将这个列表进行清空防止帧之间的数据混淆，同时防止列表溢出（虽然不大可能）。在这里插入图片描述

四、车辆碰撞检测算法及代码解读 1、算法流程

首先将yolov5检测出来的车辆框通过两层for循环嵌套读取出来，将任意两个车辆框进行两两匹配，再通过一系列的if和elif语句判断两框位置关系，若两框相交区域的宽度或高度超过一定像素t，或存在包含与被包含关系，则判定为碰撞，将其车辆ID存入列表中。t值可以按照实际需求自己调整，t值越大，碰撞越不容易检出。当t为一个较小的值时，可当做碰撞预警功能来使用。

2、核心代码

本项目将这个功能封装到了find_accidents()函数里面，直接调用即可。核心判断语句代码如下：

# 通过A、B框位置关系判断其是否发生碰撞，若相交、存在包含关系，则判定为碰撞，将其索引加入crash_index列表中。 if (B_xmin

【本文地址】

智能交通系统

智能交通系统

今日新闻

推荐新闻