# PPE Detection YOLOv8n(6 Classes)v1 ## 注册信息 | 字段 | 值 | |---|---| | Model ID | `ppe-6classes-yolov8n@1` | | 注册名称 | `PPE Detection YOLOv8n (6 Classes)` | | Backend | `ultralytics-yolo` | | 任务 | 六类 PPE 正向目标检测 | | 权重 | `models/detection/ppe-6classes-yolov8n/v1/best.pt` | | 权重格式 | PyTorch/Ultralytics `.pt` checkpoint | | SHA256 | `07172ef3ae9e256c40a1fb0ce3eefe5547d90170645aa73dded0fffc382cdb31` | Model ID 的 `@1` 与本目录的 `v1` 对应。替换权重前必须重新核对 SHA256、checkpoint 标签和评估结果;不兼容的新权重应注册为新版本,不能覆盖本文件记录的 `v1`。 ## 有序标签 checkpoint 的类别编号必须与下表严格一致,大小写也不能改变: | ID | 标签 | |---:|---| | 0 | `Gloves` | | 1 | `Vest` | | 2 | `goggles` | | 3 | `helmet` | | 4 | `mask` | | 5 | `safety_shoe` | 这六个类别都表示“画面中检测到对应装备”。模型没有 `Person` 或 `No-*` 类,不能把 “没有检测到 helmet”直接解释为“某个人没有佩戴安全帽”。如需人员违规判断,必须增加 人员检测、人员与 PPE 空间关联以及缺失判定逻辑。 ## 输入与运行参数 - 输入必须是已经解码的 packed `BGR8` 或 `RGB8` 图像,buffer 大小为 `width × height × 3`;adapter 会把 `RGB8` 转换为 backend 使用的 BGR 顺序。 - 训练/常用推理尺寸为 `640`;实际推理尺寸由 `model_options.imgsz` 控制。 - 当前项目锁定的 Ultralytics 版本为 `8.4.31`。部署 `.pt` 时还需要与目标设备匹配的 PyTorch 运行时。 - CPU 部署使用 `device: cpu`、`half: false`。CUDA/Jetson 必须使用与驱动或 JetPack 匹配的独立环境,不能只修改 `device`。 常用参数: | 参数 | 说明 | |---|---| | `weights` | 本模型权重路径,必填 | | `device` | `cpu`、`cuda:0` 或非负 GPU 编号 | | `imgsz` | 推理尺寸,默认 `640` | | `iou` | NMS IoU 阈值,默认 `0.70` | | `half` | 是否使用 FP16,CPU 应保持 `false` | | `max_det` | 单帧最大检测框数 | | `agnostic_nms` | 是否启用类别无关 NMS | ## YAML 示例 以下相对路径假定从仓库根目录启动: ```yaml detector: uses: detection.model@1 with: model: ppe-6classes-yolov8n@1 detect_labels: - Gloves - Vest - goggles - helmet - mask - safety_shoe confidence: 0.50 max_fps: 10 attach_frame: false model_options: weights: models/detection/ppe-6classes-yolov8n/v1/best.pt device: cpu imgsz: 640 iou: 0.70 half: false max_det: 100 ``` ## 训练与评估摘要 原模型说明记录了 YOLOv8n 基础模型、50 个 epoch、`imgsz=640`、`batch=32`,以及以下 近似结果: | 指标 | 原说明记录值 | |---|---:| | mAP50 | `~0.81` | | mAP50-95 | `~0.53` | | Precision | `~0.80` | | Recall | `~0.74` | 原说明记录的逐类 mAP50 约为:`Gloves 0.69`、`Vest 0.90`、`goggles 0.90`、 `helmet 0.90`、`mask 0.80`、`safety_shoe 0.64`。这些数据未由 cmvr-edge-ai 在目标园区 数据上独立复现,不能替代部署前的现场验证。 ## 限制与风险 - 低照度、运动模糊、远距离小目标、遮挡和不同地区的 PPE 外观会降低准确率。 - 原数据存在类别不均衡,`goggles` 和 `safety_shoe` 等类别应重点做现场回归测试。 - 模型只检测 PPE 的出现,不进行人员检测、目标跟踪或人员/PPE 归属判断。 - 不应在没有人工复核的情况下用于处罚、法律执法或其他高风险决定。 - 摄像采集和平台上报仍需遵守工作场所隐私、告知和数据保留要求。 ## 许可证与来源 - 原模型卡声明权重许可证为 MIT。 - 模型基于 Ultralytics YOLOv8;Ultralytics 软件及商业使用可能受其许可证约束,部署方 必须自行核对当前适用条款。 - 原模型说明称训练数据为自定义 Roboflow 格式 PPE 数据集,但没有在本目录提供完整的 数据授权链。部署方应在商业或高风险使用前核对数据来源和许可。 - 本目录仅记录当前 `.pt` 权重,不表示仓库包含其他导出格式。 - 权重来源为 Hugging Face 上的 [`Tanishjain9/yolov8n-ppe-detection-6classes`](https://huggingface.co/Tanishjain9/yolov8n-ppe-detection-6classes); 仓库内以本目录路径和上述 SHA256 作为部署制品标识。