yolov5图片及视频检测推理

yolov5检测+推理

说明

本文环境为vscode，anaconda下的yolo环境，代码为官网yolov5-7.0

切换环境

在训练及检测之前需切换至搭建的对应的yolo环境

1.检测代码为detect.py

如你要检测你自己的图片，需已成功训练对应数据集
这是我训练官网数据集得到的结果，此路径下的best.pt便是此次训练的最优模型文件

2.更改模型及需检测的图片的路径

打开detect代码，找到我框的weights和source参数，weights是模型文件的路径，source是需要检测的图片或者视频的路径

改正后的效果图 (路径注意使用 / )

3.接着运行detect.py，得到检测结果

打开对应的路径即可看到检测结果 (注:如果个别图片没有检测出来,说明模型的精度不高,需要重新训练模型，加轮数或者加图片，和环境本身没有太大关系)

检测视频同理,把存放视频的路径输到source参数即可