YOLOv5 环境配置和安装

本文详细介绍了如何在Anaconda环境下创建Python 3.8的虚拟环境,安装Pytorch及其相关库如numpy、Pillow、torch、torchaudio和torchvision,以及YOLOv5的依赖。通过遵循github上的requirements文件安装,并了解如何在Linux服务器上后台运行程序,进行进程管理和测试推理检测。

系列文章

yolov5-github

 Anaconda创建虚拟环境

使用anaconda创建一个python 3.8 的虚拟环境,详细步骤参见下文

Win10 + Anaconda + Python + Jupyter notebook + Pip + TensorFlow + PyCharm

安装pytorch

Pytorch GET STARTED

pip3 install torch torchvision torchaudio

Successfully installed numpy-1.20.3 pillow-8.2.0 torch-1.8.1 torchaudio-0.8.1 torchvision-0.9.1 typing-extensions-3.10.0.0

  • numpy:NumPy(Numerical Python) 是 Python 语言的一个扩展程序库,支持大量的维度数组与矩阵运算,此外也针对数组运算提供大量的数学函数库。
  • pillow
### YOLOv5 环境配置 PyCharm 安装教程 #### 一、准备工作 为了顺利配置YOLOv5开发环境,在PyCharm中需先完成软件安装工作。确保已下载并安装最新版本的PyCharm社区版或专业版[^2]。 #### 二、创建项目与虚拟环境 启动PyCharm后,通过`File -> New Project`来新建一个Python项目。建议勾选`Create virtual environment using:`选项,并选择Conda作为解释器类型以便后续管理依赖包更方便[^1]。 #### 三、安装必要的库工具 ##### 3.1 CUDA cuDNN 的安装 对于希望利用GPU加速训练过程的情况来说,提前准备好兼容版本的CUDA Toolkit以及cuDNN是非常重要的前置条件之一。具体操作可以参照NVIDIA官方文档说明进行设置。 ##### 3.2 PyTorch安装 打开终端窗口(Terminal),输入命令以安装适合当前系统的PyTorch版本: ```bash pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 ``` 注意这里假设使用的是CUDA 11.3版本;如果不是,则需要调整URL中的版本号匹配实际使用的CUDA版本。 #### 四、克隆YOLOv5仓库并安装需求文件 继续在同一终端内执行如下指令获取YOLOv5源码及其所需的所有外部库: ```bash git clone https://github.com/ultralytics/yolov5.git cd yolov5 pip install -r requirements.txt ``` 这一步骤会自动拉取GitHub上的YOLOv5开源项目到本地目录下,并按照requirements.txt列表批量安装所需的第三方模块。 #### 五、加载预训练模型权重 访问Ultralytics官方网站或其他可信渠道下载对应于目标检测任务预先训练好的.pt格式参数文件,将其放置于项目的根路径或是指定位置供后续调用。 #### 六、验证安装成功与否 最后可以通过运行示例脚本来检验整个流程是否顺利完成。比如尝试预测一张图片内的物体类别: ```python from pathlib import Path import sys FILE = Path(__file__).resolve() ROOT = FILE.parents[0] if str(ROOT) not in sys.path: sys.path.append(str(ROOT)) # add ROOT to PATH import torch from models.experimental import attempt_load from utils.general import non_max_suppression, scale_coords from utils.torch_utils import select_device weights = 'yolov5s.pt' # 权重文件名 imgsz = 640 # 输入图像大小 device = '' # 使用CPU还是GPU ('cpu', '', or '0') model = attempt_load(weights, map_location=device) stride = int(model.stride.max()) # model stride names = model.module.names if hasattr(model, 'module') else model.names # class names print(f"Model loaded successfully with {len(names)} classes.") ``` 上述代码片段展示了如何加载YOLOv5模型并对给定图像实施推理处理的过程。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值