DenseCLNeck¶
- class mmpretrain.models.necks.DenseCLNeck(in_channels, hid_channels, out_channels, num_grid=None, init_cfg=None)[source]¶
The non-linear neck of DenseCL.
Single and dense neck in parallel: fc-relu-fc, conv-relu-conv. Borrowed from the authors’ code.
- Parameters:
- forward(x)[source]¶
Forward function of neck.
- Parameters:
x (Tuple[torch.Tensor]) – feature map of backbone.
- Returns:
avgpooled_x
: Global feature vectors.x
: Dense feature vectors.avgpooled_x2
: Dense feature vectors for queue.
- Return type:
Tuple[torch.Tensor, torch.Tensor, torch.Tensor]