MCPcopy Create free account

hub / github.com/TomTomTommi/LiteAnyStereo / functions

Functions160 in github.com/TomTomTommi/LiteAnyStereo

↓ 10 callersMethod__init__
( self, in_channels, out_channels, deconv=False, concat=True,
core/liteanystereov2_H.py:157
↓ 10 callersMethodpad
(self, *inputs)
core/utils/utils.py:18
↓ 8 callersFunctionnormalize_version
(version)
core/models.py:36
↓ 5 callersFunctionnormalize_las2_model_size
(model_size=None)
core/liteanystereov2.py:42
↓ 5 callersMethodunpad
(self, x)
core/utils/utils.py:22
↓ 4 callersMethod__init__
(self, in_channels, left_att, blocks, expanse_ratio, backbone_channels)
core/aggregation_fasternet.py:7
↓ 4 callersFunctionbuild_model
(version, fnet_pretrained=False, model_size=None, max_disp=192)
core/models.py:58
↓ 4 callersFunctionmodel_label
(version, model_size=None)
core/models.py:51
↓ 4 callersFunctionnormalize_model_size
(version, model_size=None)
core/models.py:44
↓ 4 callersFunctionround_by_divider
(x)
Utils.py:84
↓ 3 callersMethod__init__
(self, normalized_shape, eps=1e-6, data_format="channels_last")
core/aggregation.py:75
↓ 3 callersMethod__init__
(self, fnet_pretrained=False)
core/liteanystereov2.py:177
↓ 3 callersMethod__init__
(self, chan_low, chan_high)
core/submodule.py:46
↓ 3 callersMethod__init__
(self, aug_params=None, root='./data/datasets/ETH3D', split='training')
core/stereo_datasets.py:122
↓ 3 callersFunctioncontext_upsample
(depth_low, up_weights)
core/submodule.py:211
↓ 3 callersFunctiondisparity_regression
(x, maxdisp)
core/submodule.py:94
↓ 3 callersFunctionreadPFM
(file)
core/utils/frame_utils.py:34
↓ 2 callersMethod__init__
(self, pretrained=True)
core/fnet.py:45
↓ 2 callersFunction_sample_1d
Sample a [N,C,1,W] tensor with pixel x coordinates shaped [N,1,K,1].
core/liteanystereov2_H.py:33
↓ 2 callersFunctionbuild_correlation_volume
(left_feature, right_feature, max_disp)
core/submodule.py:178
↓ 2 callersFunctiongroupwise_correlation
(fea1, fea2, num_groups)
core/submodule.py:101
↓ 2 callersFunctiongroupwise_correlation_norm
(fea1, fea2, num_groups)
core/submodule.py:137
↓ 2 callersFunctionload_model_weights
(model, checkpoint, strict=True)
core/models.py:80
↓ 2 callersFunctionnorm_correlation
(fea1, fea2)
core/submodule.py:161
↓ 2 callersMethodnormalize_image
(self, img)
core/liteanystereov2_H.py:411
↓ 2 callersMethodnormalize_image
Normalize RGB images in 0-255 range using ImageNet statistics.
core/liteanystereov2.py:144
↓ 2 callersFunctionrequire_checkpoint
(path)
core/models.py:96
↓ 2 callersFunctionresolve_checkpoint
(version, restore_ckpt, model_size=None)
core/models.py:72
↓ 2 callersFunctionruntime_string
(elapsed_list)
evaluate_stereo.py:32
↓ 2 callersFunctionsync_if_needed
(device)
profile_speed.py:39
↓ 2 callersMethodupsample_disp
(self, disp, mask_feat_4, stem_2x)
core/liteanystereov2_H.py:422
↓ 2 callersFunctionvalidate_kitti
Perform validation using the KITTI train split.
evaluate_stereo.py:84
↓ 1 callersFunction_aggregation_config
(model_size)
core/liteanystereov2.py:53
↓ 1 callersMethod_prepare_context
(self, features_left)
core/liteanystereov2_H.py:425
↓ 1 callersFunctionbenchmark
(model, left, right, warmup, total, max_disp, use_amp)
profile_speed.py:45
↓ 1 callersFunctionbuild_gwc_volume_fast
(refimg_fea, targetimg_fea, maxdisp, num_groups)
core/submodule.py:122
↓ 1 callersFunctionbuild_inputs
(height, width, device, pad)
profile_speed.py:25
↓ 1 callersMethodbuild_upsample_mask
(self, mask_feat_4, stem_2x)
core/liteanystereov2_H.py:414
↓ 1 callersMethodcorr
(fmap1, fmap2)
core/liteanystereov2_H.py:86
↓ 1 callersFunctioncount_parameters
(model)
evaluate_stereo.py:28
↓ 1 callersFunctioncount_parameters
(model)
profile_speed.py:21
↓ 1 callersFunctiondefault_checkpoint
(version, model_size=None)
core/models.py:65
↓ 1 callersFunctiondepth2xyzmap
(depth:np.ndarray, K, uvs:np.ndarray=None, zmin=0.1)
Utils.py:47
↓ 1 callersFunctiondepth_uint8_decoding
(depth_uint8, scale=1000)
core/utils/frame_utils.py:160
↓ 1 callersFunctionload_model
(args, device)
demo.py:40
↓ 1 callersFunctionmain
()
evaluate_stereo.py:229
↓ 1 callersFunctionmain
()
profile_speed.py:83
↓ 1 callersFunctionparse_args
()
evaluate_stereo.py:218
↓ 1 callersFunctionparse_args
()
profile_speed.py:66
↓ 1 callersFunctionreadFlow
Read .flo file in Middlebury format
core/utils/frame_utils.py:13
↓ 1 callersFunctionsave_stereo_gif
(img0, img1, output_gif)
demo.py:26
↓ 1 callersFunctionset_logging_format
(level=logging.INFO)
Utils.py:15
↓ 1 callersFunctionset_seed
(seed)
profile_speed.py:14
↓ 1 callersFunctiontoOpen3dCloud
(points,colors=None,normals=None)
Utils.py:34
↓ 1 callersMethodupsample_disp_with_mask
(self, disp, spx_pred)
core/liteanystereov2_H.py:419
↓ 1 callersFunctionvalidate_driving
Perform validation using the DrivingStereo weather split.
evaluate_stereo.py:173
↓ 1 callersFunctionvalidate_eth3d
Perform validation using the ETH3D train split.
evaluate_stereo.py:40
↓ 1 callersFunctionvalidate_middlebury
Perform validation using the Middlebury-v3 dataset.
evaluate_stereo.py:130
↓ 1 callersFunctionvis_disparity
@disp: np array (H,W) @invalid_thres: > thres is invalid
Utils.py:99
FunctionSpatialTransformer_grid
(x, y, disp_range_samples)
core/submodule.py:190
Method__call__
(self, disp, coords, dx)
core/liteanystereov2_H.py:64
Method__getitem__
(self, index)
core/stereo_datasets.py:33
Method__init__
(self, model, max_disp)
flops_count.py:10
Method__init__
(self, in_channels, left_att, blocks, expanse_ratio, backbone_channels)
core/aggregation.py:7
Method__init__
(self, dim, layer_scale_init_value=1e-6)
core/aggregation.py:97
Method__init__
(self, dim, img_feat_dim)
core/aggregation.py:123
Method__init__
(self, fmap1, fmap2, geo_volume, num_levels=2)
core/liteanystereov2_H.py:44
Method__init__
(self, c04, hidden_dim)
core/liteanystereov2_H.py:94
Method__init__
(self, channels, ratio=16)
core/liteanystereov2_H.py:104
Method__init__
(self, kernel_size=7)
core/liteanystereov2_H.py:121
Method__init__
(self, dim, mlp_ratio=4, n_div=4)
core/liteanystereov2_H.py:133
Method__init__
(self, input_dim, output_dim=1)
core/liteanystereov2_H.py:142
Method__init__
(self, corr_levels, corr_radius, volume_dim, hidden_dim, motion_corr_dim, motion_disp_dim)
core/liteanystereov2_H.py:218
Method__init__
(self, hidden_dim, input_dim, kernel_size)
core/liteanystereov2_H.py:237
Method__init__
(self, hidden_dim, input_dim, small_kernel_size=1, large_kernel_size=3)
core/liteanystereov2_H.py:252
Method__init__
( self, corr_levels, corr_radius, volume_dim, hidden_dim, moti
core/liteanystereov2_H.py:272
Method__init__
( self, fnet_pretrained=False, valid_iters=4, corr_levels=2, corr_radi
core/liteanystereov2_H.py:308
Method__init__
(self, model_size=DEFAULT_LAS2_MODEL_SIZE, fnet_pretrained=False)
core/liteanystereov2.py:73
Method__init__
(self, fnet_pretrained=False)
core/liteanystereov2.py:182
Method__init__
(self, fnet_pretrained=False)
core/liteanystereov2.py:187
Method__init__
(self, in_channels, out_channels, kernel_size=3, stride=1, padding=0, bias=False, norm_layer=
core/submodule.py:10
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, bias=False, norm_layer=No
core/submodule.py:28
Method__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, bn=True, relu=True, **kwargs)
core/submodule.py:65
Method__init__
(self, inp, oup, stride, mlp_ratio, n_div=4, act_layer=nn.GELU)
core/aggregation_fasternet.py:88
Method__init__
(self, dim, mlp_ratio, n_div=4, act_layer=nn.GELU, layer_scale_init_value=0.0)
core/aggregation_fasternet.py:108
Method__init__
(self, dim, n_div=4)
core/aggregation_fasternet.py:135
Method__init__
(self, dim, img_feat_dim)
core/aggregation_fasternet.py:148
Method__init__
(self, aug_params=None, sparse=False, reader=None, real_world=False)
core/stereo_datasets.py:15
Method__init__
(self, aug_params=None, image_set='training', year=2015)
core/stereo_datasets.py:138
Method__init__
(self, aug_params=None, root='./data/datasets/Middlebury', split='2014', resolution='F')
core/stereo_datasets.py:160
Method__init__
(self, aug_params=None, root='./data/datasets/DrivingStereoWeather', image_set='rainy')
core/stereo_datasets.py:211
Method__init__
(self, in_channels, out_channels)
core/fnet.py:9
Method__init__
(self, pretrained=True)
core/fnet.py:90
Method__init__
(self, fnet_pretrained=True)
core/liteanystereo.py:11
Method__init__
(self, dims, mode='sintel', divis_by=8)
core/utils/utils.py:9
Method__len__
(self)
core/stereo_datasets.py:117
Method__mul__
(self, v)
core/stereo_datasets.py:109
Functionbilinear_sampler
Wrapper for grid_sample, uses pixel coordinates
core/utils/utils.py:59
Functionbuild_gwc_volume
(refimg_fea, targetimg_fea, maxdisp, num_groups)
core/submodule.py:109
Functionbuild_gwc_volume_norm
(refimg_fea, targetimg_fea, maxdisp, num_groups)
core/submodule.py:148
next →1–100 of 160, ranked by callers