MCPcopy Create free account

hub / github.com/XiandaGuo/OpenStereo / functions

Functions2,487 in github.com/XiandaGuo/OpenStereo

Method__getitem__
(self, idx)
stereo/datasets/instereo2k_dataset.py:19
Method__getitem__
(self, idx)
stereo/datasets/kitti_dataset.py:19
Method__getitem__
(self, idx)
stereo/datasets/sintel_dataset.py:18
Method__getitem__
(self, idx)
stereo/datasets/carla_dataset.py:17
Method__getitem__
(self, idx)
stereo/datasets/sceneflow_dataset.py:26
Method__getitem__
(self, idx)
stereo/datasets/sceneflow_dataset.py:83
Method__getitem__
(self, idx)
stereo/datasets/dynamic_replica.py:19
Method__getitem__
(self, idx)
stereo/datasets/vkitti2_dataset.py:19
Method__getitem__
(self, idx)
stereo/datasets/foundationstereo.py:50
Method__getitem__
(self, idx)
stereo/datasets/eth3d_dataset.py:16
Method__getitem__
(self, idx)
stereo/datasets/tartanair_dataset.py:17
Method__getitem__
(self, idx)
stereo/datasets/mono_dataset.py:19
Method__getitem__
(self, idx)
stereo/datasets/fallingthings_dataset.py:19
Method__getitem__
(self, idx)
stereo/datasets/realfill_dataset.py:19
Method__getitem__
(self, idx)
stereo/datasets/unrealstereo4k_dataset.py:18
Method__getitem__
(self, idx)
stereo/datasets/driving_dataset.py:18
Method__getitem__
(self, idx)
stereo/datasets/spring.py:20
Method__getitem__
(self, idx)
stereo/datasets/middlebury_dataset.py:17
Method__init__
(self, data_info, data_cfg, mode='testing')
tools/test_kitti.py:45
Method__init__
(self, t=0.0)
deploy/deploy_utils.py:201
Method__init__
(self, args, cfgs, local_rank, global_rank, logger, tb_writer, model)
stereo/modeling/trainer_template.py:22
Method__init__
(self)
stereo/modeling/backbones/psmnet_backbone.py:34
Method__init__
(self, resblock=True, input_channel=3, encoder_ratio=16, decoder_ratio=16)
stereo/modeling/backbones/fadnet.py:40
Method__init__
(self, num_features=8)
stereo/modeling/cost_volume/cost_volume.py:121
Method__init__
(self)
stereo/modeling/disp_refinement/disp_refinement.py:33
Method__init__
(self)
stereo/modeling/disp_refinement/disp_refinement.py:140
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
stereo/modeling/disp_refinement/gru_blocks.py:9
Method__init__
(self, output_dim=None, norm_fn='batch', downsample=3)
stereo/modeling/disp_refinement/gru_blocks.py:63
Method__init__
(self, init_fmap1, init_fmap2, geo_volume, num_levels=2, radius=4)
stereo/modeling/disp_refinement/gru_blocks.py:171
Method__init__
(self, corr_levels, corr_radius, volume_channel)
stereo/modeling/disp_refinement/gru_blocks.py:234
Method__init__
(self, input_dim=128, hidden_dim=256, output_dim=1)
stereo/modeling/disp_refinement/gru_blocks.py:272
Method__init__
(self, n_gru_layers, corr_levels, corr_radius, volume_channel, hidden_dims=None)
stereo/modeling/disp_refinement/gru_blocks.py:291
Method__init__
(self, in_channels, backbone_channels=None)
stereo/modeling/cost_aggregation/hourglass.py:8
Method__init__
(self)
stereo/modeling/common/loss.py:7
Method__init__
(self)
stereo/modeling/common/loss.py:60
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, bias=False, norm_layer=No
stereo/modeling/common/basic_block_3d.py:24
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, bias=False, norm_layer=No
stereo/modeling/common/basic_block_2d.py:25
Method__init__
(self, inp, oup, stride, expanse_ratio, dilation=1)
stereo/modeling/models/lightstereo/aggregation.py:66
Method__init__
(self, dim, img_feat_dim)
stereo/modeling/models/lightstereo/aggregation.py:105
Method__init__
(self, cfgs)
stereo/modeling/models/lightstereo/lightstereo.py:13
Method__init__
(self, backbone='MobileNetv2')
stereo/modeling/models/lightstereo/backbone.py:30
Method__init__
(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/lightstereo/trainer.py:12
Method__init__
(self, args)
stereo/modeling/models/sttr/sttr.py:27
Method__init__
(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/sttr/trainer.py:11
Method__init__
(self, threshold: int = 3, validation_max_disp: int = -1, loss_weight: list = None)
stereo/modeling/models/sttr/utilities/loss.py:18
Method__init__
(self, left, right, disp=None, sampled_cols=None, sampled_rows=None, occ_mask=None, occ_mask_
stereo/modeling/models/sttr/utilities/misc.py:13
Method__init__
(self)
stereo/modeling/models/sttr/utilities/feat_extractor_backbone_in.py:21
Method__init__
(self, *args, **kwargs)
stereo/modeling/models/sttr/utilities/__init__.py:8
Method__init__
(self)
stereo/modeling/models/sttr/utilities/feat_extractor_backbone.py:21
Method__init__
(self, num_blocks=8, feature_dim=16, expansion=3)
stereo/modeling/models/sttr/utilities/context_adjustment_layer.py:15
Method__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None)
stereo/modeling/models/sttr/utilities/pos_encoder.py:18
Method__init__
(self, cal: nn.Module, ot: bool = True)
stereo/modeling/models/sttr/utilities/regression_head.py:18
Method__init__
(self, embed_dim, num_heads)
stereo/modeling/models/sttr/utilities/attention.py:15
Method__init__
(self, hidden_dim: int, nhead: int)
stereo/modeling/models/sttr/utilities/transformer.py:121
Method__init__
(self, hidden_dim: int, nhead: int)
stereo/modeling/models/sttr/utilities/transformer.py:155
Method__init__
(self, num_layers, num_input_features, bn_size, growth_rate, drop_rate, memory_efficient=False)
stereo/modeling/models/sttr/utilities/densenet_in.py:97
Method__init__
(self, num_input_features, num_output_features)
stereo/modeling/models/sttr/utilities/densenet_in.py:118
Method__init__
(self, in_channels: int, out_channels: int, scale: int = 2)
stereo/modeling/models/sttr/utilities/feat_extractor_tokenizer.py:18
Method__init__
(self, in_channels: int, out_channels: int)
stereo/modeling/models/sttr/utilities/feat_extractor_tokenizer.py:47
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
stereo/modeling/models/igev_rt/extractor.py:12
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
stereo/modeling/models/igev_rt/extractor.py:70
Method__init__
(self, output_dim=128, norm_fn='batch', dropout=0.0, downsample=3)
stereo/modeling/models/igev_rt/extractor.py:128
Method__init__
(self, output_dim=[128], norm_fn='batch', dropout=0.0, downsample=3)
stereo/modeling/models/igev_rt/extractor.py:202
Method__init__
(self)
stereo/modeling/models/igev_rt/extractor.py:309
Method__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, bn=True, relu=True, **kwargs)
stereo/modeling/models/igev_rt/submodule.py:11
Method__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, IN=True, relu=True, **kwargs)
stereo/modeling/models/igev_rt/submodule.py:82
Method__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, concat=True, keep_concat=True, IN=True, relu=True
stereo/modeling/models/igev_rt/submodule.py:111
Method__init__
(self, cv_chan, feat_chan)
stereo/modeling/models/igev_rt/submodule.py:227
Method__init__
(self, dims, mode='sintel', divis_by=8)
stereo/modeling/models/igev_rt/utils.py:9
Method__init__
(self, geo_volume, num_levels=2, radius=4)
stereo/modeling/models/igev_rt/geometry.py:7
Method__init__
(self, in_channels)
stereo/modeling/models/igev_rt/igev_rt_stereo.py:22
Method__init__
(self, args)
stereo/modeling/models/igev_rt/igev_rt_stereo.py:90
Method__init__
(self, input_dim=128, hidden_dim=256, output_dim=2)
stereo/modeling/models/igev_rt/update.py:6
Method__init__
(self, input_dim=128, hidden_dim=128, output_dim=1)
stereo/modeling/models/igev_rt/update.py:16
Method__init__
(self, hidden_dim=128, input_dim=192+128)
stereo/modeling/models/igev_rt/update.py:42
Method__init__
(self, args)
stereo/modeling/models/igev_rt/update.py:72
Method__init__
(self, args, hidden_dim=96)
stereo/modeling/models/igev_rt/update.py:116
Method__init__
(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/igev_rt/trainer.py:16
Method__init__
(self, cfgs)
stereo/modeling/models/stereobase/stereobase_gru.py:15
Method__init__
(self, cv_chan, feat_chan)
stereo/modeling/models/stereobase/igev_blocks.py:36
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
stereo/modeling/models/stereobase/gru_blocks.py:9
Method__init__
(self, output_dim=None, norm_fn='batch', downsample=3)
stereo/modeling/models/stereobase/gru_blocks.py:63
Method__init__
(self, init_fmap1, init_fmap2, geo_volume, num_levels=2, radius=4)
stereo/modeling/models/stereobase/gru_blocks.py:171
Method__init__
(self, corr_levels, corr_radius, volume_channel)
stereo/modeling/models/stereobase/gru_blocks.py:234
Method__init__
(self, input_dim=128, hidden_dim=256, output_dim=1)
stereo/modeling/models/stereobase/gru_blocks.py:272
Method__init__
(self, n_gru_layers, corr_levels, corr_radius, volume_channel, hidden_dims=None)
stereo/modeling/models/stereobase/gru_blocks.py:291
Method__init__
(self)
stereo/modeling/models/stereobase/backbone.py:13
Method__init__
(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/stereobase/trainer.py:12
Method__init__
(self, in_channels, backbone_channels=None)
stereo/modeling/models/stereobase/hourglass.py:8
Method__init__
(self, cfgs)
stereo/modeling/models/coex/coex.py:12
Method__init__
(self, max_disparity, in_chan, hidden_chan, head=1, weighted=False)
stereo/modeling/models/coex/coex_cost_processor.py:39
Method__init__
(self, cv_chan, im_chan, D)
stereo/modeling/models/coex/coex_cost_processor.py:69
Method__init__
( self, max_disparity=192, matching_head=1, gce=True,
stereo/modeling/models/coex/coex_cost_processor.py:100
Method__init__
( self, max_disp=192, gce=True, matching_weighted=False,
stereo/modeling/models/coex/coex_cost_processor.py:241
Method__init__
(self, inplanes, planes, stride, downsample, pad, dilation)
stereo/modeling/models/coex/submodule.py:20
Method__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, bn=True, relu=True, **kwargs)
stereo/modeling/models/coex/submodule.py:45
Method__init__
( self, inplanes: int, planes: int, stride: int = 1,
stereo/modeling/models/coex/submodule.py:171
Method__init__
( self, in_planes: int, out_planes: int, kernel_size: int = 3,
stereo/modeling/models/coex/submodule.py:218
Method__init__
( self, inp: int, oup: int, stride: int, expand_ra
stereo/modeling/models/coex/submodule.py:238
Method__init__
(self, in_channels, out_channels, stride=1, bn=True, relu=True)
stereo/modeling/models/coex/submodule.py:278
← previousnext →1,001–1,100 of 2,487, ranked by callers