python - 在python中调整4维数组的图像大小

Question

我正在使用 Python 3。

我有一个类型为 unit8 且大小为 (60000, 1, 28, 28) 的变量数据，这意味着我有 60000 具有 1 个 channel 且大小为 28x28 像素的图像。
现在我想将图像大小调整为 60x60 像素，这样我就可以得到 (60000, 1, 60, 60)

的大小

有人可以帮助我吗？

Answer 1

以下解决方案对我有用。列表理解后，列表将被堆叠回 numpy 的批处理:

import cv2
import numpy as np

target_shape = (60, 60)

def _resize_image(image, target):
   return cv2.resize(image, dsize=(target[0], target[1]), interpolation=cv2.INTER_LINEAR)

image = [_resize_image(image=i, target=target_shape) for i in data]
data_batch = np.stack(image, axis=0)