我正在使用 U_Net 分割模型通過 Kersa 和 Tensorflow 2 進行醫學影像分割。我想在模型中添加一個 dropout，但我不知道在哪里添加它？任何人都可以幫忙嗎？
謝謝
uj5u.com熱心網友回復：
是的，在 unet 的實作中沒有 dropout 層，但是你可以使用正則化器
set_regularization(model, kernel_regularizer=keras.regularizers.l2(0.001),bias_regularizer=keras.regularizers.l2(0.001))
您也可以嘗試資料增強。但是如果有必要添加 dropout 你可以在一些層之后停止并在它之后添加你想要的 dropout 層，如下所示：
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)
轉載請註明出處，本文鏈接：https://www.uj5u.com/shujuku/339681.html
上一篇：用于使用python上傳csv檔案的oracle系統的Ap??i
下一篇：無法在本機反應中加載靜態影像