yolov5训练过程中的问题

古城微笑少年丶 2022-10-09 03:19 152阅读 0赞

用yolov5发现都是将.xml文件转换为.txt的，网上对直接拥有.txt文件的训练反而教程很少。我自己摸爬打滚。最终运行出来，发现还是一个道理。

**这里：我们只有原图和标注后的.txt文件**

*  数据集

数据集在我的datasets文件夹下：这个文件夹放在yolov5文件夹下

datasets文件夹包括：

*  train   ：这个文件夹下包括images和labels文件夹
 *  valid：这个文件夹下包括images和labels文件夹
 *  test：这个文件夹下包括images和labels文件夹

三者并列的关系

images文件夹存放的是你的原图

labels文件夹下存放的是你在labelimg工具下标注的.txt文件夹

**以上存在的问题是：数据集的划分我是手工操作的，因为我的数据集很少**

*  制作数据集的.yaml文件，这个文件包含

train: /content/drive/My Drive/objectDetection/yolov5/datasets/train/images
    val: /content/drive/My Drive/objectDetection/yolov5/datasets/valid/images
    
    nc: 1
    names: ['Apple_Black_rot']

叫：dieases.yaml

以上工作完成的差不多了，就是训练和测试了。在cloab中

三条命令：

#训练的
    !python train.py  --data data/dieases.yaml --cfg models/yolov5s.yaml --weights weights/yolov5s.pt --batch-size 16
    #测试的
    !python test.py --data data/dieases.yaml --weights runs/train/exp37//weights/best.pt
    
    #推理的
    !python detect.py --weights runs/train/exp37//weights/best.pt --source datasets/dsets/
    
    #detect的图片，我另外创建一个文件夹来存储，图片是我自己胡乱选的
    #

比如我在colab中总是中断，这个时候需要使用：

!python train.py --resume

注意：以上命令需要终端进入yolov5文件夹再运行。

发表评论取消回复

表情：

评论列表（有 0 条评论，152人围观）

还没有评论，来说两句吧...

相关阅读

相关用yolov5训练自己的数据集

用yolov5训练自己的数据集一、配置环境在windows中配置yolov5所需环境，具体操作请看：[https://blog.csdn.net/Wxy9711

素颜马尾好姑娘i/ 2023年10月11日 15:26/ 0 赞/ 9 阅读

相关 YOLOv5训练自己的数据集

一、准备工作在data下新建几个文件夹 ![在这里插入图片描述][20210114092311280.png] XML文件放到Annotations 图片文

墨蓝/ 2023年01月06日 05:45/ 0 赞/ 202 阅读

相关 yolov5训练自己的VOC数据集

一.VOC数据准备 ![20201201204327773.png][] voc数据集目录结构： ----voc

深藏阁楼爱情的钟/ 2022年12月25日 11:59/ 0 赞/ 235 阅读

相关 yolov4&v5训练代码理解

1 简介本文依据[https://github.com/Tianxiaomo/pytorch-YOLOv4][https_github.com_Tianxiaomo_p

待我称王封你为后i/ 2022年12月22日 09:48/ 0 赞/ 722 阅读

相关 YOLOV5训练

简介 `YOLOv4`还没有退热，`YOLOv5`就已经来了！ 6月9日，`Ultralytics`公司开源了`YOLOv5`，离上一次`YOLOv4`发布不到50天，不过

Dear 丶/ 2022年12月13日 04:08/ 0 赞/ 147 阅读

相关 yolov5训练自己的数据集

[https://zhuanlan.zhihu.com/p/156835045?from\_voters\_page=true][https_zhuanlan.zhihu.co

﹏ヽ暗。殇╰゛Y/ 2022年11月12日 13:46/ 0 赞/ 217 阅读

相关 YOLOV4训练自己数据集过程

源码 tensorflow和keras框架： [https://github.com/bubbliiiing/yolov4-keras][https_github.c

素颜马尾好姑娘i/ 2022年11月01日 05:37/ 0 赞/ 272 阅读

相关 YOLOv5训练量化压缩

YOLOv5并量化压缩下载源码和安装依赖库：源码地址：https://github.com/ultralytics/yolov5 下载后解压，在目录内打开cm

逃离我推掉我的手/ 2022年10月15日 12:55/ 0 赞/ 193 阅读

相关 yolov5训练过程中的问题

用yolov5发现都是将.xml文件转换为.txt的，网上对直接拥有.txt文件的训练反而教程很少。我自己摸爬打滚。最终运行出来，发现还是一个道理。这里：我们只有原图和标注

古城微笑少年丶/ 2022年10月09日 03:19/ 0 赞/ 153 阅读

相关 COCO格式转yolov5训练格式

import json with open('/home/lixuan/fromLX/场景识别/新场景样本/休闲细支新主题陈列/场景识别/训练样本初始

桃扇骨/ 2022年10月08日 15:19/ 0 赞/ 155 阅读