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/