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