训练Yolov5神经网络(本地)

使用coco数据集训练:

        首先运行train.py,自动下载coco128数据集,并开始训练(之后也可以调整位置训练自己的数据集)

制作和训练自己的数据集:

1、首先创建dataset.yaml

指出数据的根目录位置,如下所示:

path: ../datasets/coco128 # 数据集根目录
train: images/train2017 # 训练图片位置
val: images/train2017 # 验证图片位置
test: # 测试图片位置

然后指定训练中有多少个类,列出类的名字,如下所示:

# Classes
nc:2 # 类的数量
names: [ 'person','car'] # 类的名字

 在文件夹data/scripts中创建demo.yaml

path: ../datasets/coco128 # 数据集根目录
train: images/train2017 # 训练图片位置
val: images/train2017 # 验证图片位置
test: # 测试图片位置

# Classes
nc:2 # 类的数量
names: [ 'person','car'] # 类的名字

2、创建标签

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值