合并TFRecord数据集

Question

我有两个TFRecordDatasets。其中一个有一个图像和相应的嵌入。另一种则具有相应的嵌入和图像特征。我想做一个数据集，它现在有图像，嵌入和功能。

d1(image,embeddings),d2(embeddings,features) ----> D(image,embeddings, features)

Answer 1

可以使用zip函数组合TfRecordDataset。找到下面的示例工作示例。

import tensorflow as tf
images = tf.data.Dataset.range(1, 4)  
embeddings = tf.data.Dataset.range(4, 7)  
features = tf.data.Dataset.range(7, 9)
ds = tf.data.Dataset.zip((a, b, d))
for element in ds.as_numpy_iterator():
  print(element)

输出

(1, 4, 7)
(2, 5, 8)