Tensorflow FIFOQueue将稀疏数组排入队列

时间:2017-06-10 21:56:45

标签: arrays tensorflow deep-learning tensorflow-gpu

如何使用稀疏数组提供FIFOQueue?

我为每个图像设置了可变大小的bbox,因此我使用两个队列来接收图像和相应的对象bbox。

data = [[[1619, 1153, 1977, 1733]], [[1488, 3569, 1594, 3604], [1282, 3476, 1382, 3526], [1667, 3613, 1814, 3653], [2374, 3459, 2848, 3632]],[[2207, 3513, 2671, 3632]], [[2554, 1222, 2865, 1377]]]
data_queue = tf.FIFOQueue(capacity=5, dtypes=[tf.int32, tf.int32], shapes=[[None], [5]])

data_enqueue_op = data_queue.enqueue_many([data])

此代码提供ValueError: setting an array element with a sequence.

0 个答案:

没有答案