【实验|个人实现】Pytorch搭建YoloV5目标检测平台

【主要内容】

Pytorch搭建YoloV5目标检测平台

github源码

【实验过程】

步骤一：torch环境搭建与配置

1、在Anaconda创建虚拟环境

conda create -n zj python==3.7.5

激活环境，前面就会变成（zj）,成功进入环境

conda activate zj

2、安装torch（进Pytorch官网找对应命令，↓CPU）

conda install pytorch torchvision torchaudio cpuonly -c pytorch

3、接下来就是安装各种包，两种方式，一种是在上述基础上进行，另一种是在项目terminal里。

conda install xxx
conda install xxx==版本

conda uninstall xxx

如果要查看当前环境下安装了什么：

conda list

 步骤二：yolo_v5

1、下载yolov5，导入

用pycharm打开 yolov5-pytorch-main，然后去setting里导入：project->Python Interpreter -> Conda Envir -> Existing Envir->找到Anaconda安装目录下envs\zj\python.exe

 2、在requirements.txt里会显示所有需要安装的内容，可点击Install，也可命令

        conda install xxx 或 pip install xxx（如出错，可能是==的版本太详细找不到 直接名字即可）

3、按照要求，将权值参数文件下载放到对应的位置