Tensorflow resize_image_with_crop_or_pad

Question

我想调用tf.image.resize_image_with_crop_or_pad(images,height,width)来调整输入图像的大小。由于我输入的图像都是二维像素数字数组的形式，而resize_image_with_crop_or_pad的图像输入必须是三维或四维张量，这将导致一个错误。我该怎么办？

Answer 1

假设您得到了images，这是一个[n, W, H] numpy nd数组，其中n是图像的数量，W和H是图像的width和height。

1. 将图像转换为张量，以便能够使用tensorflow函数： tf_images = tf.constant(images)
2. 将tf_images转换为tensorflow使用的图像数据格式(从而从n, W, H转换为n, H, W) tf_images = tf.transpose(tf_images, perm=[0,2,1])
3. 在tensorflow中，每个图像都有一个深度channell，因此，尽管您使用的是灰度图像，但我们必须添加depth=1 channell。 tf_images = tf.expand_dims(tf_images, 2)
4. 现在您可以使用tf.image.resize_image_with_crop_or_pad来调整批处理的大小( [n, H, W, 1] (4-d张量)的形状是怎样的)： resized = tf.image.resize_image_with_crop_or_pad(tf_images,height,width)