ValueError: были найдены недопустимые тензоры «normalized_input_image_tensor» - PullRequest
Новая
       55

ValueError: были найдены недопустимые тензоры «normalized_input_image_tensor»

0 голосов
/ 01 марта 2020

ОС: Ubuntu 18.04, модель Tensorflow: ssd_mobilenet_v2_quantized_300x300_coco_2019_01_03

Я переобучил модель ssd_mobilenet_v1_quantized_coco, используя мои собственные данные. Я успешно сгенерировал frozen_inference_graph.pb, используя скрипт " export_inference_graph.py ". Но когда я запустил скрипт, " tflite_convert.py ," ошибка ", ValueError: были найдены недопустимые тензоры normalized_input_image_tensor ' ." вспыхнуло Параметры сценария " tflite_convert.py " равны 

 python tflite_convert.py \
--output_file="converted_quant_traffic_tflite/traffic_tflite_graph.tflite" \
--graph_def_file="traffic_inference_graph_lite/frozen_inference_graph.pb" \
--input_arrays='normalized_input_image_tensor' \
--inference_type=QUANTIZED_UINT8 \
--output_arrays='TFLite_Detection_PostProcess','TFLite_Detection_PostProcess:1','TFLite_Detection_PostProcess:2','TFLite_Detection_PostProcess:3' \
--mean_values=128 \
--std_dev_values=128 \
--input_shapes=1,300,300,3 \
--default_ranges_min=0 \
--default_ranges_max=6 \
--change_concat_input_ranges=false \
--allow_nudging_weights_to_use_fast_gemm_kernel=true \
--allow_custom_ops

Очевидно, input_arrays не был установлен правильно. Посоветуйте, пожалуйста, как установить input_arrays .

...