Есть ли эквивалент serving_input_receiver_fn
для обработки выходных данных?Кажется, это не так.Какой лучший обходной путь?У меня есть следующий пример использования:
- Учитывая входное изображение в кодировке base64, сделайте некоторые манипуляции (например, атаку состязанием) для этого изображения, используя предварительно подготовленные модели keras.Затем base64 кодирует это обработанное изображение и возвращает.
Модель Keras может компилировать только без кодировки base64, поскольку стандартные функции потерь не допускают строковые выходные данные.Итак, в следующем коде я не уверен, куда можно вставить выходную кодировку base64.
estimator = keras.estimator.model_to_estimator(
keras_model_path=KERAS_MODEL_PATH,
model_dir=MODEL_DIR)
estimator.export_savedmodel(
EXPORT_PATH,
serving_input_receiver_fn=serving_input_receiver_fn)
Я также прекрасно переписываю свою модель Keras в ванильном тензорном потоке, если это позволяет мне base64 кодировать выводобраз.Я хотел бы знать общий подход к этому.