当前位置:   article > 正文

【yolov5目标检测】使用yolov5训练自己的训练集_yolov5里面.pt

yolov5里面.pt

数据集准备

首先得准备好数据集,你的数据集至少包含images和labels,严格来说你的images应该包含训练集train、验证集val和测试集test,不过为了简单说明使用步骤,其中test可以不要,val和train可以用同一个,因此我这里只用了一个images

其中images装的是图片数据,labels装的是与图片一一对应同名的yolo格式txt,即类别号,经过归一化的中心x和y坐标以及宽和高

 

下载yolov5

到GitHub上下载整个项目的压缩包zip下来

 下来解压zip,把我们刚刚的数据集也放进去

再下载一个yolov5的预训练模型,我这里选择yolov5n.pt,下下来也放到解压文件夹中

然后用pycharm打开这个文件夹,看看哪里标红还差什么软件包没安装给安装上

配置yaml

先配置一下yolov5预训练模型的yaml,我下载的是yolov5n.pt模型,因此需要配置一下yolov5n.yaml,修改nc的数值为类别数目,我这里的数据集只有乌骨鸡和狮头鹅,因此改成2

  1. # YOLOv5
    声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/452552
    推荐阅读
    相关标签