R/createCustomUnetModel.R
createHippMapp3rUnetModel3D.Rd
Creates a keras model implementation of the u-net architecture described here:
createHippMapp3rUnetModel3D(inputImageSize, doFirstNetwork = TRUE)
inputImageSize | Used for specifying the input tensor shape. The shape (or dimension) of that tensor is the image dimensions followed by the number of channels (e.g., red, green, and blue). |
---|---|
doFirstNetwork | boolean dictating if the model built should be the first (initial) network or second (refinement) network. |
a u-net keras model
\url{https://onlinelibrary.wiley.com/doi/pdf/10.1002/hbm.24811}
with the implementation available here:
\url{https://github.com/mgoubran/HippMapp3r}
Tustison NJ
if (FALSE) { model1 <- createHippMapp3rUnetModel3D( c( 160, 160, 128, 1 ), doFirstNetwork = TRUE ) model2 <- createHippMapp3rUnetModel3D( c( 112, 112, 64, 1 ), doFirstNetwork = FALSE ) json_config <- model_to_json( model1 ) writeLines( json_config, "/Users/ntustison/Desktop/model1_config.json" ) }