python - Tensorflow:裁剪图像的最大中心正方形区域

Question

我的网络拍摄大小为 100 x 100 像素的图像。因此，我必须调整不同大小的数据集图像的大小。我希望能够从给定图像中提取最大的中心正方形区域，然后将其调整为 100 x 100。

更准确地说，假设图像的宽度为 200 像素，高度为 50 像素。然后我想提取最大的中心正方形区域，在此示例中为 50 x 50，然后将图像大小调整为 100 x 100 像素。

使用 Tensorflow 的正确方法是什么？现在我正在使用 tf.image.resize_images() 来扭曲图像，我想摆脱它。

Answer 1

听起来像 crop_to_bounding_box正在做你需要的事情:

import tensorflow as tf

def crop_center(image):
    h, w = image.shape[-3], image.shape[-2]
    if h > w:
        cropped_image = tf.image.crop_to_bounding_box(image, (h - w) // 2, 0, w, w)
    else:
        cropped_image = tf.image.crop_to_bounding_box(image, 0, (w - h) // 2, h, h)
    return tf.image.resize_images(cropped_image, (100, 100))