yolov5.0模型训练

关于yolov5.0训练模型

环境配置

1.pycharm中yolov5配置

首先先去github下载源码文件GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite（这是我从别的博主那里搞的，还行）

然后在终端运行

pip install -r requirements.txt

之后你就会在命令行里面看到你缺少的库函数，跟着下载就可以了

2.程序运行

因为pycahrm的运行是在当前文件下，首先，我们需要打开train.py这个文件里面，然后运行，运行效果如下

可以看到命令行中以image开头的两段，这就说明我们的训练没有问题。

之后再打开detect.py文件并运行，至此，你的模型已经训练完了，在runs的detect中可以看到训练完的效果图

 训练自己的模型

1.文件配置

创建一个文件并分为训练集和数据集

类似于这样，将文件命名为mydata，在这个文件里面分别建立train和image文件，分别村塾训练集和数据集，然后用拆分文件对你的模型进行拆分，具体的大家可以去看这个up的视频

大家可以去找他搞一个拆分文件，应该挺好用的。。。。。。。