网站建设资讯

NEWS

网站建设资讯

linux下yolov5环境配置-创新互联

一、模型下载
1.首先从github上找到官方yolov5代码。
2.点击master下的tags选择下载的版本。
在这里插入图片描述
3.选择好版本后,点code下的Downloadzip。这里选择的是v6.0。
在这里插入图片描述

汉寿ssl适用于网站、小程序/APP、API接口等需要进行数据传输应用场景,ssl证书未来市场广阔!成为成都创新互联的ssl证书销售渠道,可以享受市场价格4-6折优惠!如果有意向欢迎电话联系或者加微信:028-86922220(备注:SSL证书合作)期待与您的合作!

4.下载权重参数文件
点击图中右下角:+9 releases
在这里插入图片描述
找到自己需要的版本
在这里插入图片描述
在对应的版本中下载权重文件,找到Asserts点击将其展开。
在这里插入图片描述
选择预训练权重文件。这里选择的是yolov5s.pt,下载并解压至yolov5-6.0文件夹中。

二、环境配置
1.查看yolov5-6.0中readme文件夹,确定python,pytorch的版本。
2.去pytorch官网找到对应的cuda版本。选择与自己计算机匹配或更低的cuda,这里选择的是pytorch=1.8.0,cuda=11.1。复制命令行。
在这里插入图片描述

3.创建虚拟环境
打开Anaconda prompt,创建YOLOv5虚拟环境,python版本是3.7

1  |conda create -n yolov5 python=3.7

4.激活环境
打开pycharm,终端输入命令行。

1  |conda activate yolov5

5.下载pytorch和对应版本的cuda。

1  |conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

6.修改pip安装路径。参考博客Conda虚拟环境用conda和pip安装软件包的路径问题
使得pip下载安装到创建的环境而非Base环境中,首先找到site.py文件它在anconda3/envs/yolo/lib/python3.7中。
路径如下
打开site.py文件可以看到如下内容
在这里插入图片描述
在终端激活自己的环境输入以下命令行得到当前环境搜索路径

1  |python -m site

在这里插入图片描述

将其复制粘贴到site.py文件中相应位置并保存。
在这里插入图片描述

7.下载requirements文件内容

1  |pip install -r requirements.txt

三、运行代码
终端输入

1  |python detect.py

得到运行结果。
在这里插入图片描述

在这里插入图片描述

你是否还在寻找稳定的海外服务器提供商?创新互联www.cdcxhl.cn海外机房具备T级流量清洗系统配攻击溯源,准确流量调度确保服务器高可用性,企业级服务器适合批量采购,新人活动首月15元起,快前往官网查看详情吧


分享名称:linux下yolov5环境配置-创新互联
网站地址:http://njwzjz.com/article/ccpeos.html