为U_Net Segmentation_Models添加Dropout图层

Question

我使用U_Net分割模型与Kersa和Tensorflow 2进行医学图像分割。我想添加一个dropout模型，但我不知道在哪里添加它?

Answer 1

是的，在unet的实现中没有退出层，但是您可以使用正则化器

set_regularization(model, kernel_regularizer=keras.regularizers.l2(0.001),bias_regularizer=keras.regularizers.l2(0.001))

您还可以尝试数据增强。但是如果需要添加渐变效果，你可以在一些图层之后再添加你想要的渐变图层，就像这样:

model = sm.Unet(......)
model_input = model.input
model_output = model.get_layer('final_conv').output (any layer you want)
#add dropout
model_output = keras.layers.Dropout(0.3)(model_output)
#add activation
output = keras.layers.Activation(activation, name=activation)(model_output)
model_dp = keras.models.Model(model_input, output)