YOLO自制数据集及训练

使用 Make Sense 网站进行标注

python 复制代码
https://www.makesense.ai/

可以让AI帮你先标一下

一定要点一下 + ,不然不会加进去

导出标签



长这个样子

按照你标签的顺序

分别是 car tree tree tree

标签格式

类 x中心 y中心 w h

xy坐标归一化到(0,1)

文件夹格式

布置训练集

在 my_data 文件夹下分别建立 images labels

修改--data (.yaml)

发现coco中没有 tree 类别,故删掉 ,添加

修改这里

修改成

训练时,报了个错

UnicodeDecodeError: 'gbk' codec can't decode byte 0xb7 in position 318: illegal multibyte sequence

原因:.yaml 文件里不能有中文标注,去掉就好嘞了。
其他问题,先按照上一篇文章的修改错误部分进行修改,之后就可以运行了

SPPF

python 复制代码
https://blog.csdn.net/Steven_Cary/article/details/120886696

RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 512, 8, 8] to have 1024 channels, but got 512 channels instead

python 复制代码
https://blog.csdn.net/qq_43533724/article/details/121743384

RuntimeError: result type Float can't be cast to the desired output type long int

python 复制代码
https://blog.csdn.net/abdddwxh/article/details/127509089

本地 GPU 对自制数据集训练完成

6张图片,4个类别,训练了5.28分钟(300轮epochs)

使用训练好的模型进行检测

对这里进行修改

改成权重的相对路径

会对这个目录下的所有图片进行预测,这里先改成训练集的位置

自己打的标签

预测的结果







相关推荐
见行AGV机器人1 小时前
无人机脉动线中的AGV小车
人工智能·无人机·agv·非标定制agv
廋到被风吹走1 小时前
【AI】从 OpenAI Codex 到 GitHub Copilot:AI 编程助手的技术演进脉络
人工智能·github·copilot
newsxun1 小时前
DHA之后,大脑营养进入GPC时代?
人工智能
sg_knight1 小时前
设计模式实战:模板方法模式(Template Method)
python·设计模式·模板方法模式
程序员Better2 小时前
2026年AI大模型选择指南:8大主流模型深度对比,小白秒懂如何选!
人工智能
FreakStudio2 小时前
ESP32居然能当 DNS 服务器用?内含NCSI欺骗和DNS劫持实现
python·单片机·嵌入式·面向对象·并行计算·电子diy
ai_xiaogui2 小时前
AIStarter新版后端原型图详解:架构全面升级+共享环境一键部署,本地AI模型插件工作流管理新时代来临(2026开发者必看)
人工智能·架构·推动开源ai落地·原型图细节·aistarter新版·aistarter新版原型图·架构全面升级+共享环境一键部署
2501_926978332 小时前
“LLM的智能本质--AGI的可能路径--人类的意识本质”三者的统一基底(5.0理论解读)
人工智能·经验分享·笔记·深度学习·机器学习·ai写作·agi
拾光向日葵2 小时前
2026贵州高职专科报考全问答合集:专业、就业与实力大盘点
大数据·人工智能·物联网
لا معنى له2 小时前
WAM与AC-WM:具身智能时代的世界动作模型与动作条件世界模型
人工智能·笔记·学习