未能导出张量流估计器模型

时间:2018-10-18 12:37:59

标签: python tensorflow tensorflow-estimator

我使用Estimator类训练了张量流模型。现在,我要导出模型以进行预测。输入的是224x224x3图像。通过遵循guide,我尝试了以下解决方案:

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)

但是,在两种情况下我都收到错误消息

  

TypeError:无法将<类型'dict'>类型的对象转换为Tensor。   内容:{'input':< tf.Tensor'ParseExample / ParseExample:0'   shape =(?,)dtype = string>}。考虑将元素投射到受支持的对象   类型。

该模型使用三元组损失的this张量流实现进行了训练。可能是什么问题?

0 个答案:

没有答案