yolov5训练自己的目标检测模型

yolov5训练自己的目标检测模型

yolov5

yolo是当前最火的目标检测模型之一，效果很好速度很快，也很适合小白上手。yolov5是yolo的第五代模型。

项目地址：https://github.com/ultralytics/yolov5

克隆下来之后，用pycharm打开，选择3.8版本的python解释器，自动安装依赖。

制作数据集

数据集在线打标网站：https://www.makesense.ai/

打标完成后可以下载符合yolo格式的txt文件。

然后需要对数据集进行组织。

我的数据集目录如下：

screwdata
├── images
│   ├── test
│   │   ├── 若干测试图片.jpg
│   └── train
│       ├── 若干训练图片.jpg
└── labels
    ├── test
    │   ├── 若干测试图片的标签.txt
    ├── train
    │   ├── 若干训练图片的标签.txt

注意：images目录下的图片文件名需和labels目录下的文件名保持一致（后缀名不同）

其中测试/训练图片的标签文件为文本文件，其格式如下：

0 0.991121 0.932710 0.017757 0.059813