钟二网络头像

钟二网络

探索SQL查询技巧、Linux系统运维以及Web开发前沿技术,提供一站式的学习体验

  • 文章92531
  • 阅读785646
首页 Linux 正文内容

Linux安装yolo

钟逸 Linux 2025-03-06 23:03:00 13

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 将在图像中检测并标记对象。

文章目录
    搜索