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

Question

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

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

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

Answer 1

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

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