我需要检测使用斑马线的行人。我使用YOLO算法实现了,但它检测到所有人,而不仅仅是行人。所以有没有专门用于检测行人的方法或特殊算法?如果没有,我该如何训练我的新模型?
回答:
从YOLO中,您不仅应该获得检测结果,还应该获得类别。您的模型很可能是在COCO数据集上训练的,该数据集有一张特定的对象表,它可以提供这些信息。
您可以在这里找到这样的列表:https://github.com/pjreddie/darknet/blob/master/data/coco.names
您可以看到“person”类在列表中的第0个位置,所以如果您只使用该类的检测结果,您将只得到包含人的边界框!