natanielruiz / android-yolo

Real-time object detection on Android using the YOLO network with TensorFlow
Apache License 2.0
667 stars 212 forks source link

detect multiple objects #5

Closed shootingliu closed 7 years ago

shootingliu commented 7 years ago

Thank you for your nice implementation. I would like to know whether the APP can detect multiple objects at the same time,just like YOLO's demo picture dog.jpg? Thanks~

natanielruiz commented 7 years ago

Hello, This app currently only supports a single object detection (highest confidence). I might work on supporting multiple objects in the future.

If you want to do it yourself I can give you some general pointers. You need to play with the output of the c++ native code (check the YOLO paper to see how they get multiple bounding boxes), do non-maxima suppression and code the appearance of multiple bounding boxes. If you have any more questions just post them here. Cheers