Vgg16 применяется к изображениям в градациях серого

Я хочу использовать предварительно обученные нейронные сети VGG16, используя изображения в градациях серого в керасах. Если я подключаю свои изображения непосредственно к модели VGG, я получаю эту ошибку:

ValueError: Error when checking input: expected vgg16_input to have 4 
dimensions, but got array with shape (10517, 48, 48)

Это очевидно: я использую 1-канальные изображения (в градациях серого) в качестве входных данных, в то время как vgg16 ожидает 3 канала (RGB).

Возможным решением будет дублирование каждого изображения в градациях серого для каждого из 3 каналов. Но я брожу, если существуют другие, более умные и эффективные способы.

Любое предложение?

0 ответов

Другие вопросы по тегам