Method __init__

sam2/modeling/backbones/image_encoder.py:16–28

(
        self,
        trunk: nn.Module,
        neck: nn.Module,
        scalp: int = 0,
    )

class ImageEncoder(nn.Module):

    def __init__(
        self,
        trunk: nn.Module,
        neck: nn.Module,
        scalp: int = 0,
    ):
        super().__init__()
        self.trunk = trunk
        self.neck = neck
        self.scalp = scalp
        assert (
            self.trunk.channel_list == self.neck.backbone_channel_list
        ), f"Channel dims of trunk and neck do not match. Trunk: {self.trunk.channel_list}, neck: {self.neck.backbone_channel_list}"

    def forward(self, sample: torch.Tensor):
        # Forward through backbone