下記のようにpythonでtensorflowの出力の結果について質問させてください。
python
1#tflite読み込み(tfliteファイルは移動してます) 2interpreter = tf.lite.Interpreter(model_path="*****.tflite") 3interpreter.allocate_tensors() 4 5#input output tensor取得 6input_details = interpreter.get_input_details() 7output_details = interpreter.get_output_details() 8 9# 入出力フォーマットを確認 10print('入出力フォーマットを確認') 11print(input_details) 12print(output_details)
console
1[{'name': 'input', 'index': 0, 'shape': array([ 1, 128, 128, 3]), 'dtype': <class 'numpy.float32'>, 'quantization': (0.0, 0)}] 2[{'name': 'regressors', 'index': 175, 'shape': array([ 1, 896, 16]), 'dtype': <class 'numpy.float32'>, 'quantization': (0.0, 0)}, {'name': 'classificators', 'index': 174, 'shape': array([ 1, 896, 1]), 'dtype': <class 'numpy.float32'>, 'quantization': (0.0, 0)}] 3 4ValueError: Cannot set tensor: Dimension mismatch. Got 256 but expected 128 for dimension 1 of input 0.
input_detailsの特にindexの意味が分かりません。
いまいちイメージがつかめないので質問させていただきました。
あなたの回答
tips
プレビュー