YOLO v5 实现目标检测（参考数据集&自制数据集）

redhat9i · 发表于 2023-3-7 12:57

你就按照文章要求，放在yolo同级目录下，不要用绝对路径

FeastSC · 发表于 2023-3-7 13:06

谢谢，作者大大

Zephus · 发表于 2023-3-7 13:07

请问用GPU训练的话该如何修改参数or运行

RedZero9 · 发表于 2023-3-7 13:15

可以直接使用

RedZero9 · 发表于 2023-3-7 13:17

请问那个命令行参数是--img还是--img-size？

BlaXuan · 发表于 2023-3-7 13:24

今天参考你的教程跑了一下代码，看到你的epochs设置为5，有一种思维定势，就全部设置为5。
在训练coco128时，yolov5s训练完5epochs是没有图框的；采用yolov5m训练5epochs可以看到图框；
在训练mask时，yolov5m训练20个epochs都没有图框；最后我在yolov5s上训练了200epochs才看到效果。
如果有人遇到类似的问题希望能帮助一点点。

unityloverz · 发表于 2023-3-7 13:31

因为我当时用5个epoch就出结果了了，所以写blog的时候就没改，[害羞]

yukamu · 发表于 2023-3-7 13:40

感谢大佬[捂脸]这两天在搜yolo v3要怎么用训练数据一直没找到，看到了你这yolo v5的相关介绍，才发现只要直接用coco128就可以开始跑训练了[抱拳了]

mypro334 · 发表于 2023-3-7 13:47

以运行train.py来举例。train.py的上一级就是coco128所在的目录，所以是../coco128，相对的是所运行的文件的位置，而不是相对yaml文件的位置../../coco128哈~

量子计算9 · 发表于 2023-3-7 13:57

大佬，我想问一下训练集的图片顺序和标签顺序需要一致吗？还是只是名字一一对应就行，顺序可以随便。

		自动登录	找回密码
密码			立即注册