Преобразовал изображение в массив numpy, изменил его размер, чтобы его можно было разделить на 3, и при этом он по-прежн ⇐ Python

[Anonymous]

Я попробовал этот код из руководства по маскировке изображения:

Код: Выделить всё

def show_mask(mask, ax):
color = np.array([30/255, 144/255, 255/255, 0.6])
h, w = np.array(mask).shape[-2:]
mask_image = np.array(mask).reshape(h, w, -1) * color.reshape(1, 1, -1)
ax.imshow(mask_image)

fig, axes = plt.subplots()
axes.imshow(pic)
show_mask(pix, axes)

и выдает эту ошибку

Код: Выделить всё

ValueError: operands could not be broadcast together
with shapes (1420,3,1080) (1,1,4)

Изображение и предполагаемая маска являются трехмерными, поэтому умножение матрицы (изображение маски) должно быть допустимым.
На самом деле в наборе данных есть заданная маска, это то же самое, что я собираюсь сделать?
Ошибка до этого заключалась в том, что изображение не является пустым массивом, я уже изменил размер изображения и конвертировал его в numpy, но проблема остается. Я что-то пропустил?

Подробнее здесь: https://stackoverflow.com/questions/792 ... 3-and-stil