deep-learning - 如何在 YOLOv5 中仅在感兴趣的区域进行检测？

Question

我只想检测指定区域中的对象，而忽略 ROI 之外的所有其他检测。

Answer 1

如果我对您的问题的理解正确，您想要检测路面上存在的物体。

一种方法是首先检测路面(可能通过检测车道标记:https://github.com/amusi/awesome-lane-detection)或使用道路自由空间检测模型:https://github.com/fabvio/ld-lsi/ )，然后只将该部分提供给 YOLOv5 以检测物体，或者将完整的图像提供给它，然后根据它们是否存在于路面上过滤检测到的物体(即物体的边界框与道路重叠表面)。如果是，则保留它们，否则忽略它们。