YOLOv5 是一种非常流行的实时目标检测模型,它提供了出色的性能和精度。下面是一个使用 Python 和 YOLOv5 进行目标检测的基本应用案例。
步骤 1:安装 YOLOv5 和相关依赖
首先,确保你安装了所有必要的依赖。这通常包括 torch
、torchvision
和 opencv-python
。你可以使用 pip
来安装这些库:
bash复制代码
pip install torch torchvision opencv-python |
然后,你需要从 YOLOv5 的官方 GitHub 仓库克隆代码并安装:
bash复制代码
git clone https://github.com/ultralytics/yolov5.git |
|
cd yolov5 |
|
pip install -r requirements.txt |
步骤 2:下载预训练模型
YOLOv5 提供了多个预训练模型,你可以从官方仓库或相关资源中下载。例如,你可以下载最小的 yolov5s.pt
模型:
bash复制代码
wge |