Добавление входов в сверточную нейронную сеть после свертки и объединения слоев

Question

Я строю сверточную нейронную сеть, которая будет содержать определенное количество слоев свертки и объединения.Проблема в том, что я хотел добавить несколько дополнительных входов после шагов извлечения объектов (свертка + объединение).

Эти дополнительные входы будут добавлены к сглаженным картам объектов (первый слой полностью связанного слоя).Я хотел спросить, есть ли какая-либо документация для реализации этого, в tenorflow или (если мне повезет) в Керасе.Заранее спасибо и хорошего дня.

Answer 1

Вы можете создать такую ​​модель с классом tf.keras.models.Model.

Сначала мы можем построить модель tf.keras.models.Sequential для слоев Convolution и Pooling.

conv_model = tf.keras.models.Sequential( [ ... ] )

Затем, как вы сказали, нам нужна полностью подключенная плотная сеть.Мы создаем его аналогично вышеприведенной модели.

fc_model = tf.keras.models.Sequential( [ ... ] )

Затем собираем слои Input с созданными нами моделями.

input1 = Input( ... )
input2 = Input( ... )

cnn_output = conv_model( input1 )
output = fc_model( [ cnn_output , input2 ] )

model = tf.keras.models.Model( [ input1 , input2 ] , output )