如何在AWS DJL中加载经过tensorflow训练的自定义模型

Question

我正在尝试Aws Djl平台。我想加载自定义训练的张量流模型并执行推理。我在执行此操作的官方github中找不到直接的示例。有人可以指导我吗？

Answer 1

这是演示项目：https://github.com/aws-samples/djl-demo/tree/master/pneumonia-detection

这是有关加载tensorflow模型的文档：https://github.com/awslabs/djl/blob/master/docs/tensorflow/how_to_import_keras_models_in_DJL.md 和： https://github.com/awslabs/djl/blob/master/docs/load_model.md