yolov5的训练集图片和训练集标签的名字要一一对应

昨天做好训练集之后运行train.py,结果一直提示“can not train without labels”.

尝试了诸如修改训练集路径等方法后依旧无果,于是参考大佬的数据集设置,发现jpg文件的命名跟txt的命名是一一对应的,而我在保存标签时都加上了“train”,导致运行时找不到对应的标签。

然后因为这种低级错误浪费了一天时间。

这里保存一下修改文件名用的代码    代码来源(@liguo_yao)

# -*- coding: utf-8 -*-
import os
#设定文件路径
path='G:\\whorl_nut\\'
i=1
#对目录下的文件进行遍历
for file in os.listdir(path):
#判
  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值