基于yolov5的目标检测和单目测距

废话在前头

  因为我的毕业设计就是搞目标检测这一块的,前段时间好不容易实现了yolov5的目标检测,后来还需要实现目标测距,于是在B站和博客里不断摸索,效果很不理想,知识点零散,幸运的是最终琢磨懂了代码和原理。全文通俗易懂。文章被限制,需要原文截图的点这里百度网盘拿!!!

正文开始

1、最终目的

  实现目标检测跟踪和目标测距 !

2、实现过程

2.1 实现的前提

  前提是你要会用yolov5实现目标检测,我是跟着一位博主学做出来的——目标检测—教你利用yolov5训练自己的目标检测模型 。重点是在B站里也有他做的 保姆级视频本文着重讲解单目测距。 实现了yolov5检测之后直接去我的gitee上下载我的测距项目: yolov5-main。下载并解压项目到PyCharm中,配置上运行yolov5的环境,然后跑一遍detect.py,看看我的案例效果,成功运行后再跟着教程慢慢搞。请添加图片描述

2.2 知识点简要梳理

  因为条件的限制,我没有选择双目测距来研究,其实单目和双目各有优缺点。它们的原理网上搜也是一大把,本文直接跳过原理讲解,我们来讲公式:

D = (F*W)/P
其中D是目标到摄像机的距离(最终要推测的距离), F是摄像机焦距(根据下面教程来调试自己的镜头焦距), W是目标的宽度或者高度(看情况来定,行人检测一般以人的身高为基准), P是指目标在图像中所占据的x方向像素(宽)或者y方向像素(高)(程序来计算)。
2.3 单目测距代码

  先来看看核心代码

foc = 1810.0        # 镜头焦距
real_hight_person = 64.96   # 行人高度,注意单位是英寸

# 自定义函数,单目测距
def detect_distance_person(h):
    dis_inch = (real_hight_person * foc) / (h - 2)<
  • 57
    点赞
  • 431
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 101
    评论
这个任务需要涉及到YOLOv5目标检测单目测距两个部分,下面给出一个大致的实现步骤: 1. 安装YOLOv5并下载预训练权重。可以参考官方文档进行安装和权重下载。 2. 读取图像或视频,并使用YOLOv5进行目标检测。这里可以使用官方提供的detect.py脚本,也可以使用Python代码调用YOLOv5模型进行目标检测。 3. 对检测到的目标进行单目测距。可以使用OpenCV中的相机标定方法获取相机内参和外参等参数,然后通过目标在图像中的位置和相机参数,计算目标到相机的距离。具体实现可以参考OpenCV官方文档。 4. 将检测到的目标和距离信息绘制在图像上。可以使用OpenCV中的绘图函数进行绘制。 下面是一个大致的Python代码实现: ```python import cv2 import numpy as np import torch # 安装YOLOv5并下载预训练权重,这里使用torch.hub方式加载模型 model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True) # 读取图像或视频 cap = cv2.VideoCapture('video.mp4') # 相机参数设置(可以通过相机标定获取) fx = 1000 fy = 1000 cx = 960 cy = 540 while True: ret, frame = cap.read() if not ret: break # 使用YOLOv5进行目标检测 results = model(frame) # 获取检测到的目标信息 boxes = results.xyxy[0].numpy() confs = results.xyxy[0][:, 4].numpy() classes = results.xyxy[0][:, 5].numpy().astype(np.int) # 对每个目标进行单目测距 for i in range(len(boxes)): x1, y1, x2, y2 = boxes[i] conf = confs[i] cls = classes[i] # 计算目标到相机的距离 w = x2 - x1 h = y2 - y1 z = (fx * w) / (2 * (x2 - cx)) distance = z / 1000 # 将单位转换为米 # 在图像上绘制目标和距离信息 cv2.rectangle(frame, (int(x1), int(y1)), (int(x2), int(y2)), (0, 255, 0), 2) cv2.putText(frame, f'{cls} {distance:.2f}m', (int(x1), int(y1 - 10)), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2, cv2.LINE_AA) cv2.imshow('frame', frame) if cv2.waitKey(1) == ord('q'): break cap.release() cv2.destroyAllWindows() ``` 注意:这里只是一个简单的示例代码,实际运行可能还需要进行一些调整和优化。另外,由于YOLOv5模型是通过PyTorch实现的,因此在安装和使用时需要注意PyTorch版本和环境的配置。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 101
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Lange小蓝子

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值