Как уже упоминалось, это еще одно возможное решение. Я не уверен, насколько это эффективно по сравнению с другими предлагаемыми ответами, которые выглядят гораздо более элегантно и чисто.
я есть изображение в формате RGB или в градациях серого (скажем, я конвертировал его в Gimp), теперь каждый раз, когда я загружаю изображение в градациях серого или просто преобразую его в формат градаций серого, форма всегда говорит [высота, ширина] без третьего размерность (количество цветовых каналов).
Я знаю, что обычно черно-белые изображения хранятся в таком формате, но мне[height, width, 1] форма изображения, то, что вы получите, скажем:
numpy.zeros(shape=[400, 400, 1])