我使用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张量流实现进行了训练。可能是什么问题?