win10 安装yolov7 训练自己的数据集

1.安装

https://www.pudn.com/news/62ddf431864d5c73acfb1a09.html

  1. 下载yolov7源码 yolov7
  2. 修改requirements.txt
    修改torch以及torchvision版本,这里直接指定好版本让其版本对应
    在这里插入图片描述
  3. 安装
pip install -r requirements.txt -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

2.标注自己的数据

  1. 安装labelImg
pip install labelImg
labelImg

在这里插入图片描述
一定要标定后选择yolo格式保存,然后就会生成对应的txt文件
在这里插入图片描述

3.修改配置文件

  1. 首先文件copy到源码里 图片放到images里 txt 放在labels里面 ,格式如下。
    在这里插入图片描述
    .cache文件是运行train.py文件生成的缓存文件,为了训练的时候数据加速读取。
    2.修改 coco.yaml
    在这里插入图片描述
    3 .修改train.py文件
    在这里插入图片描述
    第一个默认应该是yolov7.pt 我这里是因为我训练我上次的
    就可以训练了。

在这里插入图片描述

4. 错误

  1. caffe2_detectron_ops_gpu.dll" or one of its dependencies.
    因为 batch_size 太大了 ,我改成1就没问题了。

https://blog.csdn.net/qq_43117155/article/details/125672691

  1. 载入数据集的时候报could not convert string to float错及解决方法
    这个问题是因为我一开始标注使用的labelme ,然后转换为txt的时候问题,就是这个文章的labelme 转yolov5的脚本 ,解决方式就是从新把数据用labelimg 标注 ,因为我只有几个图片,如果很多图片 ,我建议把txt数据读取出来,用utf-8格式写入txt中 ,这个是数据编码问题
  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值