AMI-system / species_classifier

This repository contains the code to create on-device machine learning models for species classification.
MIT License
2 stars 0 forks source link

Compress the trained species classifier model using tensorflow lite #10

Closed KatrionaGoldmann closed 1 year ago

KatrionaGoldmann commented 1 year ago

Convert pytorch models to tensorflow lite following: https://github.com/omerferhatt/torch2tflite

KatrionaGoldmann commented 1 year ago

tflite-runtime requires python ≥ 3.7, whereas baskerville only has 3.6.8 by default.

So to run:

KatrionaGoldmann commented 1 year ago

Conversion steps:

Evaluation:

KatrionaGoldmann commented 1 year ago

See 10_tflite_models branch