Tensorflow中的RaggedTensor环

zbdgwd5y  于 2023-03-09  发布在  其他
关注(0)|答案(1)|浏览(169)

我想知道是否有任何方法也循环过粗糙Tensor,类似于tf.map_fn。我的粗糙Tensor有不同数量的行,但包含4个点,我想检索。
输入如下所示:

ragged_tensor[0] equals (100, 4)
ragged_tensor[1] equals (50, 4)

现在,我可以通过在RaggedTensor内部的第一个Tensor上循环来检索所有的点:

test = tf.map_fn(lambda box: tf.image.crop_to_bounding_box(img, box[0], box[1], box[2], box[3]), tf.cast(boxes, tf.int32), dtype=tf.float32)

有没有人有这方面的经验,或者可能给予我一些提示和技巧?所有的帮助都很感激。

ssgvzors

ssgvzors1#

这是一种获得整个点数组的方法:

points = tf.reshape(ragged_tensor.flat_values, [-1, 4])

相关问题