Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- inputs = Input(shape=patch_shape)
- noise = GaussianNoise(2.0)(inputs)
- conv2 = Conv3D(32, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(noise)
- conv2 = Conv3D(32, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(conv2)
- pool2 = MaxPooling3D(pool_size=(2, 2, 2),data_format="channels_first")(conv2)
- conv3 = Conv3D(64, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(pool2)
- conv3 = Conv3D(64, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(conv3)
- pool3 = MaxPooling3D(pool_size=(2, 2, 2),data_format="channels_first")(conv3)
- conv4 = Conv3D(128, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(pool3)
- conv4 = Conv3D(128, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(conv4)
- up5 = UpSampling3D(size=(2, 2, 2), data_format="channels_first")(conv4)
- up5 = concatenate([up5, conv3],axis=1)
- conv5 = Conv3D(64, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(up5)
- conv5 = Conv3D(64, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(conv5)
- up6 = UpSampling3D(size=(2, 2, 2), data_format="channels_first")(conv5)
- up6 = concatenate([up6, conv2],axis=1)
- conv6 = Conv3D(32, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(up6)
- conv6 = Conv3D(32, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(conv6)
- conv8 = Conv3D(num_y_channel, (3, 3, 3), padding="same", activation='sigmoid',data_format="channels_first")(conv6)
- model = Model(input=inputs, output=conv8)
- def dice_coef_loss(y_true, y_pred):
- y_true_f = K.flatten(y_true)
- y_pred_f = K.flatten(y_pred)
- intersection = K.sum(y_true_f * y_pred_f)
- return -(2. * intersection) / (K.sum(y_true_f) + K.sum(y_pred_f))
Add Comment
Please, Sign In to add comment