首页 > 解决方案 > 是否可以将多个图像打包到一个 TFRecord 示例中?

问题描述

我的 ML 模型的输入是每个示例的可变数量的图像(但每组只有 1 个标签),因此我希望能够将多个图像打包到单个 TFRecord 示例中。但是,我在网上遇到的每个示例都是单个图像、单个标签,这是可以理解的,因为这是最常见的用例。我也想知道解码...似乎 tf.image.decode_png 一次只处理一张图像,但也许我可以将所有图像转换为 tf.string 并使用 tf.decode_raw,然后调整大小以获取所有图像?

谢谢

标签: tensorflowtfrecord

解决方案


推荐阅读