【小白教学】如何用YOLOv7训练自己的数据集

唐小亮 · 发表于 2022-11-25 17:00:27

数据集是公开的哈，网上搜一下可以找得到；train_list.txt这些的生成就是简单的python os模块操作，也不难的哈

天晴了 · 发表于 2022-11-25 17:00:42

求问博主，数据集的存放目录格式必须按照文中提到的方式嘛？（读取数据和txt的代码是在哪个部分呀）

晶品昌田 · 发表于 2022-11-25 17:00:49

如果你不想改代码的话，建议数据集还是按照文中的方式存放哈。读取数据和标签的代码在utils/datasets.py的create_dataloader类里面

地的西瓜 · 发表于 2022-11-25 17:01:31

好的，谢谢博主！

那年风起 · 发表于 2022-11-25 17:02:12

大佬！请问一下训练一个模型大概需要多少样本啊

金字塔 · 发表于 2022-11-25 17:02:25

我这个是5000train，600val

陈强 · 发表于 2022-11-25 17:03:08

好的好的！感谢大佬！[赞同]

小菇凉慢慢长大了 · 发表于 2022-11-25 17:03:21

我的数据集是先按照test/train/vaild分，再分label和img，也可以用

马狗猪 · 发表于 2022-11-25 17:03:36

按照文件名和路径配对检索的，只要文件名对应就可以了，比如image/train/001.jpg的图片，就会自动找label/train.001.txt作为标签

服务无止境 · 发表于 2022-11-25 17:04:28

推理时输入的图片要先改成正方形的32倍数尺寸的图片，比如640*640或者320*320，然后在参数的--img中设定宽度为修改后的宽度，你看看是不是这个问题

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