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

Question

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

Answer 1

我是怎么做到的？

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