2017-04-12 31 views
2

我一直在關注「Android上的Tensorflow」的github存儲庫。分析Tensorboard上的張量流圖或.pb文件

  1. 我能夠使用bazel構建代碼,然後將Android項目導入到Android Studio中,如here所述。
  2. 正如你可以看到here,建設APK,使用Android工作室,模型文件後/圖表包含在tensorflow/examples/android/assets
  3. 默認情況下,tensorflow_inception_graph.pbimagenet_comp_graph_label_strings.txt都包括在內,從inception5文件,該文件被下載,而APK是建成。

什麼問題?

  1. 我有一個重新訓練圖(InceptionV3模式,tensorflow/examples/image_retraining/retrain.py提到的),我能夠在Android的目錄中的文件夾資產放置,並生成一個工作APK。
  2. 推理時間,而我用的是默認的圖形或.pb文件是〜500ms的和我的retrained.pb或圖形是〜1400ms(OnePlus3T設備上測試)

請幫助我瞭解

  1. 如何分析默認tensorflow_inception_graph.pb上Tensorboard
+0

是因爲'tensorflow_inception_v3_stripped_optimized_quantized.pb'被剝離,優化和量化嗎?有人可以主張聲明嗎? –

回答

1

去年五月他們已經推出了名爲一個腳本就是這麼做的。

usage: import_pb_to_tensorboard.py [-h] [--model_dir MODEL_DIR] 
            [--log_dir LOG_DIR] 

optional arguments: 
    -h, --help   show this help message and exit 
    --model_dir MODEL_DIR 
         The location of the protobuf ('pb') model to 
         visualize. 
    --log_dir LOG_DIR  The location for the Tensorboard log to begin 
         visualization from. 

注意,目前,在master的版本似乎已經收到比tensorflow的最新1.2.1分佈中所存在更多的愛,所以我會建議使用這一個。

+0

我無法使'import_pb_to_tensorboard.py'工作,所以我打開了這個問題:https://github.com/tensorflow/tensorflow/issues/11519 –