MCPcopy Create free account

hub / github.com/ashutosh1807/PixelFormer / types & classes

Types & classes28 in github.com/ashutosh1807/PixelFormer

↓ 5 callersClassNewDataLoader
pixelformer/dataloaders/dataloader_kittipred.py:28
↓ 5 callersClassPixelFormer
pixelformer/networks/PixelFormer.py:41
↓ 4 callersClassSAM
pixelformer/networks/SAM.py:247
↓ 3 callersClassDataLoadPreprocess
pixelformer/dataloaders/dataloader.py:65
↓ 3 callersClassDataLoadPreprocess
pixelformer/dataloaders/dataloader_kittipred.py:64
↓ 2 callersClassDistributedSamplerNoEvenlyDivisible
Sampler that restricts data loading to a subset of the dataset. It is especially useful in conjunction with :class:`torch.nn.parallel.Distrib
pixelformer/utils.py:187
↓ 1 callersClassBCP
Multilayer perceptron.
pixelformer/networks/PixelFormer.py:10
↓ 1 callersClassBasicLayer
A basic Swin Transformer layer for one stage. Args: dim (int): Number of feature channels depth (int): Depths of this stage.
pixelformer/networks/swin_transformer.py:292
↓ 1 callersClassDispHead
pixelformer/networks/PixelFormer.py:151
↓ 1 callersClassGLWidget
pixelformer/demo.py:313
↓ 1 callersClassMlp
Multilayer perceptron.
pixelformer/networks/SAM.py:9
↓ 1 callersClassMlp
Multilayer perceptron.
pixelformer/networks/swin_transformer.py:11
↓ 1 callersClassPPM
Pooling Pyramid Module used in PSPNet. Args: pool_scales (tuple[int]): Pooling scales used in Pooling Pyramid Module.
pixelformer/networks/PQI.py:9
↓ 1 callersClassPSP
Unified Perceptual Parsing for Scene Understanding. This head is the implementation of `UPerNet <https://arxiv.org/abs/1807.10221>`_. Ar
pixelformer/networks/PQI.py:318
↓ 1 callersClassPatchEmbed
Image to Patch Embedding Args: patch_size (int): Patch token size. Default: 4. in_chans (int): Number of input image channels. D
pixelformer/networks/swin_transformer.py:396
↓ 1 callersClassSAMBLOCK
Args: dim (int): Number of feature channels num_heads (int): Number of attention head. window_size (int): Local window s
pixelformer/networks/SAM.py:146
↓ 1 callersClassSwinTransformer
Swin Transformer backbone. A PyTorch impl of : `Swin Transformer: Hierarchical Vision Transformer using Shifted Windows` - https:/
pixelformer/networks/swin_transformer.py:439
↓ 1 callersClassSwinTransformerBlock
Swin Transformer Block. Args: dim (int): Number of input channels. num_heads (int): Number of attention heads. window_si
pixelformer/networks/swin_transformer.py:147
↓ 1 callersClassToTensor
pixelformer/dataloaders/dataloader.py:256
↓ 1 callersClassToTensor
pixelformer/dataloaders/dataloader_kittipred.py:222
↓ 1 callersClassWindow
pixelformer/demo.py:119
↓ 1 callersClassWindowAttention
Window based multi-head self attention (W-MSA) module with relative position bias. It supports both of shifted and non-shifted window. Args:
pixelformer/networks/SAM.py:62
↓ 1 callersClassWindowAttention
Window based multi-head self attention (W-MSA) module with relative position bias. It supports both of shifted and non-shifted window. Args:
pixelformer/networks/swin_transformer.py:64
↓ 1 callersClasssilog_loss
pixelformer/utils.py:102
ClassBaseDecodeHead
Base class for BaseDecodeHead. Args: in_channels (int|Sequence[int]): Input channels. channels (int): Channels after modules, bef
pixelformer/networks/PQI.py:61
ClassNewDataLoader
pixelformer/dataloaders/dataloader.py:29
ClassPatchMerging
Patch Merging Layer Args: dim (int): Number of input channels. norm_layer (nn.Module, optional): Normalization layer. Default:
pixelformer/networks/swin_transformer.py:249
ClassUPerHead
pixelformer/networks/PQI.py:255