基于YOLOv5的双目测距——代码讲解

本文讲解如何在YOLOv5中加入双目测距SGBM算法，实现目标检测和测距任务。

代码准备

1. 1 下载yolov5-v6.1代码

github 下载地址：https://github.com/bubbliiiing/yolov5-v6.1-pytorch

1. 2 加入SGBM脚本

将上图所示的测距脚本sgbm-video.py 和视频素材复制到yolov5-v6.1-pytorch的根目录上。将yolov5-v6.1的预训练文件yolov5_s_v6.1.pth拷贝到model_data文件夹下。

2 修改代码

2.1 修改predict脚本

• 创