Преобразование изображения ndarray в изображение в градациях серого с той же формой массива

Question

Я пытаюсь написать функцию python, которая принимает изображение ndarray в качестве параметра и возвращает ndarray версии этого изображения в градациях серого, используя широковещательную передачу.

Моя главная проблема сейчас заключается в том, что мое получающееся изображениене имеет той же формы (800, 400, 3), что и оригинал.

Редактировать: получается, что я просто забыл преобразовать полученный массив в np.uint8

Answer 1

Ваша первая convert_bw функция не имеет никаких проблем, я думаю, проблема в том, как вы видите изображение результата (библиотека, которую вы используете, не может показать вам результат хорошо). например, вы можете увидеть результат с помощью следующего кода:

import matplotlib.pyplot as plt
bw = convert_bw(img)
plt.imshow(bw.astype(np.uint8)) 

Это сработало для меня.