От RGB до GREY с использованием opencv удаляет 1 канал - PullRequest
0 голосов
/ 31 октября 2018

Я использую их для преобразования изображения RGB в СЕРЫЙ

i = cv2.resize(img, (100, 100))
i = cv2.cvtColor(i, cv2.COLOR_BGR2GRAY)

Это дает мне (100,100), но я хочу (100,100,1), так как это требуется для ввода в CNN следующим образом

model.add(Conv2D(16, (2,2), input_shape=(100,100,1 ), activation='relu'))

Покажите, как мне преобразовать RGB в СЕРЫЙ, чтобы коэффициент 1 канала оставался неизменным.

Ответы [ 2 ]

0 голосов
/ 31 октября 2018

Вы можете просто добавить дополнительное измерение
np.expand_dims(i, axis=0)

0 голосов
/ 31 октября 2018

Использовать NumPy Reshape:

i = i.reshape(i.shape + (1,))
...