【YOLOv5代码整理】

YOLO v5代码整理


1 yolo前期准备-安装pytorch
注意,需要安装pytorch 1.6版本以上
创建虚拟环境命令行

conda create -n my_pt38 python=3.8

激活my_pt38环境:

conda activate my_pt38

在所创建的环境下安装pytorch的1.6版本以上(官网https://pytorch.org/),执行命令:
独立显卡

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

独立显卡

conda install pytorch torchvision torchaudio cpuonly -c pytorch

(查看cuda版本,命令行 nvcc -V
{查看Pytorchb版本:

import torch
torch._version_

}
2 yolo v5项目克隆和安装

2.1 克隆YOLO v5项目

安装Git软件(https://git-scm.com/),克隆项目到本地

git clone https://github.com/ultralytics/yolo5.git

在这里插入图片描述

00 训练自己的数据集
00.1 数据打标签。labelImg可以直接利用命令行安装

pip install labelImg

加载yolo v5 所需环境
pip install -r requirements.txt
建议添加镜像

pip install -r requirements.txt -i https://pypi.doubanio.com/simple/
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值