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

Я успешно переобучил мобильную квантованную модель (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: Найдены недопустимые выходы тензоров.

Источник

0 ответов

Другие вопросы по тегам