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

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

代码准备

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

基于YOLOv5的双目测距——代码讲解_第1张图片
github 下载地址:https://github.com/bubbliiiing/yolov5-v6.1-pytorch

1. 2 加入SGBM脚本

基于YOLOv5的双目测距——代码讲解_第2张图片
将上图所示的测距脚本sgbm-video.py 和视频素材复制到yolov5-v6.1-pytorch的根目录上。将yolov5-v6.1的预训练文件yolov5_s_v6.1.pth拷贝到model_data文件夹下。
基于YOLOv5的双目测距——代码讲解_第3张图片

2 修改代码

2.1 修改predict脚本

你可能感兴趣的:(object,detection,python,深度学习,目标检测)