TensorFlow数据集:从一个序列化的原型中解析多个示例

时间:2019-06-13 07:18:34

标签: python tensorflow tensorflow-datasets tfrecord

我有带序列化TF示例的TFRecords,其中包含序列化的图像,图像中对象周围的边界框以及每个边界框中的对象标签。

这意味着每个TF示例都有一个长度可变的标签列表,其中边框的数量正好一样。

我想使用这些数据来训练图像分类器。我的模型可以。这意味着每个原型都应返回多个示例:图像中的每个标签/边界框一个。

我不确定该怎么做

  1. 将每个protobuf解析为多个tf.Examples或
  2. 如何向Dataset API表示,每次调用map,我返回的都是多个示例,而不仅仅是一个。

指导和建议表示赞赏。谢谢。

0 个答案:

没有答案