我们需要关注两个文件：coco.yaml和coco128.yaml文件，这两个文件是作者提供的数据集文件，我们在训练自己的数据集的时候，需要借用他的文件修改相应的参数。

在训练时，需要在yolov5/data/文件夹中为自己的数据集创建一个data.yaml文件,指明数据集图片和标签的存放地址，以及类别的数目和名称。