sleap.nn.architectures.common

Common utilities for architecture and model building.

class sleap.nn.architectures.common.IntermediateFeature(tensor: tensorflow.python.framework.ops.Tensor, stride: int)

Intermediate feature tensor for use in skip connections.

This class is effectively a named tuple to store the stride (resolution) metadata.

tensor

The tensor output from an intermediate layer.

stride

Stride of the tensor relative to the input.

property scale

Return the absolute scale of the tensor relative to the input.

This is equivalent to the reciprocal of the stride, e.g., stride 2 => scale 0.5.