嘗試轉換使用tensorflow我構建的自定義模型時遇到以下問題。該模型有兩個輸入,當前是saved_model格式。運行時:
python mo_tf.py --saved_model_dir C:/path_to/model_saved --input_shape [1,128,128,3],[1,1,1,64]
拋出一個錯誤
[ ERROR ] Please provide input layer names for input layer shapes.
我應該如何提供輸入名稱?
uj5u.com熱心網友回復:
在為多個圖層指定輸入形狀時,您必須為將要覆寫其形狀的輸入提供名稱。對于多輸入拓撲,指定輸入如下:
python mo.py --saved_model_dir --input input_name_1,input_name_2 --input_shape [input_shape_1],[input_shape_2]
請參閱轉換 TensorFlow 模型檔案。
轉載請註明出處,本文鏈接:https://www.uj5u.com/shujuku/479256.html
