object-detection - 使用 Yolo v4 检测安全帽和背心

Question

我正在训练一个 Yolo v4 物体检测模型，该模型检测建筑工地安全帽和背心的佩戴情况，我想显示这个人是否戴着帽子和背心，这个人的边界框必须显示“戴帽子和背心的人”或“不戴帽子和背心的人”取决于对象边界框的结果：人、帽子、背心。对我该怎么做有什么想法吗？如何找到边界框之间的距离或建立关系？或者至少我如何访问边界框坐标以使用它来编码。检测示例

Answer 1

我是怎么做到的？

1. 获取数据并找出哪种迁移学习模型最有效？----->我选择了 YOLO v4
2. 将提供的数据集转换为 DarkNet 数据集格式，用于在 YOLO v4 模型@Roboflow.com 上进行训练
3. 在我在 Google colab 上的新自定义数据集上训练 YOLO v4 模型（为什么不使用本地系统？因为我需要使用 GPU 来更快地训练模型）
4. 逐帧运行视频并使用模型在面部周围制作边界框并对其进行标记。
5. 以视频格式保存标记的图片。