YOLOV7-pose训练数据遇到的问题。

1问题描述

Traceback (most recent call last):
  File "F:/code/yolov7-pose/train.py", line 562, in <module>
    train(hyp, opt, device, tb_writer)
  File "F:/code/yolov7-pose/train.py", line 204, in train
    image_weights=opt.image_weights, quad=opt.quad, prefix=colorstr('train: '), kpt_label=kpt_label)
  File "F:\code\yolov7-pose\utils\datasets.py", line 74, in create_dataloader
    kpt_label=kpt_label)
  File "F:\code\yolov7-pose\utils\datasets.py", line 414, in __init__
    labels, shapes, self.segments = zip(*cache.values())
ValueError: not enough values to unpack (expected 3, got 0)
train: WARNING: Ignoring corrupted image and/or label F:\code\yolov7-pose\person\images\train\00.jpg: cannot reshape array of size 55 into shape (2)
train: WARNING: Ignoring corrupted image and/or label F:\code\yolov7-pose\person\images\train\01.jpg: cannot reshape array of size

2解决方案:不要使用右键运行,要使用命令行运行.

还需要注意两点:第一点,在进行命令行操作之前,需要把数据集中train和val文件夹里面的labels.cache删除掉。尤其上一次运行没成功,这两个labels.cache文件一定要删除,不然指定运行不通。      第二点:检查数据中的图片和标签是否对应。不对应也会报错。

另外,如果训练的准确率为0,一般都是数据集的问题。比如,有一次,我使用了目标检测的数据集进行了关键点检测代码的训练,出现了准确率为0的现象。

python train.py --data data/coco_kpts_128.yaml --cfg cfg/yolov7-w6-pose.yaml --weights yolov7-w6-person.pt --kpt-label(千万要注意配置文件和路径要正确,这里的yaml文件和路径要注意修改)

评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

神笔馬良

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值