Конвертировать pb-файл InceptionV3 в tflite

Question

Я пытаюсь преобразовать модель InceptionV3 из файла pb в файл tflite с помощью TOCO. Я использую следующую команду:

tflite_convert --output_file=/home/luca/Scrivania/prova.tflite --graph_def_file=/home/luca/Scrivania/inception_v3_2016_08_28_frozen.pb/inception_v3_2016_08_28_frozen.pb --input_arrays=input --output_arrays="InceptionV3/Predictions/Reshape_1:0"

Но я получаю следующую ошибку:

ValueError: Invalid tensors 'InceptionV3/Predictions/Reshape_1:0' were found.

Как я могу это исправить?

Answer 1

Эта ошибка означает, что введенный вами output_array неверен. Как правило, выходной массив для моделей InceptionV3 равен InceptionV3/Predictions/Reshape.

Если это не сработает, следующим шагом будет визуализация модели TensorFlow .pb в TensorBoard и поиск выходного массива.