Method forward

p3former/decode_heads/p3former_head.py:181–184

(self, input)

            nn.Linear(channels[-2], channels[-1]))
        
    def forward(self, input):
        for layer in self.mlp:
            input = layer(input)
        return input

@MODELS.register_module()
class _P3FormerHead(nn.Module):