4 ms·
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-trivi
by mikeshi42 9y ago
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/tfjs-yolo-tiny https://github.com/ModelDepot/tfjs-yolo-tiny !