Кто-нибудь знает, почему последний аргумент в input_shape должен быть указан 3 в приложении керас?

Question

Я хочу использовать предварительно обученную сеть, такую ​​как VGG, ResNet.Находясь в Keras, необходимо указать формат в (w, h, 3) в input_shape.Если я хочу указать канал 1, есть ли еще хитрости?

conv_vgg = keras.application.VGG16(input_shape=(224,224,3))

Я хочу указать от 3 до 1:

conv_vgg = keras.application.VGG16(input_shape=(224,224,1))

Заранее спасибо!

Answer 1

Предварительно обученные сети, прошедшие обучение по imagenet или другим наборам данных изображений.Это означает, что обучение с изображениями RGB, поэтому для использования предварительно обученной сети требуется три канала.

Если вы хотите использовать предварительно обученные сети для одного изображения канала, вы можете повторить ваш канал три раза и продолжить.(Повторите копирование еще два раза для вашего 1-канального изображения, от (224 224,1) формы до (224 224,3) формы (3-канальное изображение).