Не удалось экспортировать модель оценки тензорного потока - PullRequest
Новая
       7

Не удалось экспортировать модель оценки тензорного потока

0 голосов
/ 18 октября 2018

Я обучил модель тензорного потока с помощью класса Estimator.Теперь я хочу экспортировать модель для прогнозирования.На входе 224x224x3 изображения.Следуя руководству , я попробовал следующие решения: 

feature_spec = {'input': tf.FixedLenFeature([], dtype=tf.string)}
input_fn = tf.estimator.export.build_parsing_serving_input_receiver_fn(feature_spec)
exported_model_path = estimator.export_savedmodel(args.model_dir, input_fn)

features = {'input': tf.placeholder(tf.float32, [224, 224, 3], name="input")}
input_fn = tf.estimator.export.build_raw_serving_input_receiver_fn(features, 1)
exported_model_path = estimator.export_savedmodel(args.model_dir, input_fn)

Однако в обоих случаях я получаю ошибку

Ошибка типа: Ошибка преобразования объекта типа< введите 'dict'> для Тензор.Содержание: {'input': < tf.Tensor 'ParseExample / ParseExample: 0' shape = (?,) dtype = string>}.Рассмотрим приведение элементов к поддерживаемому типу.

Модель была обучена с использованием этой реализации тензорного потока с триплетными потерями.В чем может быть проблема?

...