python - 使用 cv2 调整图像大小

Question

我正在尝试将从 cifar10 检索到的原始 32x32 图像调整为 96x96 以便与 MobileNetV2 一起使用，但是我遇到了此错误。尝试了多种解决方案，但似乎没有任何效果。

我的代码:

for a in range(len(train_images)):
    train_images[a] = cv2.resize(train_images[a], dsize=(minSize, minSize), interpolation=cv2.INTER_CUBIC)

我遇到的错误:

----> 8     train_images[a] = cv2.resize(train_images[a], dsize=(minSize, minSize), interpolation=cv2.INTER_CUBIC)
ValueError: could not broadcast input array from shape (96,96,3) into shape (32,32,3)

Answer 1

这只是因为您正在从 train_images 读取 32x32 图像并尝试将 reshape 后的图像 (96x96) 保存在同一个数组中，这是不可能的!尝试这样的事情:

train_images_reshaped = np.array((num_images, 96, 96, 3))
for a in range(len(train_images)):
    train_images_reshaped[a] = cv2.resize(train_images[a], dsize=(minSize, minSize), interpolation=cv2.INTER_CUBIC)