官方的 tensorflow Show 和 Tell-im2txt模型带有一个 bazel 构建依赖项,用于为图像生成标题。对于每个图像的执行,会创建一个会话,并为每个图像重复加载检查点。
如何避免重复加载检查点?
我们可以创建一个将图像作为输入并通过预加载的检查点运行它的函数吗?
官方的 tensorflow Show 和 Tell-im2txt模型带有一个 bazel 构建依赖项,用于为图像生成标题。对于每个图像的执行,会创建一个会话,并为每个图像重复加载检查点。
如何避免重复加载检查点?
我们可以创建一个将图像作为输入并通过预加载的检查点运行它的函数吗?
我终于解耦了它的工作并创建了一个支持对象创建的存储库。
如果有人正在努力实现同样的目标,可以参考这个 repo:https ://github.com/Akashtyagi08/Pretrained-im2txt-single_instance