yolov9直接调用zed相机实现三维测距(python)

yolov9直接调用zed相机实现三维测距(python)

  • [1. 相关配置](#1. 相关配置)
  • [2. 相关代码](#2. 相关代码)
    • [2.1 相机设置](#2.1 相机设置)
    • [2.2 测距模块](#2.2 测距模块)
    • [2.2 实验结果](#2.2 实验结果)

相关链接

此项目直接调用zed相机实现三维测距,无需标定,相关内容如下:

1. yolov4直接调用zed相机实现三维测距
2.yolov5直接调用zed相机实现三维测距(python)
3. yolov8直接调用zed相机实现三维测距(python)
4.具体实现效果已在哔哩哔哩发布,点击此链接跳转

本篇博文工程源码下载(麻烦github给个星星)
下载链接:https://github.com/up-up-up-up/zed-yolov9

附:Zed调用YOLOv7测距也已经实现,但是3060笔记本6G显存带不动,在大现存服务器上可以运行,可能是由于YOLOv7网络结构导致的,由于不具备普适性,就不再写相关文章了,有需要的可以仿照这个代码去改写

1. 相关配置

python==3.7

Windows-pycharm

zed api 具体配置见 (zed api 配置步骤)

2. 相关代码

2.1 相机设置

python 复制代码
zed = sl.Camera()
input_type = sl.InputType()
 if opt.svo is not None:
     input_type.set_from_svo_file(opt.svo)

 # Create a InitParameters object and set configuration parameters
 init_params = sl.InitParameters(input_t=input_type, svo_real_time_mode=True)
 init_params.camera_resolution = sl.RESOLUTION.HD720
 init_params.coordinate_units = sl.UNIT.METER
 init_params.depth_mode = sl.DEPTH_MODE.ULTRA  # QUALITY
 init_params.coordinate_system = sl.COORDINATE_SYSTEM.RIGHT_HANDED_Y_UP
 init_params.depth_maximum_distance = 5

 runtime_params = sl.RuntimeParameters()
 status = zed.open(init_params)

2.2 测距模块

python 复制代码
for *xyxy, conf, cls in reversed(det):
     xywh = (xyxy2xywh(torch.tensor(xyxy).view(1, 4)) / gn).view(-1).tolist()  # normalized xywh
     cent_x = round(xywh[0] * im0.shape[1])
     cent_y = round(xywh[1] * im0.shape[0])
     cent_w = round(xywh[2] * im0.shape[1])
     point_1 = round(cent_x - 0.4 * cent_w)
     point_2 = round(cent_x + 0.4 * cent_w)
     wide_value_1 = point_cloud.get_value(point_1, cent_y)[1]
     wide_value_2 = point_cloud.get_value(point_2, cent_y)[1]

     try:
         wide = round(wide_value_1[0], 4) - round(wide_value_2[0], 4)
         wide = round(abs(wide * 1000))
     except:
         wide = 0.00
         pass
     point_cloud_value = point_cloud.get_value(cent_x, cent_y)[1]
     point_cloud_value = point_cloud_value * -1000.00
     if point_cloud_value[2] > 0.00:
         try:
             point_cloud_value[0] = round(point_cloud_value[0])
             point_cloud_value[1] = round(point_cloud_value[1])
             point_cloud_value[2] = round(point_cloud_value[2])
             print("x:", point_cloud_value[0], "y:", point_cloud_value[1], "z:",
                   point_cloud_value[2], "W:", wide)
             txt = 'x:{0} y:{1} z:{2} w:{3}'.format(point_cloud_value[0], point_cloud_value[1],
                                                    point_cloud_value[2], wide)
             a=point_cloud_value[0]
             b=point_cloud_value[1]
             c=point_cloud_value[2]
             distance = ((a ** 2 + b ** 2 + c ** 2) ** 0.5)

             # annotator.box_label(xyxy, txt, color=(255, 0, 0))
             label = f'{names[int(cls)]} {conf:.2f} '
             label = label + " " +"dis:" +str(distance)
             annotator.box_label(xyxy, label, color=colors(c, True))

         except:
             pass

2.2 实验结果

测距功能

视频展示

相关推荐
Swift社区1 小时前
在 Swift 中实现字符串分割问题:以字典中的单词构造句子
开发语言·ios·swift
没头脑的ht1 小时前
Swift内存访问冲突
开发语言·ios·swift
没头脑的ht1 小时前
Swift闭包的本质
开发语言·ios·swift
wjs20241 小时前
Swift 数组
开发语言
stm 学习ing2 小时前
FPGA 第十讲 避免latch的产生
c语言·开发语言·单片机·嵌入式硬件·fpga开发·fpga
湫ccc3 小时前
《Python基础》之字符串格式化输出
开发语言·python
mqiqe3 小时前
Python MySQL通过Binlog 获取变更记录 恢复数据
开发语言·python·mysql
AttackingLin3 小时前
2024强网杯--babyheap house of apple2解法
linux·开发语言·python