我想知道是否有任何方法也循环过粗糙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)
有没有人有这方面的经验,或者可能给予我一些提示和技巧?所有的帮助都很感激。
1条答案
按热度按时间ssgvzors1#
这是一种获得整个点数组的方法: