前言

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预训练权重均可，也可以使用自己训练好的权重，本项目中调用的是yolov5s.pt。

if __name__ == '__main__':parser = argparse.ArgumentParser()parser.add_argument('--weights', nargs='+', type=str, default='yolov5s.pt', help='model.pt path(s)')parser.add_argument('--source', type=str, default='data/videos/test.mp4', help='source')  #  file/folder, 0 for webcamparser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')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',default=True)parser.add_argument('--save-txt', action='store_true', help='save results to *.txt')parser.add_argument('--save-conf', action='store_true', help='save confidences in --save-txt labels')parser.add_argument('--nosave', action='store_true', help='do not save images/videos')    # store_true为保存视频或者图片，路径为runs/detectparser.add_argument('--classes', nargs='+', type=int, help='filter by class: --class 0, or --class 0 2 3')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('--update', action='store_true', help='update all models')parser.add_argument('--project', default='runs/detect', help='save results to project/name')  # 结果视频的保存路径parser.add_argument('--name', default='exp', help='save results to project/name')parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')parser.add_argument("--config_deepsort", type=str, default="deep_sort/configs/deep_sort.yaml")

2、算法实现

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

3、核心代码

我的项目将测速代码封装到了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])  # 获得当前帧中跟踪到车辆的IDfor i in range(len(locations[0])):prev_IDs.append(locations[0][i][2])  # 获得前一帧中跟踪到车辆的IDfor 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

另外，我的项目中将每辆车的中心坐标轨迹和车速分别写入了根目录下的track.txt和speed.txt，实现了每辆车的速度和轨迹信息记录。

# 将每帧检测出来的目标中心坐标和车辆ID写入txt中,实现轨迹跟踪
if len(location) != 0:with open('track.txt', 'a+') as track_record:track_record.write('frame:%s\n' % str(frame_idx))for j in range(len(location)):track_record.write('id:%s,x:%s,y:%s\n' % (str(location[j][2]), str(location[j][0]), str(location[j][1])))print('done!')
locations.append(location)
print(len(locations))
# 每五帧写入一次测速的数据，进行测速
if len(locations) == 5:if len(locations[0]) and len(locations[-1]) != 0:locations = [locations[0], locations[-1]]speed = Estimated_speed(locations, fps, width)with open('speed.txt', 'a+') as speed_record:for sp in speed:speed_record.write('id:%s %skm/h\n' % (str(sp[1]), str(sp[0])))  # 将每辆车的速度写入项目根目录下的speed.txt中locations = []

4、效果展示

如图所示，每个目标车辆测出来的速度和行驶轨迹的中心坐标分别存储在两个txt里面，id值用于区分不同的车辆，frame值代表视频的第几帧，x、y分别表示横纵坐标值。

二、跟车距离测量算法及代码解读

1、主函数各参数含义

foc = 500.0        # 镜头焦距,单位为cm
real_hight_bicycle = 26.04      # 自行车高度，注意单位是英寸
real_hight_car = 59.08      # 汽车高度
real_hight_motorcycle = 47.24      # 摩托车高度
real_hight_bus = 125.98      # 公交车高度
real_hight_truck = 137.79   # 卡车高度# 自定义函数，单目测距
def detect_distance_car(h):dis_inch = (real_hight_car * foc) / (h - 2)dis_cm = dis_inch * 2.54dis_cm = int(dis_cm)dis_m = dis_cm/100return dis_m

2、算法实现

车辆距离计算公式：D = (F*W)/P，其中D是目标到摄像机的距离(即车辆距离）, Ｆ是相机焦距, W是目标的宽度或者高度, P是指目标在图像中所占据的x方向像素的宽或者y方向像素的高（由YOLOv5的目标检测结果可获取）。首先需要设置好镜头焦距，这个参数可以通过在网上查询拍摄设备的参数获取，我这里用的测试视频使用行车记录仪拍摄，焦距为500cm，然后分别设置好自行车、汽车、摩托车、公交车和卡车的实际高度（单位为英寸），利用该公式就能计算出前车距离。本质上就是通过车辆现实尺寸和像素尺寸实现了一个距离映射。

3、效果展示

如图所示，1.6km/h代表这辆车相对拍摄设备行驶的相对速度，car代表目标类别为汽车，0.83为目标的置信度，2.42m为测得的跟车距离。

三、前车碰撞预警（追尾预警）算法及代码解读

1、算法实现

首先通过detect.py函数里的time_person变量设置人脑反应后的刹车时间，单位为s，即人开始反应后踩下刹车到车辆刹停的时间，这个时间与车辆本身的速度有关，后续可通过车机系统接口读取该速度，实现更好的预警效果。这里我们的预设值为3s。

    time_person = 3   # 设置人脑反应后的刹车时间，单位为s，即从人反应后踩下刹车到车辆刹停的时间，这个时间与车辆本身的速度有关，后续可通过车机系统接口读取该速度，实现更好的预警效果

再调用plot_one_box()函数，将前述变量 time_person、所测得的车辆目标速度、类别名称等值传入。

plot_one_box(xyxy, im0, speed, outputs, time_person, label=label, color=[0, 0, 255], line_thickness=3, name=names[int(cls)])  # 调用函数进行不同类别的测距，并绘制目标框

plot_one_box()函数在plots.py中的定义如下，首先根据不同的标签名称调用不同的函数计算跟车距离，再利用测出来的速度和距离计算时间t，与预先设定的人脑反应后的刹车时间time_person在draw_speed()函数中进行比较，并返回一个标记值flag。若时间t小于time_person的1/2，则判定为高风险，并将车辆目标绘制为红色框进行预警；若时间t介于time_person和time_person的1/2之间，则判定为低风险，并将车辆目标绘制为黄色框进行预警；若时间t大于time_person，则并将车辆目标绘制为绿色框，判定为无风险。

def plot_one_box(x, img, speed, outputs, time_person, color=None, label=None, line_thickness=3, name=None):# Plots one bounding box on image imgtl = line_thickness or round(0.002 * (img.shape[0] + img.shape[1]) / 2) + 1  # line/font thicknesscolor = color or [random.randint(0, 255) for _ in range(3)]c1, c2 = (int(x[0]), int(x[1])), (int(x[2]), int(x[3]))# w = int(x[2]) - int(x[0])  # 框的宽h = int(x[3]) - int(x[1])    # 框的高dis_m = 1.00if name == 'bicycle':    # 根据标签名称调用不同函数计算距离dis_m = detect_distance_bicycle(h)elif name == 'car':dis_m = detect_distance_car(h)elif name == 'motorcycle':dis_m = detect_distance_motorcycle(h)elif name == 'bus':dis_m = detect_distance_bus(h)elif name == 'truck':dis_m = detect_distance_truck(h)label += f'  {dis_m}m'    # 在标签后追加距离# 利用测出来的速度和距离计算时间，与预先设定的人脑反应后的刹车时间进行比较，flag=''if len(outputs) > 0:bbox_xyxy = outputs[:, :4]identities = outputs[:, -2]img, flag = draw_speed(img, speed, bbox_xyxy, identities, time_person, dis_m)if flag == "High risk":   # 根据判定的不同风险等级，绘制不同颜色的目标框，起到预警的作用cv2.rectangle(img, c1, c2, [0, 0, 255], thickness=tl, lineType=cv2.LINE_AA)if label:tf = max(tl - 1, 1)  # font thicknesst_size = cv2.getTextSize(label, 0, fontScale=tl / 3, thickness=tf)[0]c2 = c1[0] + t_size[0], c1[1] - t_size[1] - 3cv2.rectangle(img, c1, c2, [0, 0, 255], -1, cv2.LINE_AA)  # filledcv2.putText(img, label, (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)elif flag == "Low risk":cv2.rectangle(img, c1, c2, [0, 215, 255], thickness=tl, lineType=cv2.LINE_AA)if label:tf = max(tl - 1, 1)  # font thicknesst_size = cv2.getTextSize(label, 0, fontScale=tl / 3, thickness=tf)[0]c2 = c1[0] + t_size[0], c1[1] - t_size[1] - 3cv2.rectangle(img, c1, c2, [0, 215, 255], -1, cv2.LINE_AA)  # filledcv2.putText(img, label, (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)else:cv2.rectangle(img, c1, c2, [48, 128, 20], thickness=tl, lineType=cv2.LINE_AA)if label:tf = max(tl - 1, 1)  # font thicknesst_size = cv2.getTextSize(label, 0, fontScale=tl / 3, thickness=tf)[0]c2 = c1[0] + t_size[0], c1[1] - t_size[1] - 3cv2.rectangle(img, c1, c2, [48, 128, 20], -1, cv2.LINE_AA)  # filledcv2.putText(img, label, (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)

2、效果展示

如图所示，正前方有四辆车由于跟车距离过近和相对速度过快，触发了系统的预警功能，目标框分别显示为红色和黄色，起到对驾驶员或自动驾驶系统进行提醒的作用。还有目标由于距离过远，对车辆的行车安全不构成威胁，所以显示为绿色框。

四、总结及源码获取

1、总结

本项目基于深度目标检测和跟踪技术，结合了一些图像逻辑后处理算法，实现了车辆检测、跟踪、测速、车间距离的测量和前车碰撞预警的功能，检测准确率较高，算法实时性较好，对于自动驾驶车辆的交通安全和环境感知具有一定参考意义和实用价值。

2、项目资源获取

本项目效果展示视频：
https://www.bilibili.com/video/BV14d4y177vE/?spm_id_from=333.999.0.0&vd_source=8c532ded7c7c9041f04e35940d11fdae
资源获取：

获取整套代码、测试视频、训练好的权重和说明文档(有偿)
上交硕士，技术够硬，也可以指导深度学习毕设、大作业等。
--------------  3582584734  ->qq  ------------

yolo-车辆测距+前车碰撞预警(追尾预警)+车辆检测识别+车辆跟踪测速(原创算法-毕业设计)相关推荐

1. 前车碰撞预警——FCW,基于深度学习和单目摄像头测距的前车碰撞预警源码

   前车碰撞预警--FCW,基于深度学习和单目摄像头测距的前车碰撞预警源码. 单目测距,多目标跟踪. 车辆检测,智能adas,FCW,价格只包括源码及设计文档讲解. 我使用的版本说明: gpu版本: an ...

2. 基于深度学习和单目摄像头测距的前车碰撞预警源码

   前车碰撞预警--FCW,基于深度学习和单目摄像头测距的前车碰撞预警源码 单目测距,多目标跟踪. 车辆检测,智能adas,FCW,编号只包括源码及设计文档讲解. 我使用的版本说明: gpu版本: ano ...

3. 车距检测预警HMW与前车碰撞预警FCW的区别

   转自:http://www.maxieye.net/cn/new/new-85-444.html 车距检测预警 Headway Monitoring Warning;前车碰撞预警Forward Col ...

4. FCW Forward Collision Warning 前车碰撞预警

   当车速大于30km/h时,FCW功能处于激活状态. 最早可以在预计发生追尾碰撞前2.7秒时,车辆图标变红闪烁,并发出紧促的滴滴声音告警. 车距监控与告警HMW 当车速大于30km/h,且车距<= ...

5. 车辆跟驰模型matlab代码实现_MATLAB——考虑驾驶员特性及前车速度的快速路模型...

   重发一下之前误删的一篇~ 目前大多数元胞自动机模型并没有考虑前车速度,大多数同向行驶的模型中车辆都是处在一个完全跟车的状态,无论前车是加速还是减速,后车驾驶者都只是根据自己的车速判断是减速跟驰还是变换 ...

6. matlab车头时距,教你如何判断车头与前车的距离？so easy！

   新手,你懂的.无论是跟车,还是停车,都离前边的车或者标线还有好大一段距离.像这种和前车或者障碍物有多少距离,很多新手都拿捏不准,特别是开不熟悉的车,更是如此 .跟车凭"感觉",感觉 ...

7. 毫米波雷达、ADAS中的应用以及毫米波雷达的检测、测距、测速和角度测量

   毫米波雷达的检测.测距.测速和角度测量 毫米波(Millimeter-Wave,缩写:MMW),是指长度在1~10mm的电磁波,对应的频率范围为30~300GHz．根据波的传播理论,频率越高,波长越短 ...

8. 全球及中国生物识别车辆访问技术行业十四五应用现状及未来展望规划报告2022-2027年

   全球及中国生物识别车辆访问技术行业十四五应用现状及未来展望规划报告2022-2027年 [搜索鸿晟信合查看官网更多内容!]  生物识别车辆访问技术位于起步阶段,指纹识别.虹膜扫描.面部识别等技术可以取 ...

9. 智能驾驶ADAS算法设计及Prescan仿真(1): 前向碰撞预警FCW算法设计与simulink仿真

   前言: 本系列文章将对智能驾驶ADAS算法进行完整的介绍,并在simulink中搭建算法模型,并在Prescan中搭建场景并且与simulink联合仿真对ADAS算法进行验证.智能驾驶ADAS算法详解 ...

最新文章

1. HDU3791（二叉搜索树）
2. 《可扩展的艺术》内容回顾与作者采访
3. 一条路走不通了，就走另一条：
4. 2 时间格式化_U盘无法格式化怎么办 U盘无法格式化解决教程【详解】
5. 10分钟让你明白什么是ERP
6. cmake （4）引用子目录的库
7. 大油井隐藏的箱子_魔兽世界：历经版本变迁的5大隐藏boss！全部都经历过的才是大神...
8. RedisUtil - Redis功能介绍，五种数据类型的使用，Spring和Redis的集成
9. mosquitto支持websocket的使用方法
10. 银监局计算机英语面试题及答案,2012年国家公务员银监局面试真题及参考答案...
11. Nand_ECC_校验和纠错_详解
12. count是java关键字吗_你是一直认为 count(1) 比 count(*) 效率高么？
13. 线条边框简笔画图片大全_表情包丨表情包简笔画图片大全可爱
14. steam for linux 安装目录,我该如何安装Steam？
15. win10 + Ubuntu 20.04 LTS 双系统 引导界面美化
16. cocos2dx游戏-可爱的小精灵的各种用法大全
17. 使用 NGINX 搭建 RTMP 流媒体服务器实现直播功能
18. google chrome F11全屏快捷键
19. 【Block-Level Verification】 芯片开发通识_验证目标_ 验证语言_ 验证职业前景 _挑战和瓶颈_验证周期_功能描述文档_验证计划_回归测试_硅后测试_逃逸分析...
20. flash 的鼠标事件详解

热门文章

1. 隐藏电脑桌面desktop.ini文件图标方法
2. 怀仁在2021年高考云东成绩查询,2020年怀仁市云东中学高考喜报
3. 罗技驱动调节灵敏度方法介绍
4. 公交智能调度系统车载终端应用设计与实现
5. 将txt文本数据转换为json对象
6. [转载]面向 Java 开发人员的 db4o 指南: 超越简单对象
7. mysql 存储过程建表_MySQL 存储过程创建表
8. java微信天气查询接口,全国天气预报信息 API 接口
9. 博客！新人从零开始学习编程的滥觞！写给自己的灯塔
10. 台风大涡模拟三维可视化