This repository provides ready to use script to convert Posenet Tensorflow JS models to Tensorflow Lite model.
- Modify config.yml
Currently script can build 3 different achitectures of MobileNetv1 models.
Modify 'checkpoints_index' parameter in config.yml to change the architecture.
Similarly modify 'outputStride' and 'image_size' parameters as required.
Each parameter effects the accuracy and speed of detection. To know more visit Posenet-TfJs
- Run
$ python tfjs2tflite.py
- Output
find the model in _models/ folder