В Keras мы можем определить сеть следующим образом. Есть ли способ вывести форму после каждого слоя. Например, я хочу, чтобы распечатать форму inputs
после строки, определяющей inputs
, а затем распечатать форму conv1
после строки, определяющей conv1
и т.д.распечатать форму каждого слоя в сетевой архитектуре
inputs = Input((1, img_rows, img_cols))
conv1 = Convolution2D(64, 3, 3, activation='relu', init='lecun_uniform', W_constraint=maxnorm(3), border_mode='same')(inputs)
conv1 = Convolution2D(64, 3, 3, activation='relu', init='lecun_uniform', W_constraint=maxnorm(3), border_mode='same')(conv1)
pool1 = MaxPooling2D(pool_size=(2, 2))(conv1)
conv2 = Convolution2D(128, 3, 3, activation='relu', init='lecun_uniform', W_constraint=maxnorm(3), border_mode='same')(pool1)
conv2 = Convolution2D(128, 3, 3, activation='relu', init='lecun_uniform', W_constraint=maxnorm(3), border_mode='same')(conv2)
pool2 = MaxPooling2D(pool_size=(2, 2))(conv2)
Можете ли вы подробнее рассказать о том, как его использовать? Например, я хотел бы распечатать информацию о пуле1. Я попробовал config = pool1.get_config(), это не сработало. – user288609
отредактировал ответ. пожалуйста, проверьте –