deepLandmarkRegressionWithHeatmaps: Deep heatmap-based landmark regression stage

In stnava/patchMatchR: Local patch matching with flexible transformation models in n-dimensional images

Deep heatmap-based landmark regression stage

Description

Deep heatmap-based landmark regression stage

Usage

deepLandmarkRegressionWithHeatmaps(
  model,
  activation = c("none", "relu", "trelu", "softmax", "sigmoid"),
  theta,
  useMask = FALSE
)

Arguments

model	input deep model, presumably a unet. it should have an input dimensionality that is dimension-appropriate. the number of output channels should match the number of landmarks and should be an image.
activation	the activation function for the regression maps
theta	the theta parameter for thresholded relu
useMask	boolean adds a 2nd input for a mask that gets applied to the output activations.

Value

the augmented model

stnava/patchMatchR documentation built on March 23, 2022, 6:47 a.m.