YOLO
文章平均质量分 52
YOLO实战 数据标准 模型训练 报错
阿凡达蘑菇灯
好好修炼
展开
-
YOLO中 数据集格式转换 --------xml转为 txt格式
其中txt中数据格式为(cls, x,y,w,h), cls为类别, (x,y)为矩形框中心坐标, w,h分别为 矩形框的宽、高。都是归一化后的值,都在0-1之间。(2)遍历object(矩形框),获取name(类别),获取xmin,xmax,ymin,ymax。(3)通过convert函数,将数据转为 txt的格式。思路:(1)从xml size中读取图片尺寸---w,h。(4)保存数据到txt中。原创 2024-09-11 20:03:30 · 237 阅读 · 0 评论 -
YOLOV5 v 6.1 版本报错
YOLOV5训练时遇到TuntimeError错误,问题出现在损失函数计算过程中。错误表明浮点数不能转换为长整型。解决方案是修改utils/loss.py文件中build_targets函数的两处代码,将特定索引的计算更改为考虑形状信息,避免类型不匹配。修复后的代码使得索引计算正确适应图像和网格的尺寸。原创 2024-09-10 01:01:43 · 358 阅读 · 1 评论 -
YOLO 自动标注工具autolabelimg(适用yolov5 v6.1)
首先,手动标注一部分数据集,训练YOLOV5 模型,得到best.pt。二、下载和安装autolabelimg(1) github链接:(2)安装。原创 2024-09-10 13:49:54 · 412 阅读 · 0 评论 -
YOLOV5 ‘FreeTypeFont‘ object has no attribute ‘getsize‘报错解决
将utils 文件下的 plots.py 文件做如下修改。原创 2024-09-10 01:11:52 · 230 阅读 · 0 评论 -
YOLOV5 构建自己的数据集,模型训练(图像识别)
(4)在yolov5/data文件中, 找到coco128.yaml ,复制一份,重命名data.yaml,这里面可以指定自定义数据集路径。注意: 标注了一部分数据集,关闭软件后,随后又打开labelimg, 之前的classes.txt 会消失。(3)从github/gitee上下载 yolov5安装包, 注意有不同的版本,老版本会有各种问题,建议下载较新的版本。2. 在 大约114行初,添加以下代码,把之前的classes.txt的路径(绝对路径),写到代码中。(7) 训练自己的数据集。原创 2024-09-10 13:07:18 · 404 阅读 · 0 评论