Functions — github.com/ShuweiShao/IEBins

hub / github.com/ShuweiShao/IEBins / functions

Functions322 in github.com/ShuweiShao/IEBins

⨍ Functions 322◇ Types & classes 72

↓ 1 callersFunctionload

(module, prefix='')

iebins_kittiofficial/networks/checkpoint.py:99

↓ 1 callersFunctionload

(module, prefix='')

iebins_kittiofficial/networks/newcrf_utils.py:99

↓ 1 callersFunctionload

(module, prefix='')

iebins/networks/newcrf_utils.py:99

↓ 1 callersFunctionload_checkpoint

Load checkpoint from a file or URI. Args: model (Module): Module to load checkpoint. filename (str): Accept local filepath, URL,

iebins/networks/newcrf_utils.py:194

↓ 1 callersFunctionload_checkpoint_swin

Load checkpoint from a file or URI. Args: model (Module): Module to load checkpoint. filename (str): Accept local filepath, URL,

iebins_kittiofficial/networks/checkpoint.py:318

↓ 1 callersFunctionload_fileclient_dist

In distributed setting, this function only download checkpoint at local rank 0.

iebins_kittiofficial/networks/checkpoint.py:183

↓ 1 callersFunctionload_pavimodel_dist

In distributed setting, this function only download checkpoint at local rank 0.

iebins_kittiofficial/networks/checkpoint.py:155

↓ 1 callersFunctionload_state_dict

Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F

iebins_kittiofficial/networks/checkpoint.py:73

↓ 1 callersFunctionload_state_dict

Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F

iebins_kittiofficial/networks/newcrf_utils.py:73

↓ 1 callersFunctionload_state_dict

Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F

iebins/networks/newcrf_utils.py:73

↓ 1 callersFunctionload_url_dist

In distributed setting, this function only download checkpoint at local rank 0.

iebins_kittiofficial/networks/newcrf_utils.py:141

↓ 1 callersFunctionload_url_dist

In distributed setting, this function only download checkpoint at local rank 0.

iebins/networks/newcrf_utils.py:141

↓ 1 callersFunctionmain

()

iebins_kittiofficial/eval.py:146

↓ 1 callersFunctionmain

()

iebins_kittiofficial/train.py:395

↓ 1 callersFunctionmain

()

iebins_kittiofficial/inference_single_image.py:105

↓ 1 callersFunctionmain

()

iebins/eval.py:165

↓ 1 callersFunctionmain

()

iebins/train.py:415

↓ 1 callersFunctionmain

()

iebins/inference_single_image.py:105

↓ 1 callersFunctionmain

()

iebins/eval_sun.py:167

↓ 1 callersFunctionmain_worker

(args)

iebins_kittiofficial/eval.py:109

↓ 1 callersFunctionmain_worker

(gpu, ngpus_per_node, args)

iebins_kittiofficial/train.py:154

↓ 1 callersFunctionmain_worker

(args)

iebins_kittiofficial/inference_single_image.py:72

↓ 1 callersFunctionmain_worker

(args)

iebins/eval.py:128

↓ 1 callersFunctionmain_worker

(gpu, ngpus_per_node, args)

iebins/train.py:173

↓ 1 callersFunctionmain_worker

(args)

iebins/inference_single_image.py:72

↓ 1 callersFunctionmain_worker

(args)

iebins/eval_sun.py:130

↓ 1 callersFunctiononline_eval

(model, dataloader_eval, gpu, epoch, ngpus, group, post_process=False)

iebins_kittiofficial/train.py:97

↓ 1 callersFunctiononline_eval

(model, dataloader_eval, gpu, epoch, ngpus, group, post_process=False)

iebins/train.py:98

↓ 1 callersMethodpsp_forward

Forward function of PSP module.

iebins_kittiofficial/networks/uper_crf_head.py:350

↓ 1 callersMethodpsp_forward

Forward function of PSP module.

iebins/networks/uper_crf_head.py:350

↓ 1 callersMethodrandom_crop

(self, img, depth, height, width)

iebins_kittiofficial/dataloaders/dataloader.py:196

↓ 1 callersMethodrandom_crop

(self, img, depth, height, width)

iebins/dataloaders/dataloader_sun.py:199

↓ 1 callersMethodrandom_crop

(self, img, depth, height, width)

iebins/dataloaders/dataloader.py:196

↓ 1 callersMethodset_epoch

(self, epoch)

iebins_kittiofficial/utils.py:376

↓ 1 callersMethodset_epoch

(self, epoch)

iebins/utils.py:326

↓ 1 callersFunctiontest

Test function.

iebins_kittiofficial/test.py:58

↓ 1 callersFunctiontest

Test function.

iebins/test.py:64

↓ 1 callersMethodtrain_preprocess

(self, image, depth_gt)

iebins_kittiofficial/dataloaders/dataloader.py:207

↓ 1 callersMethodtrain_preprocess

(self, image, depth_gt)

iebins/dataloaders/dataloader_sun.py:210

↓ 1 callersMethodtrain_preprocess

(self, image, depth_gt)

iebins/dataloaders/dataloader.py:207

↓ 1 callersFunctionupdate_sample

(bin_edges, target_bin_left, target_bin_right, depth_r, pred_label, depth_num, min_depth, max_depth, uncertain

iebins_kittiofficial/networks/depth_update.py:5

↓ 1 callersFunctionupdate_sample

(bin_edges, target_bin_left, target_bin_right, depth_r, pred_label, depth_num, min_depth, max_depth, uncertain

iebins/networks/depth_update.py:5

↓ 1 callersFunctionweights_to_cpu

Copy a model state_dict to cpu. Args: state_dict (OrderedDict): Model weights on GPU. Returns: OrderedDict: Model weights on

iebins_kittiofficial/networks/checkpoint.py:467

↓ 1 callersFunctionwindow_reverse

Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image

iebins_kittiofficial/networks/newcrf_layers.py:45

↓ 1 callersFunctionwindow_reverse

Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image

iebins/networks/swin_transformer.py:47

↓ 1 callersFunctionwindow_reverse

Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image

iebins/networks/newcrf_layers.py:45

Method__call__

(self, sample)

iebins_kittiofficial/dataloaders/dataloader.py:278

Method__call__

(self, sample)

iebins/dataloaders/dataloader_sun.py:281

Method__call__

(self, sample_dataset)

iebins/dataloaders/dataloader.py:278

Method__getitem__

(self, idx)

iebins_kittiofficial/dataloaders/dataloader.py:80

Method__getitem__

(self, idx)

iebins/dataloaders/dataloader_sun.py:81

Method__getitem__

(self, idx)

iebins/dataloaders/dataloader.py:80

Method__init__

(self, batch_size, height, width)

iebins_kittiofficial/utils.py:152

Method__init__

(self, dataset, num_replicas=None, rank=None, shuffle=True)

iebins_kittiofficial/utils.py:331

Method__init__

(self, size=None, scale_factor=None, mode='nearest',

iebins_kittiofficial/networks/resize.py:32

Method__init__

(self, in_channels, channels, *, num_class

iebins_kittiofficial/networks/uper_crf_head.py:93

Method__init__

(self, pool_scales=(1, 2, 3, 6), **kwargs)

iebins_kittiofficial/networks/uper_crf_head.py:256

Method__init__

(self, pool_scales=(1, 2, 3, 6), **kwargs)

iebins_kittiofficial/networks/uper_crf_head.py:329

Method__init__

(self, dim, window_size, num_heads, v_dim, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)

iebins_kittiofficial/networks/newcrf_layers.py:76

Method__init__

(self, dim, num_heads, v_dim, window_size=7, shift_size=0, mlp_ratio=4., qkv_bias=True, qk_sc

iebins_kittiofficial/networks/newcrf_layers.py:170

Method__init__

(self, dim, depth, num_heads, v_dim,

iebins_kittiofficial/networks/newcrf_layers.py:279

Method__init__

(self, input_dim=96, embed_dim=96, v_dim=64,

iebins_kittiofficial/networks/newcrf_layers.py:367

Method__init__

(self, version=None, inv_depth=False, pretrained=None, frozen_stages=-1, min_depth=0.1, m

iebins_kittiofficial/networks/NewCRFDepth.py:16

Method__init__

(self, input_dim=100)

iebins_kittiofficial/networks/NewCRFDepth.py:155

Method__init__

(self, hidden_dim=128, context_dim=192)

iebins_kittiofficial/networks/NewCRFDepth.py:170

Method__init__

(self, hidden_dim=128, input_dim=128+192)

iebins_kittiofficial/networks/NewCRFDepth.py:234

Method__init__

(self, hidden_dim, out_chs)

iebins_kittiofficial/networks/NewCRFDepth.py:263

Method__init__

(self, in_chs, out_chs)

iebins_kittiofficial/networks/NewCRFDepth.py:280

Method__init__

(self, normalized_shape, norm_layer=None)

iebins_kittiofficial/networks/swin_transformer_v2.py:27

Method__init__

(self, normalized_shape: _shape_t, eps: float = 1e-5, elementwise_affine: bool = True)

iebins_kittiofficial/networks/swin_transformer_v2.py:42

Method__init__

(self, in_features, out_features, bias=True)

iebins_kittiofficial/networks/swin_transformer_v2.py:51

Method__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0., norm

iebins_kittiofficial/networks/swin_transformer_v2.py:93

Method__init__

(self, dim, window_size, num_heads, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.,

iebins_kittiofficial/networks/swin_transformer_v2.py:163

Method__init__

(self, dim, num_heads, window_size=7, shift_size=0, mlp_ratio=4., qkv_bias=True, qk_scale=Non

iebins_kittiofficial/networks/swin_transformer_v2.py:372

Method__init__

(self, dim, num_heads, window_size=7, shift_size=0, mlp_ratio=4., qkv_bias=True, qk_scale=Non

iebins_kittiofficial/networks/swin_transformer_v2.py:508

Method__init__

(self, dim, norm_layer=nn.LayerNorm, postnorm=True)

iebins_kittiofficial/networks/swin_transformer_v2.py:640

Method__init__

(self, dim, norm_layer=nn.LayerNorm, postnorm=True)

iebins_kittiofficial/networks/swin_transformer_v2.py:689

Method__init__

(self, dim, norm_layer=nn.LayerNorm, postnorm=True)

iebins_kittiofficial/networks/swin_transformer_v2.py:719

Method__init__

(self, dim, depth, num_heads, window_size=

iebins_kittiofficial/networks/swin_transformer_v2.py:769

Method__init__

(self, patch_size=4, in_chans=3, embed_dim=96, norm_layer=None)

iebins_kittiofficial/networks/swin_transformer_v2.py:927

Method__init__

(self, in_chans=3, embed_dim=96, norm_layer=None)

iebins_kittiofficial/networks/swin_transformer_v2.py:961

Method__init__

(self, pretrain_img_size=224, patch_size=4, in_chans=3,

iebins_kittiofficial/networks/swin_transformer_v2.py:1024

Method__init__

(self, args, mode)

iebins_kittiofficial/dataloaders/dataloader.py:30

Method__init__

(self, args, mode, transform=None, is_for_online_eval=False)

iebins_kittiofficial/dataloaders/dataloader.py:66

Method__init__

(self, mode)

iebins_kittiofficial/dataloaders/dataloader.py:274

Method__init__

(self, dataset, num_replicas=None, rank=None, shuffle=True)

iebins/utils.py:281

Method__init__

(self, batch_size, height, width)

iebins/utils.py:333

Method__init__

(self)

iebins/sum_depth.py:6

Method__init__

(self, size=None, scale_factor=None, mode='nearest',

iebins/networks/resize.py:32

Method__init__

(self, in_channels, channels, *, num_class

iebins/networks/uper_crf_head.py:93

Method__init__

(self, pool_scales=(1, 2, 3, 6), **kwargs)

iebins/networks/uper_crf_head.py:256

Method__init__

(self, pool_scales=(1, 2, 3, 6), **kwargs)

iebins/networks/uper_crf_head.py:329

Method__init__

(self, dim, window_size, num_heads, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)

iebins/networks/swin_transformer.py:78

Method__init__

(self, dim, num_heads, window_size=7, shift_size=0, mlp_ratio=4., qkv_bias=True, qk_scale=Non

iebins/networks/swin_transformer.py:165

Method__init__

(self, dim, norm_layer=nn.LayerNorm)

iebins/networks/swin_transformer.py:256

Method__init__

(self, dim, depth, num_heads, window_size=

iebins/networks/swin_transformer.py:311

Method__init__

(self, patch_size=4, in_chans=3, embed_dim=96, norm_layer=None)

iebins/networks/swin_transformer.py:406

Method__init__

(self, pretrain_img_size=224, patch_size=4, in_chans=3,

iebins/networks/swin_transformer.py:468

Method__init__

(self, dim, window_size, num_heads, v_dim, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)

iebins/networks/newcrf_layers.py:76

Method__init__

(self, dim, num_heads, v_dim, window_size=7, shift_size=0, mlp_ratio=4., qkv_bias=True, qk_sc

iebins/networks/newcrf_layers.py:170

← previous next →101–200 of 322, ranked by callers