How to crop tensor in the center in tensorflow?

Question

The tensor shape is [batch, height, width, channel]. Let's take the tensor as [8, 64, 64, 1], how could I crop it in the center to [8, 56, 56, 1]? Note that the feature 56*56 is center taken from 64*64.

Answer 1

我认为你要找的是 tf.image.central_crop

Doc link for tf.image.central_crop

在处理批处理图像的实例中，您需要使用tf.map_fn函数。它允许您将单个张量函数应用于一系列函数。例如，您可能需要以下内容：

cropped_image=tf.map_fn(lambda img: tf.image.central_crop(img,0.5), imgin, parallel_iterations=8, name="crop1")

这应该像您通常在图表中找到的一样批量返回裁剪后的图像！