來自Tensorflow的圖形文件（.pb）的大小對於android使用來說太大了，如何減少它呢？

Question

我嘗試構建一個使用張量流實現Yolo對象檢測的網絡，並且我希望它可以在Android上使用。在構建結構之後，我使用tf.train.write_graph來獲取圖形文件，並且想要在android demo中替換原始文件。 但是pb文件太大（1.1G），這在Android上不可用。那麼，我怎麼能減小尺寸？

Answer 1

我建議你先嚐試量化你的圖形，因爲你只需要一個官方的TensorFlow腳本。這裏有一個很好的教程由皮特·沃登： https://petewarden.com/2016/05/03/how-to-quantize-neural-networks-with-tensorflow/

如果您使用的32位浮點模型將要結束了〜4倍（〜250MB）小，因爲圖中的值將被轉換爲8個整數理論（推斷它對性能沒有顯着影響）。請注意，這在您壓縮協議緩衝區文件時起作用。