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

Question

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

标签： deep-learningcomputer-visionartificial-intelligenceobject-detectionyolov5

Answer 1

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

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