tensorflow - 如何从两个不同的应用程序中添加两个不同的 tensorflow 模型？

Question

我使用来自两个不同应用程序的两个模型。第一个是EAST Scene Text Detector，第二个是Calamari Ocr。他们都使用 TensorFlow。代码如下：

east_detector = EASTDetector(model_path='east_models/EAST_IC15+13_model.h5')
calamari = ocr.Predictor(checkpoint='calamari_models/model_00623553.ckpt')

没有为这两个应用程序设置 Tensorflow 会话的参数、方法等。有没有办法让两个模型在同一个 tf 会话中运行？

标签： tensorflowocr

Answer 1

我还没有检查过如何在 TensorFlow 2 中做到这一点，但我已经在 TF-1 中实现了很多年。

你可以查看这个python Notebook

在这里，我通过平均 softmax 层的输出将两个 Xception 模型合并为一个（有点像集成技术。）

对于 Keras 模型（h5），您需要将该模型从 h5 转换为 ckpt 甲酸盐，以使用上述代码使用tf.Saver。