Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/NVlabs/SegFormer
/ functions
Functions
711 in github.com/NVlabs/SegFormer
⨍
Functions
711
◇
Types & classes
147
↳
Endpoints
8
Method
forward
Forward function.
mmseg/models/backbones/unet.py:211
Method
forward
(self, x)
mmseg/models/backbones/unet.py:375
Method
forward
(self, x)
mmseg/models/backbones/cgnet.py:37
Method
forward
(self, x)
mmseg/models/backbones/cgnet.py:142
Method
forward
(self, x)
mmseg/models/backbones/cgnet.py:179
Method
forward
(self, x)
mmseg/models/backbones/cgnet.py:309
Method
forward
(self, x)
mmseg/models/backbones/mobilenet_v3.py:233
Method
forward
Forward function.
mmseg/models/backbones/hrnet.py:185
Method
forward
Forward function.
mmseg/models/backbones/hrnet.py:510
Method
forward
(self, x)
mmseg/models/backbones/fast_scnn.py:66
Method
forward
(self, x)
mmseg/models/backbones/fast_scnn.py:172
Method
forward
(self, higher_res_feature, lower_res_feature)
mmseg/models/backbones/fast_scnn.py:235
Method
forward
(self, x)
mmseg/models/backbones/fast_scnn.py:367
Method
forward
(self, x)
mmseg/models/backbones/mobilenet_v2.py:149
Method
forward
Forward function.
mmseg/models/backbones/resnet.py:67
Method
forward
Forward function.
mmseg/models/backbones/resnet.py:264
Method
forward
Forward function.
mmseg/models/backbones/resnet.py:632
Method
forward
(self, x)
mmseg/models/utils/drop.py:122
Method
forward
(self, x)
mmseg/models/utils/drop.py:158
Method
forward
Forward function.
mmseg/models/utils/self_attention_block.py:131
Method
forward
(self, x)
mmseg/models/utils/se_layer.py:53
Method
forward
Forward function.
mmseg/models/utils/up_conv_block.py:94
Method
forward
(self, x)
mmseg/models/utils/inverted_residual.py:81
Method
forward
(self, x)
mmseg/models/utils/inverted_residual.py:183
Method
forward
Forward function.
mmseg/models/decode_heads/sep_aspp_head.py:78
Method
forward
Forward function.
mmseg/models/decode_heads/uper_head.py:86
Method
forward
(self, x)
mmseg/models/decode_heads/segformer_head.py:28
Method
forward
(self, inputs)
mmseg/models/decode_heads/segformer_head.py:64
Method
forward
Forward function.
mmseg/models/decode_heads/gc_head.py:38
Method
forward
(self, inputs)
mmseg/models/decode_heads/fpn_head.py:54
Method
forward
Forward function.
mmseg/models/decode_heads/aspp_head.py:43
Method
forward
Forward function.
mmseg/models/decode_heads/aspp_head.py:93
Method
forward
Forward function.
mmseg/models/decode_heads/cc_head.py:32
Method
forward
(self, x)
mmseg/models/decode_heads/dnl_head.py:33
Method
forward
Forward function.
mmseg/models/decode_heads/dnl_head.py:122
Method
forward
Forward function.
mmseg/models/decode_heads/psp_head.py:46
Method
forward
Forward function.
mmseg/models/decode_heads/psp_head.py:93
Method
forward
Forward function.
mmseg/models/decode_heads/enc_head.py:50
Method
forward
Forward function.
mmseg/models/decode_heads/ocr_head.py:23
Method
forward
Forward function.
mmseg/models/decode_heads/ocr_head.py:73
Method
forward
Forward function.
mmseg/models/decode_heads/fcn_head.py:69
Method
forward
Forward function.
mmseg/models/decode_heads/ann_head.py:22
Method
forward
Forward function.
mmseg/models/decode_heads/ann_head.py:125
Method
forward
Forward function.
mmseg/models/decode_heads/ann_head.py:175
Method
forward
Forward function.
mmseg/models/decode_heads/ann_head.py:236
Method
forward
Forward function.
mmseg/models/decode_heads/lraspp_head.py:68
Method
forward
Forward function.
mmseg/models/decode_heads/dm_head.py:60
Method
forward
Forward function.
mmseg/models/decode_heads/dm_head.py:131
Method
forward
Forward function.
mmseg/models/decode_heads/psa_head.py:113
Method
forward
Forward function.
mmseg/models/decode_heads/ema_head.py:44
Method
forward
Forward function.
mmseg/models/decode_heads/ema_head.py:154
Method
forward
Forward function.
mmseg/models/decode_heads/nl_head.py:40
Method
forward
Forward function.
mmseg/models/decode_heads/apc_head.py:78
Method
forward
Forward function.
mmseg/models/decode_heads/apc_head.py:149
Method
forward
Forward function.
mmseg/models/decode_heads/da_head.py:41
Method
forward
Forward function.
mmseg/models/decode_heads/da_head.py:56
Method
forward
Forward function.
mmseg/models/losses/lovasz_loss.py:274
Method
forward
Forward function to calculate accuracy. Args: pred (torch.Tensor): Prediction of models. target (torch.Tensor): Targe
mmseg/models/losses/accuracy.py:68
Method
forward
Forward function.
mmseg/models/losses/cross_entropy_loss.py:175
Method
forward
(self, inputs)
tests/test_models/test_segmentor.py:67
Method
forward
(self, inputs, prev_out)
tests/test_models/test_segmentor.py:77
Method
forward_dummy
Dummy forward function.
mmseg/models/segmentors/encoder_decoder.py:130
Method
forward_test
Forward function for testing. Args: inputs (list[Tensor]): List of multi-level img features. img_metas (list[dict]):
mmseg/models/decode_heads/decode_head.py:192
Method
forward_test
Forward function for testing. Args: inputs (list[Tensor]): List of multi-level img features. prev_output (Tensor): Th
mmseg/models/decode_heads/cascade_decode_head.py:41
Method
forward_train
Forward function for training. Args: inputs (list[Tensor]): List of multi-level img features. img_metas (list[dict]):
mmseg/models/decode_heads/decode_head.py:172
Method
forward_train
Forward function for training. Args: inputs (list[Tensor]): List of multi-level img features. prev_output (Tensor): Th
mmseg/models/decode_heads/point_head.py:165
Method
forward_train
Forward function for training. Args: inputs (list[Tensor]): List of multi-level img features. prev_output (Tensor): Th
mmseg/models/decode_heads/cascade_decode_head.py:18
Method
freeze_patch_emb
(self)
mmseg/models/backbones/mix_transformer.py:299
Function
generate_labels
(img_id, detail, out_dir)
tools/convert_datasets/pascal_context.py:20
Method
get_classifier
(self)
mmseg/models/backbones/mix_transformer.py:306
Function
get_world_size
(process_group)
tests/test_models/test_forward.py:177
Method
init_weights
Initialize the weights in segmentor. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to N
mmseg/models/segmentors/base.py:65
Method
init_weights
Initialize the weights in backbone and heads. Args: pretrained (str, optional): Path to pre-trained weights. Defa
mmseg/models/segmentors/cascade_encoder_decoder.py:48
Method
init_weights
(self)
mmseg/models/necks/fpn.py:157
Method
init_weights
(self, pretrained=None)
mmseg/models/backbones/mix_transformer.py:276
Method
init_weights
Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to No
mmseg/models/backbones/hrnet.py:484
Method
init_weights
(self, pretrained=None)
mmseg/models/backbones/mobilenet_v2.py:136
Method
init_weights
Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to No
mmseg/models/backbones/resnet.py:600
Method
init_weights
Initialize weights of classification layer.
mmseg/models/decode_heads/decode_head.py:135
Method
init_weights
Initialize weights of classification layer.
mmseg/models/decode_heads/point_head.py:104
Method
losses
Compute segmentation and semantic encoding loss.
mmseg/models/decode_heads/enc_head.py:178
Function
lovasz_hinge
Binary Lovasz hinge loss. Args: logits (torch.Tensor): [B, H, W], logits at each pixel (between -infty and +infty). l
mmseg/models/losses/lovasz_loss.py:83
Function
lovasz_softmax
Multi-class Lovasz-Softmax loss. Args: probs (torch.Tensor): [B, C, H, W], class probabilities at each prediction (between 0
mmseg/models/losses/lovasz_loss.py:171
Method
make_res_layer
Pack all blocks in a stage into a ``ResLayer``.
mmseg/models/backbones/resnest.py:305
Method
make_res_layer
Pack all blocks in a stage into a ``ResLayer``
mmseg/models/backbones/resnext.py:139
Function
mask_cross_entropy
Calculate the CrossEntropy loss for masks. Args: pred (torch.Tensor): The prediction with shape (N, C), C is the number of cl
mmseg/models/losses/cross_entropy_loss.py:100
Function
multi_gpu_test
(model, data_loader, tmpdir=None, gpu_collect=False)
tests/test_eval_hook.py:115
Method
no_weight_decay
(self)
mmseg/models/backbones/mix_transformer.py:303
Method
reset_classifier
(self, num_classes, global_pool='')
mmseg/models/backbones/mix_transformer.py:309
Method
reset_drop_path
(self, drop_path_rate)
mmseg/models/backbones/mix_transformer.py:281
Method
sample
Placeholder for sample function.
mmseg/core/seg/sampler/base_pixel_sampler.py:11
Method
setup_class
(cls)
tests/test_data/test_loading.py:14
Method
simple_test
Simple test with single image.
mmseg/models/segmentors/encoder_decoder.py:263
Function
test_CLAHE
()
tests/test_data/test_transform.py:422
Function
test_accuracy
()
tests/test_models/test_losses.py:89
Function
test_adjust_gamma
()
tests/test_data/test_transform.py:343
Function
test_ann_forward
()
tests/test_models/test_forward.py:108
Function
test_ann_head
()
tests/test_models/test_heads.py:541
Function
test_apc_head
()
tests/test_models/test_heads.py:226
Function
test_aspp_head
()
tests/test_models/test_heads.py:332
← previous
next →
501–600 of 711, ranked by callers