yolov5知识笔记

1、在yolov的同等目录下创建一个自己的数据集文件夹,存放自己的数据集(dataset);
2、在dataset数据文件夹里(coco128)创建(train2017)存放:imasges(存放需要训练的原图片.jpg格式),
                                   labels(存放标注结果的.txt文件)文件夹;
                                 (标注工具makesense.ai或使用labelImg-master(在D盘K210中的标记工具))
3、将对yolov5项目文件中的\data下的.yaml文件;\model下yolov5s.yaml,yolov5-master\tarin.py相关文件进行改动
4、coco128.yaml---coco128 copy.yaml(复制-重命名)
    nc: 80(使用标注的时候用到的类别数)
    names: [  ]#以标注工具取的名字为准(标注的类别名称)labelImg-master--data--predefined_classes.txt类别名称
5、yolov5有许多权重文件想要用哪个权重文件就修改哪个文件中的nc值,nc就是我们标注类别的数量:yolov5s.yaml(最小速度最快);
6、tarin.py的修改:
    parser.add_argument('--weights', type=str, default='yolov5s.pt', help='initial weights path')#default='*'选择我们训练用的权重文件;
    parser.add_argument('--cfg', type=str, default='models/yolov5s.yaml', help='model.yaml path')# default='*'输入我们需要用到的权重文件的配置文件;
    parser.add_argument('--data', type=str, default='data/coco128 copy.yaml', help='data.yaml path')# default='*'填写我们自己修改后的标注配置文件路径;
    parser.add_argument('--batch-size', type=int, default=16, help='total batch size for all GPUs')#default= 显卡不好的可以调低每次训练图片的数量,显卡好的可以尝试更高数
    parser.add_argument('--img-size', nargs='+', type=int, default=[640, 640], help='[train, test] image sizes')
    #, default=[640, 640]也可以填320,320,不过要求得到更好的训练数据可以选择默认的640,640
    参数:
    epochs:指的就是训练过程中整个数据集将被迭代多少次,显卡不行你就调小点。
    batch-size:一次看完多少张图片才进行权重更新,梯度下降的mini-batch,显卡不行你就调小点。
    cfg:存储模型结构的配置文件
    data:存储训练、测试数据的文件
    img-size:输入图片宽高,显卡不行你就调小点。
    rect:进行矩形训练
    resume:恢复最近保存的模型开始训练
    nosave:仅保存最终checkpoint
    notest:仅测试最后的epoch
    evolve:进化超参数
    bucket:gsutil bucket
    cache-images:缓存图像以加快训练速度
    weights:权重文件路径
    name: 重命名results.txt to results_name.txt
    device:cuda device, i.e. 0 or 0,1,2,3 or cpu
    adam:使用adam优化
    multi-scale:多尺度训练,img-size +/- 50%
    single-cls:单类别的训练集


 

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值