2017-08-17 50 views
1

オブジェクト検出APIでssd_mobilenetsを使用して自分のモデルをトレーニングし、.ckptファイルを取得します。それは私のコンピュータ上ではうまくいきますが、今では私の電話でモデルを使いたいと思っています。ですから、それを.pbファイルに変換する必要があります。私はそれをする方法を知らない、誰も助けることができますか?ちなみに、ssd_mobilenetsのグラフは複雑ですが、私はモデルの出力を見つけることができません。出力の名前を知っている人はいますか?.ckptファイルを.pbに変換する方法

答えて

0
  • これは、このリンクでobject_detection_tutorial.ipynbで4番目のコードセルである - https://github.com/tensorflow/models/blob/master/research/object_detection/object_detection_tutorial.ipynb

    # What model to download. 
    MODEL_NAME = 'ssd_mobilenet_v1_coco_2017_11_17' 
    MODEL_FILE = MODEL_NAME + '.tar.gz' 
    DOWNLOAD_BASE = 'http://download.tensorflow.org/models/object_detection/' 
    
    # Path to frozen detection graph. This is the actual model that is used for the object detection. 
    PATH_TO_CKPT = MODEL_NAME + '/frozen_inference_graph.pb' 
    
    # List of the strings that is used to add correct label for each box. 
    PATH_TO_LABELS = os.path.join('data', 'mscoco_label_map.pbtxt') 
    
    NUM_CLASSES = 90 
    
  • 今電池は明らかにそう/frozen_inference_graph.pb

  • ある.pbファイル名は、あなたが既に持っていると言います.pbファイルなぜ変換しますか?
  • はとにかくあなたは、グラフの凍結のためthsiリンクを参照することができます:あなたが変換するtensorflow.python.tools.freeze_graph()機能を使用する必要がhttps://github.com/jayshah19949596/Tensorboard-Visualization-Freezing-Graph
  • をご.ckptファイル以下のコード行は、あなたがそれを行う方法を示し
  • ファイル.pb

    freeze_graph.freeze_graph(input_graph_path, 
              input_saver_def_path, 
              input_binary, 
              input_checkpoint_path, 
              output_node_names, 
              restore_op_name, 
              filename_tensor_name, 
              output_graph_path, 
              clear_devices, 
              initializer_nodes) 
    
    • input_graph_path:.pbのファイルへのパスで、グラフを作成します。この.pbファイルはフリーズしていません。あなたは、グラフ
    • input_saver_def_pathを書くためにtf.train.write_graph()を使用する:あなたは空の文字列
    • input_binaryそれを維持することができます:それはgenertaedファイルがバイナリではありませんように偽のそれを維持するブール値であり、人間が読める
    • input_checkpoint_path:パス.ckpt file
    • output_graph_pathに:あなたはあなたのpbファイル
    • clear_devicesを書きたいパス:ブール値...偽
    • output_node_namesをそれを保つ:あなたはSAVしたい明示的なテンソルノード名電子
    • restore_op_name:
    • initializer_nodes = ""
: "保存/ restore_all"
  • filename_tensor_name = "0 /のConstを保存" でなければなりません文字列値
  • 関連する問題