使用Yolov5离线检测图片

5 min read
  1. 下载Yolov5的代码

从GitHub的Yolov5存储库(https://github.com/ultralytics/yolov5)上下载最新的Yolov5代码。

  1. 准备数据集

将要检测的图片放在一个文件夹中，并将其路径添加到"yolov5/inference/images"文件夹中的"test.txt"文件。

  1. 下载预训练权重

从Yolov5模型储存库(https://github.com/ultralytics/yolov5/releases)中下载最合适的预训练权重。将权重文件放在"yolov5/inference"文件夹中。

  1. 运行推理脚本

运行以下命令，使用预训练权重和测试图片进行推理。

python yolov5/detect.py --source yolov5/inference/images --weights yolov5/inference/<weights> --conf 0.5 --save-txt --save-conf --exist-ok

其中，""表示预训练权重文件名。上面的命令会在yolov5/inference/output文件夹中生成检测结果图像，并存储检测结果文本文件。