Guest User

Untitled

a guest
Jun 23rd, 2018
79
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.79 KB | None | 0 0
  1. inputs = Input(shape=patch_shape)
  2.  
  3. noise = GaussianNoise(2.0)(inputs)
  4. conv2 = Conv3D(32, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(noise)
  5. conv2 = Conv3D(32, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(conv2)
  6. pool2 = MaxPooling3D(pool_size=(2, 2, 2),data_format="channels_first")(conv2)
  7.  
  8. conv3 = Conv3D(64, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(pool2)
  9. conv3 = Conv3D(64, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(conv3)
  10. pool3 = MaxPooling3D(pool_size=(2, 2, 2),data_format="channels_first")(conv3)
  11.  
  12. conv4 = Conv3D(128, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(pool3)
  13. conv4 = Conv3D(128, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(conv4)
  14.  
  15. up5 = UpSampling3D(size=(2, 2, 2), data_format="channels_first")(conv4)
  16. up5 = concatenate([up5, conv3],axis=1)
  17. conv5 = Conv3D(64, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(up5)
  18. conv5 = Conv3D(64, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(conv5)
  19.  
  20. up6 = UpSampling3D(size=(2, 2, 2), data_format="channels_first")(conv5)
  21. up6 = concatenate([up6, conv2],axis=1)
  22. conv6 = Conv3D(32, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(up6)
  23. conv6 = Conv3D(32, (3, 3, 3), padding="same", activation="relu",data_format="channels_first")(conv6)
  24.  
  25. conv8 = Conv3D(num_y_channel, (3, 3, 3), padding="same", activation='sigmoid',data_format="channels_first")(conv6)
  26. model = Model(input=inputs, output=conv8)
  27.  
  28. def dice_coef_loss(y_true, y_pred):
  29. y_true_f = K.flatten(y_true)
  30. y_pred_f = K.flatten(y_pred)
  31. intersection = K.sum(y_true_f * y_pred_f)
  32. return -(2. * intersection) / (K.sum(y_true_f) + K.sum(y_pred_f))
Add Comment
Please, Sign In to add comment