Warm tip: This article is reproduced from stackoverflow.com, please click
computer-vision machine-learning opencv tensorflow2.0 video-processing

Any special algorithm for pedestrian detection alone?

发布于 2020-03-31 23:00:12

I need to detect the pedestrian who are using zebra crossing.I implemented by using yolo algorithm .But it detects everyone not only the pedestrian .So is there any method or special algorithm for pedestrian alone.If not how can I train my new model?

Questioner
Mythili
Viewed
61
Victor Sonck 2020-01-31 20:32

From YOLO you should not only get the detections but also the classes. Your model was most likely trained on the COCO dataset which has a certain table of objects that it can offer.

You can find such a list here: https://github.com/pjreddie/darknet/blob/master/data/coco.names

You can see that the class "person" is on the 0'th position on the list, so if you only use the detections from that class, you will only get the bounding boxes with persons in them!