search

lshug / first-order-model-tf

TensorFlow port of first-order motion model. TF Lite and TF.js compatible, supports the original's checkpoints and implements in-graph kp processing, but inference only (no training).
MIT License
34 stars 9 forks source link

I can't find testlite.py #24

Closed Apokar closed 5 months ago

Apokar commented 5 months ago

@lshug Hi, has the name of this file been changed? Can I still use it to check if the tflite model works correctly?

lshug commented 5 months ago

Hello. As I remember I replaced testlite.py with TFLite signature runner-based functionality in run.py as soon as TFLite signature runner became available in Tensorflow. TFLite models can be tested after build by running run.py with a not-so-awesomely-named --target tflite argument. When this argument is provided, run.py uses load_models_tflite from utils.py to load the models, which returns callables with the same signature as SavedModels and which are then used in animate.py the same way.