tf.random.shuffle()导致错误的tfrecord检索

时间:2018-11-17 13:00:41

标签: python tensorflow

假设我有200个tfrecord文件,每个tfrecord文件中有60个数字(在我们的情况下为整数)。每次,我想从16个tfrecord文件中获得8个数字(随机选择)。 为此,我使用

all_file_name = 'tfrecord-*'

tf.train.match_filenames_once(all_file_name)

tf.random_shuffle(all_tfrecord_files)[0:16]

获取16个tfrecord文件。

这里

for i in range(k):
    data_loaded = decode_from_tfrecords([file_k[i]], data_shape, n, min_after_dequeue, num_threads)
    I.append(data_loaded)

我从16条tfrecord中的每条中生成8个数字。

但是,我发现输出num并非来自16 tfrecord。

the output nums 然后,我删除tf.random.shuffe()(即使用all_tfrecord_files[0:16]

这次,输出数字与tfrecord值匹配。

output nums 详细信息可以在see here

中找到

您能告诉我问题吗,谢谢!

0 个答案:

没有答案