FYI: Sucessfully converted NanoDet to TensorFlow with eval implemetation in JavaScript
See original GitHub issueIn case anybody is interested, NanoDet works like a charm in TensorFlow and TensorFlow/JS
Converted models, full JS code as well as conversion notes are at:
https://github.com/vladmandic/nanodet
I’ve re-implemented score & box decoding to squeeze a bit more performance
and moved NMS as a post-decoding task to avoid calculation of low-probability boxes
(code is fully commented with explanations)
Implementation is now directly applicable to both nanodet-m
and nanodet-g
and it auto-adjusts
(nanodet-t
converted model has some corrupted layers and cannot be used just yet)
Model works in NodeJS using TensorFlow
backend and Browser using WebGL
backend, but does not work using WASM
backend due to missing implementation for an operation SparseToDense
(https://github.com/tensorflow/tfjs/issues/4824)
Thank you for a great TINY model!
Issue Analytics
- State:
- Created 3 years ago
- Reactions:7
- Comments:19
Top GitHub Comments
Hey I tried converting to tensorflow and tflite but the model seems to perform poor I tried the inference provided here: https://github.com/PINTO0309/PINTO_model_zoo/blob/main/072_NanoDet/demo/demo_tflite.py
Works well, thank you!