5.16笔记，yolov5模型训练

（1）配置文件修改

1.打开hat.yaml,yolov5_hat.yaml,train.py三个文件
将hat.yaml文件中的图片路径改为自己电脑中的路径，同时确保标签种类（number of classes）为2，标签名称（class names）是标记的两种

2.确保yolov5_hat.yaml中标签种类（number of classes）为2

 3.在train.py文件中确保权重（weight）、模型（model）、数据（data）三者的路径正确

weights：初始化的权重文件的路径地址
cfg：模型yaml文件的路径地址
data：数据yaml文件的路径地址 

（2）模型训练

1.代码中的epochs是指整个模型一共在数据集上训练多少次；batch-size则是指一次训练用几张图片

下面以epochs=50，batch-size=4为例

2.运行train.py文件，开始训练模型