Method forward

sam2_train/modeling/backbones/utils.py:91–95

(self, x: torch.Tensor)

        )

    def forward(self, x: torch.Tensor) -> torch.Tensor:
        x = self.proj(x)
        # B C H W -> B H W C
        x = x.permute(0, 2, 3, 1)
        return x