createSimpleFullyConvolutionalNeuralNetworkModel3D: Implementation of the "SCFN" architecture for Brain/Gender...

View source: R/createCustomModel.R

createSimpleFullyConvolutionalNeuralNetworkModel3DR Documentation

Implementation of the "SCFN" architecture for Brain/Gender prediction

Description

Creates a keras model implementation of the Simple Fully Convolutional Network model from the FMRIB group:

Usage

createSimpleFullyConvolutionalNeuralNetworkModel3D(
  inputImageSize,
  numberOfFiltersPerLayer = c(32, 64, 128, 256, 256, 64),
  numberOfBins = 40,
  dropoutRate = 0.5,
  doExperimentalVariant = FALSE
)

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

numberOfFiltersPerLayer

number of filters for the convolutional layers

numberOfBins

number of bins for final softmax output.

dropoutRate

dropout rate before final convolution layer.

Details

    \url{https://github.com/ha-ha-ha-han/UKBiobank_deep_pretrain}

Value

a SCFN keras model

Author(s)

Tustison NJ

Examples


library( ANTsRNet )

model <- createSimpleFullyConvolutionalNeuralNetworkModel3D( list( NULL, NULL, NULL, 1 ) )


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