Object 创建 Image Captioning im2txt 模型,避免为每张图片加载检查点

Object creation for Image Captioning im2txt model, avoid loading checkpoints for every image

官方 tensorflow Show 和 Tell-im2txt 模型带有 bazel 构建依赖项来为图像生成标题。对于每张图片的执行,都会创建一个 session,并为每张图片重复加载检查点。

如何避免重复加载检查点?

我们能否创建一个将图像作为输入并通过 pre-loaded 检查点运行的函数?

我终于解耦了它的工作并创建了一个支持对象创建的 repo。

如果有人正在努力实现同样的目标,可以参考这个 repo:https://github.com/Akashtyagi08/Pretrained-im2txt-single_instance