Добавить указанное значение в качестве нового измерения массива

Question

После прочтения изображения у меня есть массив форм (224,224,3). Однако я хотел бы преобразовать это в форму (4,224,224,3).

Я бы хотел повторить те же значения.

Я пытаюсь добавить, как показано ниже, но это не работает.

np.append(image,[[[4]]],axis=1)

Вместо этого выдается ошибка ниже

ValueError: all the input arrays must have same number of dimensions

Я ожидаю, что моя выходная форма будет (4,224,224,3)

Можете ли вы рассказать мне, как это сделать?

Answer 1

Вы можете использовать np.repeat, настроив ось на 0:

out = np.repeat([image], 4, axis=0)
out.shape
# (4, 224, 224, 3)