YOLO v5学习之YOLO V5的安装

1、下载YOLO v5的源码

首先要到YOLO v5的官网下载YOLO v5的源码,源码地址为:https://github.com/ultralytics/yolov5

或者
使用以下代码来下载源码并直接安装:
注意:直接使用以下代码安装的pytorch的是x86版本的,如果需要arm版本的请参考其他文章。

$ git clone https://github.com/ultralytics/yolov5

2、配置YOLO v5的环境

$ cd yolov5
$ pip install -r requirements.txt

为了使用方便,一般是建立一个python的虚拟环境来安装测试一些应用。
这里我使用了anaconda创建了一个python3.6.13版本名为py6的虚拟环境。

在虚拟环境配置yolov5环境完成后,如下图所示:
YOLO v5学习之YOLO V5的安装_第1张图片

3、测试yolo v5

在终端(py36虚拟环境下),运行以下代码,调用笔记本的摄像头进行yolo v5的识别。

python detect.py --source 0

YOLO v5学习之YOLO V5的安装_第2张图片
至此YOLO v5就安装完成了。

你可能感兴趣的:(毕设相关,YOLOv5,python,pytorch,机器学习)