Method __init__

modules/VIPTRv2.py:917–938

(self, in_chans=3, embed_dim=96, norm_layer=None)

class PatchEmbed(nn.Module):
    def __init__(self, in_chans=3, embed_dim=96, norm_layer=None):
        super().__init__()

        self.embed_dim = embed_dim
        self.proj = nn.Sequential(
            ConvBNLayer(
                in_channels=in_chans,
                out_channels=embed_dim // 2,
                kernel_size=3,
                stride=2,
                padding=1,
                act=nn.GELU,
                bias_attr=False),
            ConvBNLayer(
                in_channels=embed_dim // 2,
                out_channels=embed_dim,
                kernel_size=3,
                stride=2,
                padding=1,
                act=nn.GELU,
                bias_attr=False)
        )

    def forward(self, x):
        x = self.proj(x).permute(0, 2, 3, 1).contiguous()