yolov5模型训练
书接上文,完成了数据集构建,本文主要对上文数据进行训练
调整数据
在images文件夹中创建两个文件夹分别为train(用于存放训练集)和val(用于存放验证集)并将图片进行分配,原则是train中分配大部分,val分配小部分,我是给train40张,val7张,大家自己看着定义。
同理,在lables文件夹中进行同样的操作,但注意train和val文件夹中的标题要和images文件夹中的train和val文件夹相对应,不然匹配不到,可以理解为每个image对应每个lable
设定参数
更改好bvn.yaml文件的配置
开始运行,性能低的显卡运行可能会比较慢,耐心等待
运行结束后生成上图文件,注意weights中的两个权重文件,待会要用
接着打开detect.py文件,按照以下步骤配置
配置好后,开始运行
bandicam
OK 也不知道我写清楚了没有 到这喽
追加不好意思各位上面的视频传错了,稍后我再重发一份,抱歉
ok 更改完毕