yolo模型训练工具分享

📅 2026/7/9 11:47:54
yolo模型训练工具分享
前言2026年2月20日我租下了人生中第一台阿里云服务器。虽然配置仅为2核2G但已能满足我的需求。起初我被YOLO模型的训练过程所困扰——不仅需要下载各种依赖、配置复杂环境还耗费了我整整一上午才勉强搞定。更令人沮丧的是这套方法换一台电脑就可能失效。此外训练模块本身也是一大挑战想要得到一个好模型必须划分数据集、选择合适的预训练模型并调整大量参数。整套流程对新手而言实在不够友好。于是我萌生了一个想法制作一个图形化的YOLO训练界面。最初方案是打包成单个exe文件用户只需双击即可运行。完成后我又想如果能支持远程发布和版本更新岂不是更方便然而当我搭建好服务器并准备上传时却发现这个exe文件在其他电脑上无法运行。查阅资料后才知道是因为缺少运行依赖。随后我将所有依赖都打包进同一个exe文件。但新问题接踵而至Windows系统对可执行文件的大小有限制2G而我的文件已达1.9G。虽然能够打开但启动速度极其缓慢。于是我再次调整思路转向制作安装包。在Windows上我采用Inno Setup制作exe安装包在Linux上则采用deb格式制作安装包。最终我将安装包的大小优化为1.92GWindows和2.42GLinux。虽然体积较大但安装后即可直接使用无需配置环境和依赖且在不同电脑上都能正常运行。后来我开始制作网站来宣传我开发的软件搭建数据库并逐步完善网站的各种功能。至今我的网站已具备下载功能、版本信息查看功能、常见问题解答功能、登录功能、群聊功能、赞助功能以及问题反馈功能。软件介绍目前只有模型训练功能和设置是完善的其他功能还在进一步开发中。该软件内部会自动识别CPU和显卡优先使用显卡进行训练安装包里内置各个版本的预训练模型可以随时切换选择内部自动对数据集进行划分训练集85%验证集15%并且支持xml,txt,json这三种格式的数据集内部会自动识别并进行转换注意如果你使用的是yolo官方格式的数据集(.txt格式)则不会进行检查和进行转换其余格式则会进行检测判断类别名称是否填写完整。另外你还可以在设置中切换ui主题选择自己喜欢的样式你甚至可以上传一张图片作为背景图(图片储存在本地并不会上传到服务器)你也可以保存当前的训练参数设置以及ui设置方便你下载训练时可以直接使用。使用教程1.在数据集配置栏选择图片目录和标注目录以及最终模型保存的路径。需要注意的是图片文件夹中只能存放你采集的照片标注目录文件夹中只能存放你标注好的数据并且每张图片的名字要和其标注文件的名字相同(除了后缀)。如果你没有填写类别名称直接点击开始训练时如果你的标注数据格式是xml或者json就会出现如上图那样软件会提示你输入类别名称你可以直接复制进去但要保证格式正确一行一个类别名称如果你的标注数据格式为txt的话可能就没有这些检查会直接进行训练最终你得到的模型什么都识别不到。2.配置完数据集后就可以进行调整模型参数选择你的预训练模型也就是想要训练什么版本什么类型的模型预训练模式是什么这里只说明教程不过多赘述。你可以点击预训练模型那里来进行选择模型也可以点击自定义按钮选择自己的模型你也可以将自己的模型复制到软件安装的目录内这样软件中就会显示出你复制过去的模型你也可以在安装目录删除不需要的模型。你可以自己调节训练轮次轮次越少训练越快这里默认为2000你不用担心它训练很慢因为yolo有早停机制当近100轮训练中模型没有太大的变化时会自己结束训练并将效果最好的模型导出所以你可以放心的调大训练轮次这里2000就是一个比较合适的参数。图片尺寸一般为640或320320训练的更快640更精确,这里默认为640如果你是新手可以直接使用这个参数。批次大小是每轮训练有多少张图片同时进行训练这个参数直接影响着你的模型训练速度默认为16因为这个参数兼容大部分电脑。如果你的显卡很好比如4060你可以直接调整为32但如果你调得过大可能会崩溃所以这个参数还是要根据图片数量和电脑配置来做决定mosaic增强这个参数请自行查找资料默认为1.0新手可以直接使用不会有影响。断点续训当你训练模型到一半突然电脑关机或者软件崩溃时你可以将训练了一半的模型继续拿出来训练这极大地节省了时间但要注意的是你需要在预训练模型那里选择你中断的这个模型一般是在模型保存路径中train/weights文件夹内名为last.pt选择这个模型作为预训练模型然后再勾选断点续训就可以直接点击开始训练啦模型参数可以不填模型会按照上次的参数配置进行继续训练。注意数据集配置中的参数还是要重新选择。图片目录和标注目录要和中断时的目录一样。3.数据集配置和模型参数填写完毕后就可以直接点击开始训练啦如下图模型训练完成后会自动将train文件保存到你填写的模型保存路径如果训练过程中有报错也会在训练日志中提示以便查明原因。最后你可以在设置中点击保存配置这样下次打开软件就还是现在的配置。至此教程结束说明使用此软件需要进行联网不联网无法使用用户的图片以及模型等信息不会上传至服务器服务器只记录用户的累计使用软件时长只用于统计无其他用途。本软件拥有自更新功能用户可以在应用内进行更新。本软件支持Windows和Linux(Ubuntu)系统。如果你需要不联网使用此软件功能或者需要定制的版本可以去官网联系开发者此软件工作量巨大本人不仅要维护网站还要维护windows和linux系统软件正常使用再加上本人正常参与学校竞赛所以更新的时间很少请见谅。如果你有任何的问题可以前往官方加入群聊或者点击问题反馈功能填写表单进行反馈问题。官网yolotool.top 软件下载地址yolotool.top以上内容如有说的不对的地方或有侵权行为请指出谢谢