tensorflow - 在 TensorFlow 中提取地标周围的多个补丁

Question

我正在尝试使用 TensorFlow 在同一图像中围绕一组 2D 地标提取多个补丁。

给定一个形状为 2D 地标的输入张量[batch_size, num_landmarks, 2]和一个形状为输入图像的张量，[batch_size, num_rows, num_cols, num_channels]我想返回一个包含[batch_size, num_landmarks, patch_rows, patch_cols, num_channels].

现在我们可以假设，batch_size=1如果是这样，以下代码将执行上述操作：

im = tf.tile(im, (num_landmarks, 1, 1, 1))
patches = tf.image.extract_glimpse(im, (patch_cols, patch_rows), landmarks, centered=False, normalized=False)

我基本上重复输入图像的次数与我有地标一样多，然后提取一瞥。当我有很多地标时，这当然是疯狂的，所以我想知道是否存在更好的方法。

编辑：

我认为tf.gather_nd可以做到这一点，所以我正在构建提取补丁所需的索引。

标签： tensorflow