search

ModelDepot / tfjs-yolo-tiny

In-Browser Object Detection using Tiny YOLO on Tensorflow.js
https://modeldepot.io/mikeshi/tiny-yolo-in-javascript
MIT License
531 stars 92 forks source link

help ? #5

Closed hiddentn closed 6 years ago

hiddentn commented 6 years ago

can you explain to me where are the weights please ? ,thanks

&& btw look at these numbers :) screenshot 1 GTX850M

MikeShi42 commented 6 years ago

That's awesome! 5fps! :D

For a zip of the weight files used, you can check it out on ModelDepot, clicking on "Download Model" https://modeldepot.io/mikeshi/tiny-yolo-in-javascript

The location reference in this package is here in the source code. You can check out the repo hosting the weight files here (it's a bit of a mess, the config file is model2.json)

Ideally I can get it hosted somewhere better than GH but we don't have the data transfer capacity at ModelDepot to justify hosting the live model version there for now (hopefully in the future we can though!).

The weights were created through pulling the weights from the original darknet website, passing it through YAD2k (see this issue for loading tiny yolo weights), and then using tfjs's keras -> tfjs convertor to create the ultimate tf.js model weights.

Let me know if you have any further questions :)