Tensorflow 模型轉(zhuǎn)換 .pb convert to .lite實例

更新時間：2020年02月12日 09:28:14 作者：cefengxu

今天小編就為大家分享一篇Tensorflow 模型轉(zhuǎn)換 .pb convert to .lite實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

2017年底，Tensorflow 推出Lite版本，可實現(xiàn)移動端的快速運行，其中，一個很關(guān)鍵的問題，如何把現(xiàn)有分類模型(.pb) 轉(zhuǎn)換為(.lite)模型呢？

其實，步驟如下

1- 進入 Tensorflow 源碼文件夾（以便bazel可以無需配置找打相應(yīng)路徑）

2- 轉(zhuǎn)換可執(zhí)行文件

bazel run --config=opt \
 //tensorflow/contrib/lite/toco:toco -- \
 --input_file=/tmp/mobilenet_v1_1.0_224_frozen.pb \  # 待轉(zhuǎn)換模型路徑
 --output_file=/tmp/tflite_model2.tflite \      # 目標模型路徑
 --input_format=TENSORFLOW_GRAPHDEF \
 --output_format=TFLITE \
 --input_shape=1,224,224,3 \              # 輸入圖像寬高
 --input_array=input \                 # 輸入節(jié)點名稱<參考1>
 --output_array=MobilenetV1/Predictions/Reshape_1 \  # 輸出節(jié)點名稱<參考1>
 --inference_type=FLOAT \               # 圖像數(shù)據(jù)類型
 --input_data_type=FLOAT

參考1： 如果輸入輸出節(jié)點不知道：進入python，import tf ，并通過如下命令（ tf.GraphDef() ）查找:

>>> import tensorflow as tf
>>> gf = tf.GraphDef()
>>> gf.ParseFromString(open('/your/path/to/graphname.pb','rb').read())
>>> for n in gf.node:
>>>   print ( n.name +' ===> '+n.op )

參轉(zhuǎn)換后，在android手機上運行，在不考慮精度的前提下，不同實現(xiàn)方式得到的結(jié)果如下:

模型	type	model size	speed
mobilenet-v1-224	.pb	17 mb	106 ms
mobilenet-v1-224	.lite	16.4 mb	72 ms

以上這篇Tensorflow 模型轉(zhuǎn)換 .pb convert to .lite實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

終于搞懂了Keras中multiloss的對應(yīng)關(guān)系介紹
這篇文章主要介紹了終于搞懂了Keras中multiloss的對應(yīng)關(guān)系介紹,具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-06-06
一篇文章帶你了解python正則表達式的正確用法
這篇文章主要介紹了Python中正則表達式的詳細教程,正則表達式是Python學習進階當中的重要內(nèi)容,需要的朋友可以參考下
2021-08-08
python?DataFrame數(shù)據(jù)分組統(tǒng)計groupby()函數(shù)的使用
在python的DataFrame中對數(shù)據(jù)進行分組統(tǒng)計主要使用groupby()函數(shù)，本文主要介紹了python?DataFrame數(shù)據(jù)分組統(tǒng)計groupby()函數(shù)的使用，具有一定的參考價值，感興趣的可以了解一下
2022-03-03
Python函數(shù)__new__及__init__作用及區(qū)別解析
這篇文章主要介紹了Python函數(shù)__new__及__init__作用及區(qū)別解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2020-08-08
使用Python的Supervisor進行進程監(jiān)控以及自動啟動
這篇文章主要介紹了使用Python的Supervisor進行進程監(jiān)控以及自動啟動,使用python supervisor實現(xiàn),需要的朋友可以參考下
2014-05-05
Python如何批量更改圖像尺寸統(tǒng)一大小
這篇文章主要介紹了Python如何批量更改圖像尺寸統(tǒng)一大小問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2023-03-03
python獲得兩個數(shù)組交集、并集、差集的方法
這篇文章主要介紹了python獲得兩個數(shù)組交集、并集、差集的方法,實例分析了Python操作list集合的技巧,具有一定參考借鑒價值,需要的朋友可以參考下
2015-03-03
Python語言實現(xiàn)機器學習的K-近鄰算法
今天學習的算法是KNN近鄰算法。KNN算法是一個監(jiān)督學習分類器類別的算法。下面我們來詳細的探討下
2015-06-06
python OpenCV的imread不能讀取中文路徑問題及解決
這篇文章主要介紹了python OpenCV的imread不能讀取中文路徑問題及解決方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2022-07-07
python爬蟲之驗證碼篇3-滑動驗證碼識別技術(shù)
本篇涉及到的驗證碼為滑動驗證碼，不同于極驗證，本驗證碼難度略低，需要的將滑塊拖動到矩形區(qū)域右側(cè)即可完成。對python爬蟲滑動驗證碼識別技術(shù)感興趣的朋友跟隨小編一起看看吧
2019-04-04