【YOLOv5代码整理】

news/2024/7/21 6:09:05 标签: pytorch, 深度学习, python, 视觉检测, 图像处理

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版本:

python">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可以直接利用命令行安装

python">pip install labelImg

加载yolo v5 所需环境
pip install -r requirements.txt
建议添加镜像

python">pip install -r requirements.txt -i https://pypi.doubanio.com/simple/

http://www.niftyadmin.cn/n/970692.html

相关文章

Web 地理定位(Geo-Location)知识大全

在线地图已经是很平常的应用,Google Map 甚至可以用作驾驶导航仪,然而,这些应用的实现涉及到千丝万缕的艰辛工作,仅仅是定位你现在的位置,就有很多技术,有幸的是,如今在你的 Web 站点中&#xf…

【已解决】AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor’

项目yolo v5中,终端执行detect.py文件时出现AttributeError属性报错,解决如下: 1 光标定位Upsampling.py,点击报错跳转 2 如行154示,直接删除or注释, recompute_scale_factorself.recompute_scale_factor …

为你写的心情,我整理成了一篇文章,释放我悠悠的思念...(3)

习惯了不跟你联系,习惯了默默的等着你的消息,习惯了被动的了解你的变化就这样,你说我这是有心病。要带我去庙里住上几天我很期待...未完待续转载于:https://blog.51cto.com/victory15/344944

经过一年时间的沉淀 再次回首 TCP Socket服务器编程 (二)

------------------ 前言 ------------------ 发了第一篇文章后,有不少同志留言,看来socket编程仍然是软件系统里面一个比较难的部分。 第一篇文章主要介绍了传输协议的设计,这个是整个socket框架最底层基础的部分,接下来整个sock…

解决IE常见故障

IE浏览器常见经典故障 当你驾着心爱的IE在Internet上纵横驰骋时,突然桌面上弹出一个莫名其妙的警告提示框,紧接着所有IE窗口全部关闭;或无论如何点击链接,IE始终没有反应……这些故障让人心烦意乱,无可奈何。不过你也不…

Python CV 视觉图像数据标签(xml)在原图的可视化操作

CV离不开数据集,对于拿到公开的数据集抑或是人家整理的数据集时,我们有必要清楚现有的数据集标签质量如何。因此,我们可以用以下脚本去对我们现有的数据标签在原图上进行可视化操作。 本人基于企业实际工程项目,以.xml标签数据可…

Redhat_rh033_g

Redhat_rh033_g文本处理工具:抽取文本的工具:1.)文件内容:less和cat2.)文件截取:head和tail3.)按列抽取:cut4.)按关键字抽取:grep查看文件内容:cat:通过输出显示一个或多个文件-A …

文件结构树的├、└、─ 符号

文件结构树的├、└、─ 符号 ├─ 一级目录1 └─ 一级目录2 └─二级目录1 └─二级目录2 注:csdn的空格键编辑,英文键入 【直接复制使用】 后续查证后继续更新!