Используйте `tf.image.resize_image_with_crop_or_pad`, чтобы изменить размер массива

Question

Я хочу использовать tf.image.resize_image_with_crop_or_pad для массива Numpy формы (100,100,2), чтобы обрезать его или дополнить до целевой формы (h,w,2).

Однако, когда я делаю:

img = resize_image_with_crop_or_pad(img, target_height, target_width)
img = np.array(img)

img.shape оценивается как (), что не соответствует ожиданиям. Как мне превратить выходные данные этой функции в массив правильной формы?

Answer 1

img = resize_image_with_crop_or_pad(img_tensor, target_height, target_width)
with tf.Session as sess:
    img_output = sess.run(img)

Теперь img_output является массивом numpy, но обратите внимание, что img должен иметь tf.Tensor формы [1, height, width, channels], так что вы можете сделать это заранее, предполагая, что ваше входное изображение уже является массивом numpy:

img_input = np.expand_dims(img_input, 0)
img_tensor = tf.convert_to_tensor(img_input)