tfjs-yolo-tiny

Hey Everyone! After seeing the announcement of Tensorflow.js, and being a lover of both JS and ML, I couldn't help myself from creating a demo with a non-trivial ML model. YOLO is definitely one of the most interesting/useful ML models out there (as we saw on ModelDepot) and thought that'd be cool to get working. As far as I know, there hasn't been any prior work with getting YOLO object detection working in JS, I assume mainly because the post-processing code to transform the model's outputs to bounding boxes is a whole heap of math. Luckily tf.js makes it a lot easier to translate from Python and so here it is! You can drop this into your project with just a couple lines of code (and no ML required) using NPM! (via npm i tfjs-yolo-tiny) If you want to help with perf/bug fixes, PR's are more than welcome @ https://github.com/ModelDepot/