Functions — github.com/LiBingyu01/StitchFusion

hub / github.com/LiBingyu01/StitchFusion / functions

Functions314 in github.com/LiBingyu01/StitchFusion

⨍ Functions 314◇ Types & classes 106

Method__init__

Resize the input image to the given size.

semseg/augmentations_mm.py:316

Method__init__

(self, root: str = 'data/NYUDepthv2', split: str = 'train', transform = None, modals = ['img', 'depth'], case

semseg/datasets/nyu.py:27

Method__init__

(self, root: str = 'data/MCubeS/multimodal_dataset', split: str = 'train', transform = None, modals = ['image'

semseg/datasets/mcubes.py:47

Method__init__

(self, mean=(0., 0., 0.), std=(1., 1., 1.))

semseg/datasets/mcubes.py:161

Method__init__

(self, base_size, crop_size, fill=255)

semseg/datasets/mcubes.py:293

Method__init__

(self, crop_size)

semseg/datasets/mcubes.py:392

Method__init__

(self, root: str = 'data/MFNet', split: str = 'train', transform = None, modals = ['rgb', 'ther'], case = None

semseg/datasets/MFNet.py:22

Method__init__

(self, root: str = 'data/FMB', split: str = 'train', transform = None, modals = ['img', 'thermal'], case = Non

semseg/datasets/fmb.py:39

Method__init__

(self, root: str = 'data/UrBanLF/Syn', split: str = 'train', transform = None, modals = ['img', '5_1', '5_2',

semseg/datasets/urbanlf.py:24

Method__init__

(self, root: str = 'data/PST900', split: str = 'train', transform = None, modals = ['img', 'thermal'], case =

semseg/datasets/pst.py:28

Method__init__

(self, root: str = 'data/DELIVER', split: str = 'train', transform = None, modals = ['img'], case = None)

semseg/datasets/DELIVER.py:51

Method__init__

(self, root: str = 'data/KITTI360', split: str = 'train', transform = None, modals = ['img', 'depth', 'event',

semseg/datasets/kitti360.py:28

Method__init__

(self, backbone: str = 'MiT-B0', num_classes: int = 19, modals: list = ['rgb', 'depth', 'event', 'lidar'])

semseg/models/base.py:38

Method__init__

(self, backbone: str = 'stitchfusion-B0', num_classes: int = 20, modals: list = ['img', 'aolp', 'dolp', 'nir']

semseg/models/stitchfusion.py:8

Method__init__

(self, dim)

semseg/models/backbones/stitchfusion.py:21

Method__init__

(self, dim, kernel)

semseg/models/backbones/stitchfusion.py:33

Method__init__

(self, dim)

semseg/models/backbones/stitchfusion.py:48

Method__init__

(self, dim, xavier_init=False)

semseg/models/backbones/stitchfusion.py:63

Method__init__

(self, dim, head, sr_ratio)

semseg/models/backbones/stitchfusion.py:92

Method__init__

(self, c1=3, c2=32, patch_size=7, stride=4, padding=0)

semseg/models/backbones/stitchfusion.py:124

Method__init__

(self, channel, reduction=16)

semseg/models/backbones/stitchfusion.py:138

Method__init__

(self, c1, c2)

semseg/models/backbones/stitchfusion.py:162

Method__init__

(self, channels, num_modals)

semseg/models/backbones/stitchfusion.py:186

Method__init__

(self, channels, num_modals)

semseg/models/backbones/stitchfusion.py:223

Method__init__

(self, dim, num_heads=8, qkv_bias=False, attn_drop=0., drop_path=0., mlp_hidden_dim=None, norm_layer=nn.LayerN

semseg/models/backbones/stitchfusion.py:244

Method__init__

(self, dim, head, sr_ratio=1, dpr=0., num_modalities=2, fan=False)

semseg/models/backbones/stitchfusion.py:288

Method__init__

(self, dim, head, sr_ratio=1, dpr=0., num_modalities=2, fan=False)

semseg/models/backbones/stitchfusion.py:355

Method__init__

(self, dim, head, sr_ratio=1, dpr=0. , num_modalities=2, fan=False)

semseg/models/backbones/stitchfusion.py:411

Method__init__

(self, dim, head, sr_ratio=1, dpr=0. , num_modalities=2, fan=False)

semseg/models/backbones/stitchfusion.py:438

Method__init__

(self, model_name: str = 'B0', modals: list = ['rgb', 'depth', 'event', 'lidar'])

semseg/models/backbones/stitchfusion.py:485

Method__init__

(self, c1, c2, k, s=1, p=0, d=1, g=1)

semseg/models/layers/common.py:6

Method__init__

(self, c1, c2)

semseg/models/heads/segformer.py:19

Method__init__

(self, dims: list, embed_dim: int = 256, num_classes: int = 19)

semseg/models/heads/segformer.py:29

Method__init__

(self, cfg)

getting_start/infer_mm.py:24

Method__len__

(self)

semseg/datasets/nyu.py:40

Method__len__

(self)

semseg/datasets/mcubes.py:74

Method__len__

(self)

semseg/datasets/MFNet.py:36

Method__len__

(self)

semseg/datasets/fmb.py:56

Method__len__

(self)

semseg/datasets/urbanlf.py:38

Method__len__

(self)

semseg/datasets/pst.py:45

Method__len__

(self)

semseg/datasets/DELIVER.py:70

Method__len__

(self)

semseg/datasets/kitti360.py:49

Method_init_weights

(self, m: nn.Module)

semseg/models/base.py:45

Functioncount_parameters

(model: nn.Module)

semseg/utils/utils.py:52

Functionevaluate_msf

(model, dataloader, device, scales, flip)

getting_start/visulize_mm_label.py:153

Functionevaluate_msf

(model, dataloader, device, scales, flip)

getting_start/visulize_mm.py:126

Functionevaluate_msf

(model, dataloader, device, scales, flip)

getting_start/visulize_mm_tsne.py:194

Methodforward

(self, x: Tensor, H, W)

stitchfusion_with_tips_you_can_copy.py:16

Methodforward

(self, x: Tensor, H, W)

stitchfusion_with_tips_you_can_copy.py:25

Methodforward

(self, x: Tensor, H, W)

stitchfusion_with_tips_you_can_copy.py:40

Methodforward

(self, x: Tensor, H, W)

stitchfusion_with_tips_you_can_copy.py:56

Methodforward

(self, x)

stitchfusion_with_tips_you_can_copy.py:79

Methodforward

(self, x: Tensor, H, W)

stitchfusion_with_tips_you_can_copy.py:105

Methodforward

(self, x: Tensor)

stitchfusion_with_tips_you_can_copy.py:129

Methodforward

(self, x, H, W)

stitchfusion_with_tips_you_can_copy.py:153

Methodforward

(self, x: Tensor, H, W)

stitchfusion_with_tips_you_can_copy.py:176

Methodforward

(self, x, layer_idx)

stitchfusion_with_tips_you_can_copy.py:212

Methodforward

(self, x, layer_idx)

stitchfusion_with_tips_you_can_copy.py:235

Methodforward

(self, x, H, W, atten=None)

stitchfusion_with_tips_you_can_copy.py:258

Methodforward

(self, inputs: list, H, W)

stitchfusion_with_tips_you_can_copy.py:300

Methodforward

(self, inputs, H, W)

stitchfusion_with_tips_you_can_copy.py:352

Methodforward

(self, inputs, H, W)

stitchfusion_with_tips_you_can_copy.py:417

Methodforward

(self, inputs: list, H, W)

stitchfusion_with_tips_you_can_copy.py:463

Methodforward

(self, x: list)

stitchfusion_with_tips_you_can_copy.py:569

Methodforward

(self, preds, labels: Tensor)

semseg/losses.py:16

Methodforward

(self, preds, labels: Tensor)

semseg/losses.py:41

Methodforward

(self, preds, targets: Tensor)

semseg/losses.py:70

Methodforward

(self, label: torch.Tensor, color_fea: torch.Tensor, other_fea: torch.Tensor)

semseg/losses.py:111

Methodforward

(self, x: list)

semseg/models/stitchfusion.py:13

Methodforward

(self, x: Tensor, H, W)

semseg/models/backbones/stitchfusion.py:16

Methodforward

(self, x: Tensor, H, W)

semseg/models/backbones/stitchfusion.py:25

Methodforward

(self, x: Tensor, H, W)

semseg/models/backbones/stitchfusion.py:40

Methodforward

(self, x: Tensor, H, W)

semseg/models/backbones/stitchfusion.py:56

Methodforward

(self, x)

semseg/models/backbones/stitchfusion.py:79

Methodforward

(self, x: Tensor, H, W)

semseg/models/backbones/stitchfusion.py:105

Methodforward

(self, x: Tensor)

semseg/models/backbones/stitchfusion.py:129

Methodforward

(self, x, H, W)

semseg/models/backbones/stitchfusion.py:153

Methodforward

(self, x: Tensor, H, W)

semseg/models/backbones/stitchfusion.py:176

Methodforward

(self, x, layer_idx)

semseg/models/backbones/stitchfusion.py:212

Methodforward

(self, x, layer_idx)

semseg/models/backbones/stitchfusion.py:235

Methodforward

(self, x, H, W, atten=None)

semseg/models/backbones/stitchfusion.py:258

Methodforward

(self, inputs, H, W)

semseg/models/backbones/stitchfusion.py:305

Methodforward

(self, inputs, H, W)

semseg/models/backbones/stitchfusion.py:373

Methodforward

(self, inputs: list, H, W)

semseg/models/backbones/stitchfusion.py:419

Methodforward

(self, inputs: list, H, W)

semseg/models/backbones/stitchfusion.py:449

Methodforward

(self, x: list)

semseg/models/backbones/stitchfusion.py:543

Methodforward

(self, x: Tensor)

semseg/models/layers/common.py:27

Methodforward

(self, x: Tensor)

semseg/models/heads/segformer.py:12

Methodforward

(self, x: Tensor)

semseg/models/heads/segformer.py:25

Methodforward

(self, features: Tuple[Tensor, Tensor, Tensor, Tensor])

semseg/models/heads/segformer.py:40

Functiongenerate_palette

(num_classes, background: bool = False)

semseg/utils/visualize.py:67

Methodget_lr

(self)

semseg/schedulers.py:28

Methodget_main_ratio

(self)

semseg/schedulers.py:51

Methodget_main_ratio

(self)

semseg/schedulers.py:65

Methodget_main_ratio

(self)

semseg/schedulers.py:76

Functionget_model_size

(model: Union[nn.Module, torch.jit.ScriptModule])

semseg/utils/utils.py:36

Functionget_train_augmentation

(size: Union[int, Tuple[int], List[int]], seg_fill: int = 0)

semseg/augmentations.py:334

Functionget_val_augmentation

(size: Union[int, Tuple[int], List[int]])

semseg/augmentations.py:348

Functionnchw_to_nlc

Flatten [N, C, H, W] shape tensor to [N, L, C] shape tensor. Args: x (Tensor): The input tensor of shape [N, C, H, W] before conversion.

semseg/utils/utils.py:178

Functionnlc2nchw2nlc

Convert [N, L, C] shape tensor `x` to [N, C, H, W] shape tensor. Use the reshaped tensor as the input of `module`, and convert the output of `

semseg/utils/utils.py:206

← previous next →201–300 of 314, ranked by callers