草庐IT

windows+Pycharm+yolov5 配置环境以及本地训练自己的模型

我与nano 2023-04-21 原文

参加比赛真的可以学到好多好多的东西,也是因为一次比赛,我接触到了模型训练,刚开始我采用openmv进行训练,效果很拉跨,接着通过查找一些资料,我发现了yolo,这个用来做目标检测效果很好,然后我就开始尝试,摸爬滚打好几天,最后终于成功了,网上的教程真的很不全,所以我想写一篇博客记录一下我这几天学习的过程。我写下这篇博客第一是为了记录我自己的学习生活。第二是为了让我的学弟学妹在学习以及正在学习模型训练的你能够少走一些弯路。

文章目录

前言

PyCharm是一种Python IDE(Integrated Development Environment,集成开发环境),带有一整套可以帮助用户在使用Python语言开发时提高其效率的工具,比如调试、语法高亮、项目管理、代码跳转、智能提示、自动完成、单元测试、版本控制。此外,该IDE提供了一些高级功能,以用于支持Django框架下的专业Web开发。
当然它也可以进行模型训练,跑yolo算法以及opencv。这篇博客主要给大家介绍怎么使用Pycharm进行模型训练。接下来我们进入正文。

一、前提准备

一台电脑
Pycharm软件:
https://www.jetbrains.com/pycharm/

二、环境配置

1.安装anaconda

https://www.anaconda.com/
安装完成后打开anaconda3.5.2

2.安装python3.6

输入

conda create -n pytorch python=3.6

这里我已经安装过了安装完成后显示

输入conda activate pytorch 进入pytorch环境

输入pip list 查看已经安装的包

这里没有pytorch接下来我们安装pytorch

3.安装pytorch

如果你的电脑没有英伟达的显卡输入:(在pytorch环境下输入)

conda install pytorch torchvision cpuonly -c pytorch

如果你的电脑有英伟达的显卡输入:(同样实在pytorch环境下)

conda install pytorch torchvision cudatoolkit -c pytorch -c defaults -c numba/label/dev

查看自己的显卡版本号(只有版本号大于396才支持安装)

输入指令

接下来就是漫长的等待…
安装完成后输入 pip list 查看已经安装的包

输入

python 
import torch
torch.cuda.is_available()


如果显示ture说明已经安装成功了

三、项目环境配置

1.下载yolov5官方项目

选择yolov5-5.0版本

https://github.com/ultralytics/yolov5

2.配置模型训练环境

使用Pycharm打开yolov5-5.0

打开后第一步点击设置配置python解释器


选择你们刚才配置的解释器,回到主界面,点击requirements.txt

在最上面一行你会看到pip语句
打开终端输入 pip install -r requirements.txt

在这个txt文件下,官方把所需环境全都给出了直接输入就可以安装所需环境了。
运行detect.py文件


在runs下你就可以看到所预测的图片。

四、自制数据集训练自己的模型

1.制作数据集

1.图片的采集

采用手机拍照或者走网上搜索你所需要的数据集。

2.图片的标注

在这里给大家推荐一个在线标注工具:https://www.makesense.ai/

vott也是一个很好用的工具,这个可能更加方便一点
点击Get Started

然后选择自己所要训练的图片,点击Object Detection(对象检测)

输入你所需要的标签,如果你的种类较多可以enter继续输入

画框打上标签,如图所示

导出模型标签

选择第一个yolo模型(如果你是用K210进行训练的话选择第二个)

然后导出,你可以看到你需要的标签已经下载下来了

打开之后在这个文件夹下会有很多文本文档(这些文本文档就是标注好的)

五、开始训练

1.新建文件夹储存数据集图片和标签


新建一个mydate文件夹,在这个文件夹下新建image和labels文件夹,image文件夹下放图片,labels文件夹下放刚才标注过的文本文档(这里一定要注意图片和导出的文本文档一定要对应)

2.在date文件夹下新建mydate.yaml文件


打开coco128.yaml文件,将里面的内容复制到mydate.yaml文件下然后进行修改

我之前用数码管训练的所以就没有修改

3.进入train.py开始训练

修改train.py文件,这里我不多赘述,直接放图,修改成和我一样就可以运行了,这里主要是放你新建的mydate.yaml文件根目录

点击运行

然后就开始训练了

train.py文件里的参数在这里我不做过多的介绍,网上这个介绍真的太多了。感谢大家看到最后一行。

总结

以上就是我进行模型训练的一些学习过程,大家在看文章的时候哪个地方遇到问题了可以在下面留言基本上看到我都会回复,最后我想以一句话结束这篇博客。
***现在的日积月累,迟早会变成别人的望尘莫及。***加油

有关windows+Pycharm+yolov5 配置环境以及本地训练自己的模型的更多相关文章

  1. ruby - 在 Ruby 程序执行时阻止 Windows 7 PC 进入休眠状态 - 2

    我需要在客户计算机上运行Ruby应用程序。通常需要几天才能完成(复制大备份文件)。问题是如果启用sleep,它会中断应用程序。否则,计算机将持续运行数周,直到我下次访问为止。有什么方法可以防止执行期间休眠并让Windows在执行后休眠吗?欢迎任何疯狂的想法;-) 最佳答案 Here建议使用SetThreadExecutionStateWinAPI函数,使应用程序能够通知系统它正在使用中,从而防止系统在应用程序运行时进入休眠状态或关闭显示。像这样的东西:require'Win32API'ES_AWAYMODE_REQUIRED=0x0

  2. ruby-on-rails - Rails - 子类化模型的设计模式是什么? - 2

    我有一个模型:classItem项目有一个属性“商店”基于存储的值,我希望Item对象对特定方法具有不同的行为。Rails中是否有针对此的通用设计模式?如果方法中没有大的if-else语句,这是如何干净利落地完成的? 最佳答案 通常通过Single-TableInheritance. 关于ruby-on-rails-Rails-子类化模型的设计模式是什么?,我们在StackOverflow上找到一个类似的问题: https://stackoverflow.co

  3. ruby - 什么是填充的 Base64 编码字符串以及如何在 ruby​​ 中生成它们? - 2

    我正在使用的第三方API的文档状态:"[O]urAPIonlyacceptspaddedBase64encodedstrings."什么是“填充的Base64编码字符串”以及如何在Ruby中生成它们。下面的代码是我第一次尝试创建转换为Base64的JSON格式数据。xa=Base64.encode64(a.to_json) 最佳答案 他们说的padding其实就是Base64本身的一部分。它是末尾的“=”和“==”。Base64将3个字节的数据包编码为4个编码字符。所以如果你的输入数据有长度n和n%3=1=>"=="末尾用于填充n%

  4. ruby-on-rails - Rails - 一个 View 中的多个模型 - 2

    我需要从一个View访问多个模型。以前,我的links_controller仅用于提供以不同方式排序的链接资源。现在我想包括一个部分(我假设)显示按分数排序的顶级用户(@users=User.all.sort_by(&:score))我知道我可以将此代码插入每个链接操作并从View访问它,但这似乎不是“ruby方式”,我将需要在不久的将来访问更多模型。这可能会变得很脏,是否有针对这种情况的任何技术?注意事项:我认为我的应用程序正朝着单一格式和动态页面内容的方向发展,本质上是一个典型的网络应用程序。我知道before_filter但考虑到我希望应用程序进入的方向,这似乎很麻烦。最终从任何

  5. ruby - 使用 C 扩展开发 ruby​​gem 时,如何使用 Rspec 在本地进行测试? - 2

    我正在编写一个包含C扩展的gem。通常当我写一个gem时,我会遵循TDD的过程,我会写一个失败的规范,然后处理代码直到它通过,等等......在“ext/mygem/mygem.c”中我的C扩展和在gemspec的“扩展”中配置的有效extconf.rb,如何运行我的规范并仍然加载我的C扩展?当我更改C代码时,我需要采取哪些步骤来重新编译代码?这可能是个愚蠢的问题,但是从我的gem的开发源代码树中输入“bundleinstall”不会构建任何native扩展。当我手动运行rubyext/mygem/extconf.rb时,我确实得到了一个Makefile(在整个项目的根目录中),然后当

  6. ruby-on-rails - 在混合/模块中覆盖模型的属性访问器 - 2

    我有一个包含模块的模型。我想在模块中覆盖模型的访问器方法。例如:classBlah这显然行不通。有什么想法可以实现吗? 最佳答案 您的代码看起来是正确的。我们正在毫无困难地使用这个确切的模式。如果我没记错的话,Rails使用#method_missing作为属性setter,因此您的模块将优先,阻止ActiveRecord的setter。如果您正在使用ActiveSupport::Concern(参见thisblogpost),那么您的实例方法需要进入一个特殊的模块:classBlah

  7. ruby-on-rails - 如何验证非模型(甚至非对象)字段 - 2

    我有一个表单,其中有很多字段取自数组(而不是模型或对象)。我如何验证这些字段的存在?solve_problem_pathdo|f|%>... 最佳答案 创建一个简单的类来包装请求参数并使用ActiveModel::Validations。#definedsomewhere,atthesimplest:require'ostruct'classSolvetrue#youcouldevencheckthesolutionwithavalidatorvalidatedoerrors.add(:base,"WRONG!!!")unlesss

  8. ruby-on-rails - form_for 中不在模型中的自定义字段 - 2

    我想向我的Controller传递一个参数,它是一个简单的复选框,但我不知道如何在模型的form_for中引入它,这是我的观点:{:id=>'go_finance'}do|f|%>Transferirde:para:Entrada:"input",:placeholder=>"Quantofoiganho?"%>Saída:"output",:placeholder=>"Quantofoigasto?"%>Nota:我想做一个额外的复选框,但我该怎么做,模型中没有一个对象,而是一个要检查的对象,以便在Controller中创建一个ifelse,如果没有检查,请帮助我,非常感谢,谢谢

  9. ruby-on-rails - 独立 ruby​​ 脚本的配置文件 - 2

    我有一个在Linux服务器上运行的ruby​​脚本。它不使用rails或任何东西。它基本上是一个命令行ruby​​脚本,可以像这样传递参数:./ruby_script.rbarg1arg2如何将参数抽象到配置文件(例如yaml文件或其他文件)中?您能否举例说明如何做到这一点?提前谢谢你。 最佳答案 首先,您可以运行一个写入YAML配置文件的独立脚本:require"yaml"File.write("path_to_yaml_file",[arg1,arg2].to_yaml)然后,在您的应用中阅读它:require"yaml"arg

  10. ruby-on-rails - 如何将验证与模型分开 - 2

    我有一些非常大的模型,我必须将它们迁移到最新版本的Rails。这些模型有相当多的验证(User有大约50个验证)。是否可以将所有这些验证移动到另一个文件中?说app/models/validations/user_validations.rb。如果可以,有人可以提供示例吗? 最佳答案 您可以为此使用关注点:#app/models/validations/user_validations.rbrequire'active_support/concern'moduleUserValidationsextendActiveSupport:

随机推荐