Как добавить две разные модели тензорного потока из двух разных приложений?

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')

Нет параметра, метод et c для установки сеанса Tensorflow для обоих приложений. Есть ли способ позволить двум моделям работать в одном сеансе tf?

Answer 1

Я не проверял, как это сделать в TensorFlow 2, но я реализовал это в TF-1 год назад.

Вы можете проверить это python Записная книжка

здесь я объединил две модели Xception в одну, усредняя их выходной сигнал из слоя softmax (с помощью техники ансамбля.)

Для модели Keras (h5) вам необходимо преобразовать эту модель из формата h5 в формат ckpt, чтобы использовать приведенный выше код, используя tf.Saver .