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

View source: R/createCustomUnetModel.R

createSysuMediaUnetModel3DR Documentation

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 April 28, 2024, 12:16 p.m.