Я новичок в глубоком обучении; Я хочу обучить сеть на участках изображения размером (256, 256, 3), чтобы предсказать три метки сегментации по пикселям. Для начала я хочу предоставить один сверточный слой:
model = Sequential()
model.add(Convolution2d(32, 16, 16, input_shape=(3, 256, 256))
На данный момент выход модели представляет собой изображение с 32 каналами. Теперь я хочу добавить плотный слой, который объединяет все эти 32 канала в три канала, каждый из которых предсказывает вероятность класса для одного пикселя.
Как я могу это сделать?