(yolov5在Windows下搭建环境运行检测demo)

前话

本文简单介绍在Windows下简单搭建Pytorch 版yolov5的检测环境,下载代码,测试安装环境是否正确,为之后自己训练,修改源码做基础。

1、下载代码

Github链接地址:

下载链接: yolov5下载.

2、创建虚拟环境下载依赖包

①使用conda命令创建虚拟环境:
conda create -n yolov5 python==3.8

②进入创建的虚拟环境:
conda activate yolov5

③按照官方要求下载依赖包:
pip install -r requirements.txt

3、下载预训练权重文件

Windows下终端不好下载预训练权重文件可以直接在github官网中下载:
链接: yolov5预训练权重文件下载地址.

在这里插入图片描述
将下载到的预训练权重放在下载的代码包weight文件夹下:

在这里插入图片描述

4、修改测试文件读入参数,运行测试

① 目前只是为了测试搭建环境的可行性,简单修改detect.py文件中参数,简单读入图片进行检测就好

在这里插入图片描述

② 会读取data文件夹中的图片检测(可以放进去自己的图片测试),并把运行结果保存在runs文件夹中:
在这里插入图片描述

③ 在conda终端cd到yolov5所在的文件夹地址,输入命令测试运行:

python detect.py

运行结果,在runs文件夹中:

在这里插入图片描述

根据官方公布,还可以检测webcam读入视频流等:

python detect.py --source 0

在这里插入图片描述

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐