J'ai un problème avec la conversion du modèle tensorflow en tflite. J'ai un modèle appris basé sur la Détection d'objets Tensorflow Je souhaite utiliser le code de conversion du convertisseur TFlite
curl https://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_0.50_128_frozen.tgz | tar xzv -C /tmp tflite_convert \ --output_file=/tmp/foo.tflite \ --graph_def_file=/tmp/mobilenet_v1_0.50_128/frozen_graph.pb \ --input_arrays=input \ --output_arrays=MobilenetV1/Predictions/Reshape_1
3 Réponses :
Si vous téléchargez le modèle sur ici a> puis il y aura un fichier mobilenet_v1_0.5_128_info.txt qui indique le nom d'entrée et de sortie Model: mobilenet_v1_0.5_128
Input: input
Output: MobilenetV1/Predictions/Reshape_1
Le moyen le plus simple que j'ai trouvé est de télécharger mon modèle sur https://lutzroeder.github.io/ netron / il montre les entrées et les sorties de n'importe quel modèle.
ps il a également une application de bureau que j'utilise, au lieu de télécharger tous les modèles < / p>