YOLO26矿井安全帽检测数据集 矿井安全帽数据集 yolo矿井识别

数据集包括3个类carry objects携带工具、helmet戴头盔、no-helmet没带头盔按8:1:1比例划分为train5110张val639张test639张yolo系列可以直接拿来进行训练本数据集面对的是复杂场景的安全帽图片数据集中包含有煤矿场景因图片目标较为单一为扩充数据集对其他安全帽数据集的图片进行了复杂场景的处理同时重新使用新的类标注图片。如何使用 YOLOv8 训练矿井安全帽检测数据集的详细步骤。YOLOv8 是 YOLO 系列的最新版本具有更高的精度和更快的推理速度。一、数据准备数据集结构假设你的数据集已经按照 8:1:1 的比例划分好并且结构如下深色版本mine_safety_helmet_dataset/├── images/│ ├── train/│ │ ├── 000001.jpg│ │ ├── 000002.jpg│ │ └── …│ ├── val/│ │ ├── 000001.jpg│ │ ├── 000002.jpg│ │ └── …│ └── test/│ ├── 000001.jpg│ ├── 000002.jpg│ └── …├── labels/│ ├── train/│ │ ├── 000001.txt│ │ ├── 000002.txt│ │ └── …│ ├── val/│ │ ├── 000001.txt│ │ ├── 000002.txt│ │ └── …│ └── test/│ ├── 000001.txt│ ├── 000002.txt│ └── …└── mine_safety_helmet.yamlimages/ 目录包含输入图像。labels/ 目录包含对应的标注文件每个标注文件中的每一行表示一个目标框格式为 class_id x_center y_center width height。mine_safety_helmet.yaml 是数据集的配置文件。2. 配置文件创建一个 mine_safety_helmet.yaml 文件内容如下yaml深色版本数据集路径path: ./mine_safety_helmet_dataset类别名称names:0: carry_objects1: helmet2: no_helmet数据集划分train: images/trainval: images/valtest: images/test二、安装依赖确保你已经安装了 YOLOv8 所需的依赖。可以使用以下命令安装bash深色版本pip install ultralytics三、训练模型导入库python深色版本import torchfrom ultralytics import YOLO加载模型python深色版本加载预训练的YOLOv8模型model YOLO(‘yolov8n.yaml’) # 你可以选择其他预训练模型如’yolov8s.yaml’、yolov8m.yaml’等3. 配置训练参数python深色版本配置训练参数model.train(data‘mine_safety_helmet.yaml’, # 数据集配置文件epochs100, # 训练轮数imgsz640, # 输入图像大小batch16, # 批量大小name‘mine_safety_helmet’, # 训练结果保存的文件夹名称device‘cuda’ # 使用GPU进行训练)四、评估模型训练完成后可以使用验证集来评估模型的性能。python深色版本评估模型results model.val(data‘mine_safety_helmet.yaml’, imgsz640, batch16, device‘cuda’)打印评估结果print(results)五、保存模型训练完成后模型会自动保存到指定的文件夹中。你也可以手动保存模型python深色版本保存模型model.save(‘mine_safety_helmet_model.pt’)六、完整代码将上述所有步骤整合成一个完整的脚本python深色版本import torchfrom ultralytics import YOLO加载预训练的YOLOv8模型model YOLO(‘yolov8n.yaml’) # 你可以选择其他预训练模型如’yolov8s.yaml’、yolov8m.yaml’等配置训练参数model.train(data‘mine_safety_helmet.yaml’, # 数据集配置文件epochs100, # 训练轮数imgsz640, # 输入图像大小batch16, # 批量大小name‘mine_safety_helmet’, # 训练结果保存的文件夹名称device‘cuda’ # 使用GPU进行训练)评估模型results model.val(data‘mine_safety_helmet.yaml’, imgsz640, batch16, device‘cuda’)打印评估结果print(results)保存模型model.save(‘mine_safety_helmet_model.pt’)七、运行脚本将上述代码保存为一个 Python 脚本文件例如 train_mine_safety_helmet.py然后在命令行中运行b