YOLO(You Only Look Once)是一种实时目标检测算法,因其速度和准确性而广受认可。本文将详细介绍如何在 Linux 系统上安装 YOLO。
先决条件
在开始安装之前,确保您的系统满足以下先决条件:
Ubuntu 18.04 或更高版本
Python 3.6 或更高版本
CUDA 10.2 或更高版本
NVIDIA 显卡
安装依赖项
首先,安装以下依赖项:
sudo apt-get update
sudo apt-get install build-essential cmake pkg-config
sudo apt-get install libjpeg-dev libpng-dev libtiff-dev
sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev
sudo apt-get install libgstreamer1.0-dev libgstreamer-plugins-base1.0-dev
安装 YOLO
克隆 YOLO 存储库:
git clone https://github.com/ultralytics/yolov5.git
进入 YOLO 目录:
cd yolov5
安装要求:
pip install -r requirements.txt
安装 PyTorch
使用 CUDA 支持安装 PyTorch:
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
测试 YOLO
下载一个示例图像:
wget https://github.com/ultralytics/yolov5/releases/download/v4.0/data.zip
解压缩图像:
unzip data.zip
检测图像中的目标:
python detect.py --source data/images/zidane.jpg
YOLO 将在图像中检测并标记对象。