createHippMapp3rUnetModel3D: Implementation of the "HippMapp3r" U-net architecture

View source: R/createCustomUnetModel.R

createHippMapp3rUnetModel3DR Documentation

Implementation of the "HippMapp3r" U-net architecture

Description

Creates a keras model implementation of the u-net architecture described here:

Usage

createHippMapp3rUnetModel3D(inputImageSize, doFirstNetwork = TRUE)

Arguments

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.

Details

\url{https://onlinelibrary.wiley.com/doi/pdf/10.1002/hbm.24811}

with the implementation available here:

\url{https://github.com/mgoubran/HippMapp3r}

Value

a u-net keras model

Author(s)

Tustison NJ

Examples

## Not run: 

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" )


## End(Not run)

ANTsX/ANTsRNet documentation built on April 28, 2024, 12:16 p.m.