Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ashutosh1807/PixelFormer
/ types & classes
Types & classes
28 in github.com/ashutosh1807/PixelFormer
⨍
Functions
155
◇
Types & classes
28
↓ 5 callers
Class
NewDataLoader
pixelformer/dataloaders/dataloader_kittipred.py:28
↓ 5 callers
Class
PixelFormer
pixelformer/networks/PixelFormer.py:41
↓ 4 callers
Class
SAM
pixelformer/networks/SAM.py:247
↓ 3 callers
Class
DataLoadPreprocess
pixelformer/dataloaders/dataloader.py:65
↓ 3 callers
Class
DataLoadPreprocess
pixelformer/dataloaders/dataloader_kittipred.py:64
↓ 2 callers
Class
DistributedSamplerNoEvenlyDivisible
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 callers
Class
BCP
Multilayer perceptron.
pixelformer/networks/PixelFormer.py:10
↓ 1 callers
Class
BasicLayer
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 callers
Class
DispHead
pixelformer/networks/PixelFormer.py:151
↓ 1 callers
Class
GLWidget
pixelformer/demo.py:313
↓ 1 callers
Class
Mlp
Multilayer perceptron.
pixelformer/networks/SAM.py:9
↓ 1 callers
Class
Mlp
Multilayer perceptron.
pixelformer/networks/swin_transformer.py:11
↓ 1 callers
Class
PPM
Pooling Pyramid Module used in PSPNet. Args: pool_scales (tuple[int]): Pooling scales used in Pooling Pyramid Module.
pixelformer/networks/PQI.py:9
↓ 1 callers
Class
PSP
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 callers
Class
PatchEmbed
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 callers
Class
SAMBLOCK
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 callers
Class
SwinTransformer
Swin Transformer backbone. A PyTorch impl of : `Swin Transformer: Hierarchical Vision Transformer using Shifted Windows` - https:/
pixelformer/networks/swin_transformer.py:439
↓ 1 callers
Class
SwinTransformerBlock
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 callers
Class
ToTensor
pixelformer/dataloaders/dataloader.py:256
↓ 1 callers
Class
ToTensor
pixelformer/dataloaders/dataloader_kittipred.py:222
↓ 1 callers
Class
Window
pixelformer/demo.py:119
↓ 1 callers
Class
WindowAttention
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 callers
Class
WindowAttention
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 callers
Class
silog_loss
pixelformer/utils.py:102
Class
BaseDecodeHead
Base class for BaseDecodeHead. Args: in_channels (int|Sequence[int]): Input channels. channels (int): Channels after modules, bef
pixelformer/networks/PQI.py:61
Class
NewDataLoader
pixelformer/dataloaders/dataloader.py:29
Class
PatchMerging
Patch Merging Layer Args: dim (int): Number of input channels. norm_layer (nn.Module, optional): Normalization layer. Default:
pixelformer/networks/swin_transformer.py:249
Class
UPerHead
pixelformer/networks/PQI.py:255