Ajouter des poids au fichier .pb exporté par TensorFlow

Mon projet utilise Python pour former un MLP sur TensorFlow puis j'exporte le graphique et les poids de cette manière:

tf.train.write_graph(sess.graph_def, "./", "inp.txt", True) saver.save(sess, 'variables/model.ckpt', global_step=1) 

Maintenant, bien qu'il soit bon d'utiliser les deux fichiers pour l'importer à Python, il semble impossible de l'utiliser pour Android ou C ++ car il ne peut pas importer le point de contrôle.

  • Comment obtenir Bitmap à partir de la ressource par défaut d'Android?
  • Comment libérer un plugin d'application utilisant Android Market?
  • Appelez setCompoundDrawables () n'indique pas le composite Drawable
  • Comment définit-je les RecyclerViews imbriqués, tout en restant performant?
  • Android Studio: java.io.IOException: une connexion existante a été fermée de force par l'hôte distant
  • Convertir AsyncTask en RxAndroid
  • En ce moment, j'utilise le script freeze_graph.py fourni par google pour rejoindre les deux fichiers en un en faisant:

     bazel-bin/tensorflow/python/tools/freeze_graph --input_graph=inp.txt --input_checkpoint=variables/model.ckpt-1 --output_graph=newoutput.pb --output_node_names=output 

    Ma question est: est-ce qu'il existe un moyen d'utiliser une autre fonction au lieu de tf.train.write_graph pour l'exporter avec les poids inclus?

  • Erreur: Cause: impossible de trouver un chemin de certification valide vers la cible demandée
  • Créer un fichier GPX à partir de GeoPoints
  • Quelle est la différence entre un fil et un gestionnaire
  • Liste de vidéos en streaming utilisant Exoplayer
  • Libcurl Certificate Pinning fonctionne sur iPhone mais pas sur Android
  • Activité de l'assistant d'activité en plein écran. Comment puis-je empêcher la barre d'action de s'afficher lorsque je interagis avec un périphérique?
  • 3 Solutions collect form web for “Ajouter des poids au fichier .pb exporté par TensorFlow”

    Pour l'instant, je suis désolé de dire qu'il n'y a aucun moyen.

    Comme discuté précédemment sur Github ( regardez ), l'équipe TensorFlow ne résout pas actuellement ce problème.

    Je ne peux rien dire pour le moment.

    pltrdy

    Actuellement, il n'y a aucun moyen de le faire, malheureusement.

    Il serait vraiment bon d'avoir un moyen de le faire directement. Je veux dire, il serait formidable d'avoir quelque chose pour le faire d'une seule manière, au lieu d'avoir à générer deux fichiers, puis d'exécuter un autre script pour les convertir.

    C'est particulièrement mauvais pour ceux qui utilisent des graphiques hybrides sur un ordinateur portable.

    De toute façon, comme le groupe TensorFlow l'a déclaré:

    Nous n'avons pas l'intention de soutenir la production de fichiers .pb directement. Si vous êtes préoccupé par le fait que trop de fichiers de points de contrôle occupent de l'espace, vous pouvez limiter le max_to_keep à 1.

    Actuellement, freezegraph est le seul moyen de résoudre ce problème.

    Pour moi, cela ne fonctionne pas très bien parce que je dois l'installer sur de nombreux ordinateurs tous les jours car les gens continuent à faire des pannes et le freezegraph me force à l'installer à partir de la source.

    Ils doivent créer une autre façon de le faire. Une manière intégrée. En particulier s'ils veulent que nous utilisions pour Android.

    coAndroid est un fan Android de Google, tout sur les téléphones Android, Android Wear, Android Dev et Android Games Apps.