使用Tensorflow Queue从3D数组中随机选择一个窗口

时间:2017-07-28 14:21:40

标签: tensorflow

我有一个3D numpy数组数据集AxBxC。如何应用Tensorflow RandomShuffleQueue在数据集的前两个维度上实现窗口大小为dequeue的随机选择(dequeue_manyaxbxC)?

A = 500
B = 600
C = 400
data = np.random.rand(A, B, C)
q. tf.RandomShuffleQueue(capacity=???, shapes=???)

我想使用Queue而不是tf.random_crop来减少placeholder引起的CPU-GPU通信的开销。

0 个答案:

没有答案