yolo中创建自己的数据集

创建文件夹
  1. 首先创建(测试集)train文件夹
    在这里插入图片描述

然后在labeLimg(在conda环境下)启动
images是保存的照片,labels是图像标框的数据

  1. 同理在(验证集)Valid也是创建相同的文件夹。
    关于测试集和验证集的说明
  • 训练集:用于学习的一组样例,用于拟合分类器的相关参数。
  • 验证集:用于调整分类器参数的一组样例,例如选择神经网络中隐含层单元数量。
  • 测试集:仅用于评估完整的最终分类器的性能的一组样例。
    通常情况下的占比
    在这里插入图片描述
    一般只要这两个文件夹
    在这里插入图片描述
    train 是训练集
    valid 是验证集 就是会在最后的exp结果展示的。
    data.yaml 文件如下
train: D:/SCI/data1/yaogan/train    #这是存放训练集的路径
val: D:/SCI/data1/yaogan/valid     #这是存放训测试集的路径
# Classes 
nc: 1 						#几个类别
names: ['aircraft']      #类别的名称
模型的选定

一般存放于这个文件中
模型也可以用
在这里插入图片描述
best.pt 是指是当此训练过最好结果的模型
last.pt 是指当此训练最后一个的模型
(在后面这些模型都可以调用)

终端训练代码:

python train.py --batch-size 8 --epochs 100 --data det_sample/data.yaml --weights pretrained/kouzhaobest.pt
# --batch-size 8 每次训练照片8张
#--epochs 100		迭代次数100此 (也就是跑100遍)
#--data det_sample/data.yaml	设定训练集和验证集的设置
#--weights pretrained/kouzhaobest.pt		选用什么路径下的模型

终端测试代码:

 python .\detect.py --source D:/SCI/data1/1 --weights runs/train/exp12/weights/best.pt
 #--source D:/SCI/data1/1   测试图片集存放的路径
 #--weights runs/train/exp12/weights/best.pt  刚刚训练完就好的模型
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值