如何在tensorflow2中替换占位符？

Question

我需要将估计器导出器代码从tensorflow 1.x转换为2，并去掉占位符。有没有人知道怎么做才对？

我已经尝试了features = {'input'：tf.compat.v1.placeholder(tf.float32，shape=(1,224,224，3)，name="input")}，但这并不能像我预期的那样工作

features = {'input': tf.placeholder(tf.float32, shape=(1, 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)

Answer 1

如果您已经迁移到使用keras模型，您可以使用以下代码

# model is a compiled and trained keras model
tf.saved_model.save(model, export_dir)

它会将keras模型导出为SavedModel格式。