我的网络拍摄的图像尺寸为100 x 100像素.因此,我必须调整数据集大小不同的图像的大小.我希望能够从给定图像中提取最大的中央正方形区域,然后将其调整为100 x 100.
更准确地说,假设图像的宽度为200像素,高度为50像素.然后,我想提取最大的中央正方形区域,在此示例中为50 x 50,然后将图像调整为100 x 100像素.
使用Tensorflow的正确方法是什么?现在我正在使用tf.image.resize_images()扭曲图像,我想摆脱它.
解决方法:
听起来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))