search

ayooshkathuria / YOLO_v3_tutorial_from_scratch

Accompanying code for Paperspace tutorial series "How to Implement YOLO v3 Object Detector from Scratch"
https://blog.paperspace.com/how-to-implement-a-yolo-object-detector-in-pytorch/
2.32k stars 724 forks source link

multi class prediction #41

Open ShangyinGao opened 5 years ago

ShangyinGao commented 5 years ago

Hello, it seems that the current code can not predict multiple classes, like Women and Person at the same time, is that right?

Thanks

Francis-Xia commented 5 years ago

Yes, selecting the max prob one.