layer_anchor_box_2d: Anchor box layer (2-D and 3-D)

In ANTsX/ANTsRNet: Neural Networks for Medical Image Processing

Anchor box layer (2-D and 3-D)

Description

Wraps a custom layer for the SSD network

Usage

layer_anchor_box_2d(
  object,
  imageSize,
  scale,
  nextScale,
  aspectRatios,
  variances,
  name = NULL,
  trainable = TRUE
)

layer_anchor_box_3d(
  object,
  imageSize,
  scale,
  nextScale,
  aspectRatios,
  variances,
  name = NULL,
  trainable = TRUE
)

Arguments

object	Object to compose layer with. This is either a keras::keras_model_sequential to add the layer to, or another Layer which this layer will call.
imageSize	size of the image, passed to create_layer
scale	box scale, passed to create_layer
nextScale	box scale, passed to create_layer
aspectRatios	list of ratios used for the boxes, passed to create_layer
variances	list of variances, passed to create_layer
name	The name of the layer
trainable	logical indicating if it is trainable or not

Value

a keras layer tensor

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