MCPcopy Create free account

hub / github.com/NVlabs/SegFormer / functions

Functions711 in github.com/NVlabs/SegFormer

↓ 1 callersMethodlosses
Compute segmentation loss.
mmseg/models/decode_heads/decode_head.py:217
↓ 1 callersMethodlosses
Compute segmentation loss.
mmseg/models/decode_heads/point_head.py:248
↓ 1 callersMethodlosses
Compute ``pam_cam``, ``pam``, ``cam`` loss.
mmseg/models/decode_heads/da_head.py:164
↓ 1 callersFunctionmain
()
tools/train.py:63
↓ 1 callersFunctionmain
()
tools/convert_model.py:25
↓ 1 callersFunctionmain
()
tools/test.py:66
↓ 1 callersFunctionmain
()
tools/benchmark.py:23
↓ 1 callersFunctionmain
()
tools/get_flops.py:76
↓ 1 callersFunctionmain
()
tools/publish_model.py:29
↓ 1 callersFunctionmain
()
tools/print_config.py:16
↓ 1 callersFunctionmain
()
tools/convert_datasets/pascal_context.py:47
↓ 1 callersFunctionmain
()
tools/convert_datasets/drive.py:24
↓ 1 callersFunctionmain
()
tools/convert_datasets/voc_aug.py:36
↓ 1 callersFunctionmain
()
tools/convert_datasets/hrf.py:34
↓ 1 callersFunctionmain
()
tools/convert_datasets/chase_db1.py:23
↓ 1 callersFunctionmain
()
tools/convert_datasets/stare.py:33
↓ 1 callersFunctionmain
()
tools/convert_datasets/cityscapes.py:25
↓ 1 callersFunctionmain
()
demo/image_demo.py:7
↓ 1 callersMethodmake_layer
Stack InvertedResidual blocks to build a layer for MobileNetV2. Args: out_channels (int): out_channels of block. num_
mmseg/models/backbones/mobilenet_v2.py:107
↓ 1 callersMethodmake_res_layer
Pack all blocks in a stage into a ``ResLayer``.
mmseg/models/backbones/resnet.py:523
↓ 1 callersMethodmake_stage_plugins
make plugins for ResNet 'stage_idx'th stage . Currently we support to insert 'context_block', 'empirical_attention_block', 'nonlocal_
mmseg/models/backbones/resnet.py:470
↓ 1 callersMethodnorm0
nn.Module: the normalization layer named "norm0"
mmseg/models/backbones/resnest.py:108
↓ 1 callersMethodnorm1
nn.Module: the normalization layer named "norm1"
mmseg/models/backbones/resnest.py:113
↓ 1 callersMethodnorm1
nn.Module: normalization layer after the first convolution layer
mmseg/models/backbones/resnet.py:58
↓ 1 callersMethodnorm1
nn.Module: the normalization layer named "norm1"
mmseg/models/backbones/resnet.py:528
↓ 1 callersMethodnorm2
nn.Module: normalization layer after the second convolution layer
mmseg/models/backbones/resnet.py:63
↓ 1 callersMethodnorm2
nn.Module: normalization layer after the second convolution layer
mmseg/models/backbones/resnet.py:255
↓ 1 callersMethodpam_cls_seg
PAM feature classification.
mmseg/models/decode_heads/da_head.py:128
↓ 1 callersFunctionparse_args
()
tools/train.py:19
↓ 1 callersFunctionparse_args
()
tools/convert_model.py:15
↓ 1 callersFunctionparse_args
()
tools/test.py:15
↓ 1 callersFunctionparse_args
()
tools/benchmark.py:13
↓ 1 callersFunctionparse_args
()
tools/pytorch2onnx.py:148
↓ 1 callersFunctionparse_args
()
tools/get_flops.py:10
↓ 1 callersFunctionparse_args
()
tools/publish_model.py:7
↓ 1 callersFunctionparse_args
()
tools/print_config.py:6
↓ 1 callersFunctionparse_args
()
tools/convert_datasets/pascal_context.py:37
↓ 1 callersFunctionparse_args
()
tools/convert_datasets/drive.py:11
↓ 1 callersFunctionparse_args
()
tools/convert_datasets/voc_aug.py:24
↓ 1 callersFunctionparse_args
()
tools/convert_datasets/hrf.py:13
↓ 1 callersFunctionparse_args
()
tools/convert_datasets/chase_db1.py:13
↓ 1 callersFunctionparse_args
()
tools/convert_datasets/stare.py:21
↓ 1 callersFunctionparse_args
()
tools/convert_datasets/cityscapes.py:13
↓ 1 callersFunctionparse_line
Parse information from a line in a requirements text file.
setup.py:38
↓ 1 callersFunctionparse_version_info
(version_str)
mmseg/version.py:6
↓ 1 callersMethodprepare_test_img
Get testing data after pipeline. Args: idx (int): Index of data. Returns: dict: Testing data after pipeline
mmseg/datasets/custom.py:210
↓ 1 callersMethodprepare_train_img
Get training data and annotations after pipeline. Args: idx (int): Index of data. Returns: dict: Training da
mmseg/datasets/custom.py:193
↓ 1 callersFunctionprocess_checkpoint
(in_file, out_file)
tools/publish_model.py:16
↓ 1 callersMethodpsp_forward
Forward function of PSP module.
mmseg/models/decode_heads/uper_head.py:76
↓ 1 callersFunctionpytorch2onnx
Export Pytorch model to ONNX model and verify the outputs are same between Pytorch and ONNX. Args: model (nn.Module): Pytorch model w
tools/pytorch2onnx.py:70
↓ 1 callersMethodrandom_sample
Randomly sample an img_scale when ``multiscale_mode=='range'``. Args: img_scales (list[tuple]): Images scale range for sampling.
mmseg/datasets/pipelines/transforms.py:64
↓ 1 callersMethodrandom_select
Randomly select an img_scale from given candidates. Args: img_scales (list[tuple]): Images scales for selection. Returns
mmseg/datasets/pipelines/transforms.py:46
↓ 1 callersMethodrandom_select
Randomly select an img_scale from given candidates. Args: img_scales (list[tuple]): Images scales for selection. Returns
mmseg/datasets/pipelines/transforms.py:295
↓ 1 callersFunctionreadme
()
setup.py:4
↓ 1 callersFunctionreduce_mean
Reduce mean when distributed training.
mmseg/models/decode_heads/ema_head.py:13
↓ 1 callersMethodresults2img
Write the segmentation results to images. Args: results (list[list | tuple | ndarray]): Testing results of the da
mmseg/datasets/cityscapes.py:50
↓ 1 callersMethodsaturation
Saturation distortion.
mmseg/datasets/pipelines/transforms.py:1153
↓ 1 callersMethodscaled_l2
(x, codewords, scale)
mmseg/ops/encoding.py:33
↓ 1 callersFunctionset_random_seed
Set random seed. Args: seed (int): Seed to be used. deterministic (bool): Whether to set the deterministic option for
mmseg/apis/train.py:14
↓ 1 callersFunctionsetup
(app)
docs/conf.py:87
↓ 1 callersFunctionshow_result_pyplot
Visualize the segmentation results on the image. Args: model (nn.Module): The loaded segmentor. img (str or np.ndarray): Image fi
mmseg/apis/inference.py:101
↓ 1 callersMethodsimple_test
Placeholder for single image test.
mmseg/models/segmentors/base.py:56
↓ 1 callersMethodslide_inference
Inference by sliding-window with overlap. If h_crop > h_img or w_crop > w_img, the small patch will be used to decode without padding
mmseg/models/segmentors/encoder_decoder.py:169
↓ 1 callersFunctiontotal_intersect_and_union
Calculate Total Intersection and Union. Args: results (list[ndarray]): List of prediction segmentation maps. gt_seg_maps (list[nd
mmseg/core/evaluation/metrics.py:62
↓ 1 callersFunctiontrain_segmentor
Launch segmentor training.
mmseg/apis/train.py:32
↓ 1 callersMethodwhole_inference
Inference with full image.
mmseg/models/segmentors/encoder_decoder.py:214
Method__call__
Call function to load images into results. Args: results (dict): A result dict contains the file name of the imag
mmseg/apis/inference.py:45
Method__call__
Call function to apply test time augment transforms on results. Args: results (dict): Result dict contains the data to transform.
mmseg/datasets/pipelines/test_time_aug.py:93
Method__call__
Call function to resize images, bounding boxes, masks, semantic segmentation map. Args: results (dict): Result dict from
mmseg/datasets/pipelines/transforms.py:209
Method__call__
Call function to resize images, bounding boxes, masks, semantic segmentation map. Args: results (dict): Result dict from
mmseg/datasets/pipelines/transforms.py:437
Method__call__
Call function to flip bounding boxes, masks, semantic segmentation maps. Args: results (dict): Result dict from loading p
mmseg/datasets/pipelines/transforms.py:486
Method__call__
Call function to pad images, masks, semantic segmentation maps. Args: results (dict): Result dict from loading pipeline.
mmseg/datasets/pipelines/transforms.py:569
Method__call__
Call function to normalize images. Args: results (dict): Result dict from loading pipeline. Returns: dict: N
mmseg/datasets/pipelines/transforms.py:608
Method__call__
Call function to rerange images. Args: results (dict): Result dict from loading pipeline. Returns: dict: Rera
mmseg/datasets/pipelines/transforms.py:650
Method__call__
Call function to Use CLAHE method process images. Args: results (dict): Result dict from loading pipeline. Returns:
mmseg/datasets/pipelines/transforms.py:699
Method__call__
Call function to randomly crop images, semantic segmentation maps. Args: results (dict): Result dict from loading pipeline.
mmseg/datasets/pipelines/transforms.py:756
Method__call__
Call function to randomly crop images, semantic segmentation maps. Args: results (dict): Result dict from loading pipeline.
mmseg/datasets/pipelines/transforms.py:824
Method__call__
Call function to rotate image, semantic segmentation maps. Args: results (dict): Result dict from loading pipeline. Retu
mmseg/datasets/pipelines/transforms.py:894
Method__call__
Call function to convert RGB image to grayscale image. Args: results (dict): Result dict from loading pipeline. Returns:
mmseg/datasets/pipelines/transforms.py:961
Method__call__
Call function to process the image with gamma correction. Args: results (dict): Result dict from loading pipeline. Retur
mmseg/datasets/pipelines/transforms.py:1009
Method__call__
Call function to process the image with gamma correction. Args: results (dict): Result dict from loading pipeline. Retur
mmseg/datasets/pipelines/transforms.py:1042
Method__call__
Call function to scale the semantic segmentation map. Args: results (dict): Result dict from loading pipeline. Returns:
mmseg/datasets/pipelines/transforms.py:1079
Method__call__
Call function to perform photometric distortion on images. Args: results (dict): Result dict from loading pipeline. Retu
mmseg/datasets/pipelines/transforms.py:1174
Method__call__
Call functions to load image and get image meta information. Args: results (dict): Result dict from :obj:`mmseg.CustomDataset`.
mmseg/datasets/pipelines/loading.py:42
Method__call__
Call function to load multiple types annotations. Args: results (dict): Result dict from :obj:`mmseg.CustomDataset`. Ret
mmseg/datasets/pipelines/loading.py:113
Method__call__
Call function to apply transforms sequentially. Args: data (dict): A result dict contains the data to transform. Returns
mmseg/datasets/pipelines/compose.py:29
Method__call__
Call function to convert data in results to :obj:`torch.Tensor`. Args: results (dict): Result dict contains the data to convert.
mmseg/datasets/pipelines/formating.py:47
Method__call__
Call function to convert image in results to :obj:`torch.Tensor` and transpose the channel order. Args: results (dict): R
mmseg/datasets/pipelines/formating.py:81
Method__call__
Call function to convert image in results to :obj:`torch.Tensor` and transpose the channel order. Args: results (dict): R
mmseg/datasets/pipelines/formating.py:117
Method__call__
Call function to convert data in results to :obj:`mmcv.DataContainer`. Args: results (dict): Result dict contains the dat
mmseg/datasets/pipelines/formating.py:155
Method__call__
Call function to transform and format common fields in results. Args: results (dict): Result dict contains the data to convert.
mmseg/datasets/pipelines/formating.py:189
Method__call__
Call function to collect keys in results. The keys in ``meta_keys`` will be converted to :obj:mmcv.DataContainer. Args: r
mmseg/datasets/pipelines/formating.py:264
Method__getitem__
Get training/test data after pipeline. Args: idx (int): Index of data. Returns: dict: Training/test data (wi
mmseg/datasets/custom.py:177
Method__getitem__
Get item from original dataset.
mmseg/datasets/dataset_wrappers.py:44
Method__getitem__
(self, idx)
tests/test_eval_hook.py:18
Method__init__
(self, channels, num_codes)
mmseg/ops/encoding.py:17
Method__init__
(self, size=None, scale_factor=None, mode='nearest',
mmseg/ops/wrappers.py:34
Method__init__
(self, context, thresh=None, min_kept=100000)
mmseg/core/seg/sampler/ohem_pixel_sampler.py:23
Method__init__
(self, **kwargs)
mmseg/core/seg/sampler/base_pixel_sampler.py:7
Method__init__
(self, dataloader, interval=1, by_epoch=False, **eval_kwargs)
mmseg/core/evaluation/eval_hooks.py:15
← previousnext →201–300 of 711, ranked by callers