Method__init__(self, max_disp=192, regression_topk=2, chans =[16, 24, 32, 96, 160])
stereo/modeling/models/coex/coex_disp_processor.py:69
Method__init__(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/coex/trainer.py:11
Method__init__(self, dim, qk_dim, num_heads=8, attn_drop=0., proj_drop=0., drop_path=0., dropout=0., normalize_before=False)
stereo/modeling/models/nmrf/NMP.py:74
Method__init__(self, dim, window_size, shift_size, num_heads, qk_scale=None, attn_drop=0.)
stereo/modeling/models/nmrf/NMP.py:155
Method__init__(self, dim, qkv_dim, num_heads, window_size=7, shift_size=0,
mlp_ratio=4., qkv_bias=True, qk_
stereo/modeling/models/nmrf/NMP.py:314
Method__init__(self, dim, qk_dim, v_dim, patches_resolution, num_heads,
split_size=7, mlp_ratio=4., qk_scal
stereo/modeling/models/nmrf/NMP.py:510
Method__init__(self, embed_dim, cost_group, layers, norm=None, return_intermediate=False)
stereo/modeling/models/nmrf/NMP.py:605
Method__init__(self, cost_group, dim, layers, norm, return_intermediate=False)
stereo/modeling/models/nmrf/NMP.py:672
Method__init__(self, embed_dim, mlp_ratio, context_dim, split_size, n_heads,
activation="gelu", attn_drop=0
stereo/modeling/models/nmrf/NMP.py:904
Method__init__(self, embed_dim, mlp_ratio, window_size, shift_size, n_heads,
activation="gelu", attn_drop=0
stereo/modeling/models/nmrf/NMP.py:933
Method__init__(self, dim, mlp_ratio, window_size, shift_size, n_heads,
activation="gelu", attn_drop=0., pro
stereo/modeling/models/nmrf/NMP.py:962
Method__init__(self, cost_group, num_proposals, feat_dim, context_dim, num_prop_layers,
prop_embed_dim, mlp
stereo/modeling/models/nmrf/DPN.py:24
Method__init__(self, inplanes, planes, stride, downsample, pad, dilation)
stereo/modeling/models/nmrf/submodule.py:29
Method__init__(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.0, qkv_bias=False)
stereo/modeling/models/nmrf/hybrid_backbone.py:12
Method__init__(
self,
dim,
window_size,
num_heads,
qkv_bias=True,
qk_scale=N
stereo/modeling/models/nmrf/swin.py:91
Method__init__(
self,
dim,
num_heads,
window_size=7,
shift_size=0,
mlp_ratio
stereo/modeling/models/nmrf/swin.py:194
Method__init__(self, in_features, hidden_features=None, out_features=None,
act_layer=nn.GELU, drop=0.)
stereo/modeling/models/nmrf/adaptor_modules.py:38
Method__init__(self, dim, num_heads=8, n_points=4, n_levels=1, deform_ratio=1.0,
with_cffn=True, cffn_ratio
stereo/modeling/models/nmrf/adaptor_modules.py:72
Method__init__(self, dim, in_channel_list, num_heads=8, n_points=4,
norm_layer=partial(nn.LayerNorm, eps=1e
stereo/modeling/models/nmrf/adaptor_modules.py:146
Method__init__(self, in_planes, planes, norm_layer=nn.InstanceNorm2d, stride=1, dilation=1)
stereo/modeling/models/nmrf/backbone.py:18
Method__init__(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/nmrf/trainer.py:13
Method__init__(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/msnet/trainer.py:13
Method__init__(self, inp, oup, stride, expanse_ratio, dilation=1)
stereo/modeling/models/msnet/msnet_utils/msnet_blocks.py:8
Method__init__(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/fast_foundationstereo/trainer.py:19
Method__init__(self, args, c04, c08, c16, output_dim=[(128,128,128), (128,128,128)], norm_fn='batch', downsample=3)
stereo/modeling/models/fast_foundationstereo/core/extractor.py:12
Method__init__(self, in_channels, out_channels, deconv=False, is_3d=False, bn=True, relu=True, norm='batch', **kwargs)
stereo/modeling/models/fast_foundationstereo/core/submodule.py:50
Method__init__(self, C_in, C_out, hidden=None, kernel_size=3, kernel_disp=None, stride=1, norm=nn.BatchNorm3d)
stereo/modeling/models/fast_foundationstereo/core/submodule.py:91
Method__init__(self, inplanes, planes, kernel_size=3, stride=1, padding=1, downsample=None, groups=1, base_width=64, dilatio
stereo/modeling/models/fast_foundationstereo/core/submodule.py:119
Method__init__(self, inplanes, planes, kernel_size=3, stride=1, padding=1, downsample=None, groups=1, base_width=64, dilatio
stereo/modeling/models/fast_foundationstereo/core/submodule.py:159
Method__init__(self, embed_dim, num_heads, dim_feedforward, dropout=0.1, act=nn.GELU, norm=nn.LayerNorm)
stereo/modeling/models/fast_foundationstereo/core/submodule.py:234
Method__init__(self, in_channels, out_channels, deconv=False, is_3d=False, IN=True, relu=True, **kwargs)
stereo/modeling/models/fast_foundationstereo/core/submodule.py:302
Method__init__(self, in_channels, out_channels, c_middle=None, deconv=False, is_3d=False, concat=True, keep_concat=True, IN=
stereo/modeling/models/fast_foundationstereo/core/submodule.py:336
Method__init__(self, d_model, nhead, dim_feedforward, dropout=0.1, act=nn.GELU, norm_first=False, num_transformer=6, max_len
stereo/modeling/models/fast_foundationstereo/core/submodule.py:582
Method__init__(self, hidden_dim=128, input_dim=256, small_kernel_size=1, large_kernel_size=3, patch_size=None)
stereo/modeling/models/fast_foundationstereo/core/update.py:61
Method__init__(self, in_channels, out_channels, deconv=False, is_3d=False, bn=True, relu=True, **kwargs)
stereo/modeling/models/igev/submodule.py:8
Method__init__(self, in_channels, out_channels, deconv=False, is_3d=False, IN=True, relu=True, **kwargs)
stereo/modeling/models/igev/submodule.py:84
Method__init__(self, in_channels, out_channels, deconv=False, is_3d=False, concat=True, keep_concat=True, IN=True,
stereo/modeling/models/igev/submodule.py:113
Method__init__(self, init_fmap1, init_fmap2, geo_volume, num_levels=2, radius=4)
stereo/modeling/models/igev/geometry.py:8
Method__init__(self, args, cfgs, local_rank, global_rank, logger, tb_writer)
stereo/modeling/models/igev/trainer.py:11