createSysuMediaUnetModel3D: 3-D variant of the sysu_media U-net architecture

In ANTsX/ANTsRNet: Neural Networks for Medical Image Processing

3-D variant of the sysu_media U-net architecture

Description

3-D variant of the sysu_media U-net architecture

Usage

createSysuMediaUnetModel3D(
  inputImageSize,
  numberOfFilters = NULL,
  anatomy = "wmh"
)

Arguments

inputImageSize	Used for specifying the input tensor shape. This will be c(200, 200, 2) for t1/flair input and c(200, 200, 1) for flair-only input for wmhs. For the claustrum, it is c(180, 180, 1).
numberOfFilters	specifies schedule for the encoding/decoding layers.
anatomy	"wmh" or "claustrum". Vestigial option from the original 2-D version.

Value

a u-net keras model

Author(s)

Tustison NJ

Examples

## Not run: 

model <- createSysuMediaUnetModel3D( c( 64, 64, 64, 2 ), numberOfFilters = c( 64, 128, 256, 512 ) )


## End(Not run)

ANTsX/ANTsRNet documentation built on Nov. 21, 2024, 4:07 a.m.