如何在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格式。