Создание объекта для изображения с субтитрами im2txt, избегайте загрузки контрольных точек для каждого изображения

Question

Официальный тензор потока Show and Tell-im2txt поставляется с зависимостью от Bazel для создания заголовка для изображения. Для выполнения каждого изображения создается сеанс, контрольные точки загружаются для каждого изображения повторно.

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

Можем ли мы создать функцию, которая принимает изображение в качестве входных данных и пропускает его через предварительно загруженные контрольные точки?

Answer 1

Я наконец-то отделил его работу и создал репо, поддерживающий создание объектов.

Если кто-то работает над достижением того же, можно обратиться к этому репо: https://github.com/Akashtyagi08/Pretrained-im2txt-single_instance