ValueError: При преобразовании из .pb в .tflite были обнаружены недопустимые выходные данные тензоров

Question

Я успешно переобучил квантованную модель мобильной сети (architecture="mobilenet_1.0_128_quantized") со своим собственным набором данных изображений:

python3 -m scripts.retrain \
  --bottleneck_dir=tf_files/bottlenecks_quant \
  --how_many_training_steps=50000 \
  --model_dir=tf_files/models/ \
  --summaries_dir=tf_files/training_summaries/"mobilenet_1.0_128_quant" \
  --output_graph=tf_files/retrained_graph_50000_1.0_128.pb \
  --output_labels=tf_files/retrained_labels.txt \
  --architecture="mobilenet_1.0_128_quantized" \
  --image_dir=images

Когда я пытаюсь преобразовать файл .pb в .tflite, используя

toco \
  --graph_def_file=tf_files/retrained_graph_50000_1.0_128.pb \
  --output_file=tf_files/retrained_graph_50000_1.0_128.tflite \
  --input_format=TENSORFLOW_GRAPHDEF --output_format=TFLITE \
  --inference_type=QUANTIZED_UINT8 \
  --input_shape="1,128,128,3" \
  --input_array=input \
  --output_array=outputs \
  --std_dev_values=127.5 --mean_value=127.5

Сбой при следующей ошибке:

ValueError: Обнаружены недопустимые выходные данные тензоров.