категории к keras imagenet сети пока я не убрал второй Dense слой модель почти моментально скатывалась во все 0 или все 1, а сейчас отлично тренируется (тьфутьфутьфу)?
x = base_model.output
x = GlobalAveragePooling2D()(x)
x = Dense(1792)(x)
x = BatchNormalization()(x)
x = Activation(tf.nn.leaky_relu)(x)
# x = Dense(448)(x)
# x = BatchNormalization()(x)
# x = Activation('relu')(x)
predictions = Dense(2, activation='sigmoid')(x)
бери резнет18-50-101 и не думай
А зачем у тебя сигмоида если нейрона два
Обсуждают сегодня