build_backbone

lightning_pose.models.backbones.vits.build_backbone(backbone_arch: str, image_size: int = 256, **kwargs)

Load backbone weights for resnet models.

Parameters:

• backbone_arch – which backbone version/weights to use

• image_size – height/width in pixels of images (must be square)

Returns:

tuple

• backbone: pytorch model

• num_fc_input_features (int): number of input features to fully connected layer