Un ensemble d'outils d'apprentissage automatique pour smartphones et appareils embarqués! Plus léger que Tensorflow.
On suppose que Python et Tensorflow (pip) sont installés sur votre PC. Tout d'abord, préparez le modèle à convertir.
import tensorflow as tf
converter = tf.lite.TFLiteConverter.from_saved_model("test_model.pb")
tflite_model = converter.convert()
open("test_model.tflite", "wb").write(tflite_model)
import tensorflow as tf
converter = tf.lite.TFLiteConverter.from_saved_model("Chemin relatif vers le modèle")
tflite_model = converter.convert()
open("Où enregistrer le modèle", "wb").write(tflite_model)
Vous pouvez également convertir sur la ligne de commande.
tflite_convert \
--saved_model_dir=test_model.pb \
--output_file=test_model.tflite
tflite_convert \
--saved_model_dir=Chemin relatif vers le modèle\
--output_file=Où enregistrer le modèle
Je vais vous expliquer comment intégrer Tensorflow Lite dans Android, alors n'hésitez pas à nous contacter. L'article est ici
Veuillez également vous référer aux informations officielles. Article officiel
Recommended Posts