Я пытаюсь экспортировать мою модель классификации изображений TensorFlow так, чтобы она принимала строки base64 в качестве входных данных.
Я пытался реализовать решение, предоставленное для этого вопроса , однако я получаю следующую ошибку:
"InvalidArgumentError: Shape должен иметь ранг0, но это ранг 1 для 'DecodeJpeg_1' (op: 'DecodeJpeg') с входными формами: [?]. "
Ошибка возникает в результате кода в строке 4.
export_dir = '~/models/1'
builder = saved_model_builder.SavedModelBuilder(export_dir)
image = tf.placeholder(dtype=tf.string, shape=[None], name='source')
decoded = tf.image.decode_jpeg(image)
scores = build_model(decoded)
signature = predict_signature_def(inputs={'image_bytes': image},
outputs={'output': scores})
with K.get_session() as sess:
builder.add_meta_graph_and_variables(sess=sess,
tags=[tag_constants.SERVING],
signature_def_map={'predict': signature})
builder.save()
sess.close()
Также,
Я вижу, что в строке 5 «Score» предоставляет выходные данные модели, основанные на функции build_model.Однако я не могу найти ни в ответах исходного вопроса, ни в документации TensorFlow, откуда взялась эта функция.