在 numpy 中向特定维度添加额外的通道

Question

我有一个形状数组

masks = encoded.reshape(n, h, w)
masks = np.expand_dims(masks, axis=3)
maskData = to_categorical(masks)
print(maskData.shape)

(16, 456, 684, 8)

我想在第 4 维中添加一个额外的通道，使其达到第 9 维，并用零填充

n,h,w,c = maskData.shape
maskData.reshape(n,h,w, 9)

似乎崩溃了。

Answer 1

我可能会创建一个新的所需大小的空 numpy 数组，然后用原始数据填充它

n,h,w,c = maskData.shape
wanted_size = 9
new_mask_data = np.zeros([n,h,w,wanted_size])
new_mask_data[:,:,:,:c] = maskData