簡體   English   中英

替代tensorflowjs_converter中的tf_frozen_model

[英]alternative to tf_frozen_model in tensorflowjs_converter

我是tensorflow的新手,並通過pip安裝了當前的tensorflowjs(1.1.2),因為我想將一個凍結的推理圖( .pb )轉換為tensorflowjs文件( --output_format=tensorflowjs )。 我按照教程建議--input_format=tf_frozen_model ,這個版本的tensorflowjs不再支持這樣的選項並返回錯誤消息error:argument --input_format: invalid choice: 'tf_frozen_model' 此選項是否已完全替換,或僅重命名? 我試圖安裝版本tensorflowjs v.0.8.0 ,這導致了一堆ImportErrors(用於Keras)和依賴沖突。 我在這里想念一下嗎?

  1. 創建一個新的虛擬環境並使用命令安裝tensorflowjs 0.8.6:pip install tensorflowjs == 0.8.6
  2. 可以使用以下方法轉換更多模型:tensorflowjs_converter --input_format = tf_frozen_model --output_json = true --output_node_names ='MobilenetV1 / Predictions / Reshape_1'--saved_model_tags = serve \\ /mobilenet/frozen_model.pb / mobilenet / web_model
  3. 參考: https//github.com/tensorflow/tfjs-converter

tensorflowjs包的現代版本不再從frozen models轉換。 來自https://github.com/tensorflow/tfjs-converter

注意:如果要轉換TensorFlow凍結模型或會話包,可以安裝舊版本的tensorflowjs pip包,即pip install tensorflowjs == 0.8.6

希望能幫助到你。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM