MCPcopy Create free account

hub / github.com/Junda24/MonSter / functions

Functions359 in github.com/Junda24/MonSter

⨍ Functions 359◇ Types & classes 92
↓ 18 callersFunctionpool2x
(x)
core/update.py:106
↓ 16 callersMethodpad
(self, *inputs)
core/utils/utils.py:18
↓ 15 callersMethod__init__
(self, hidden_dim, input_dim, kernel_size=3)
core/update.py:27
↓ 14 callersMethodunpad
(self, x)
core/utils/utils.py:22
↓ 12 callersFunctioninterp
(x, dest)
core/update.py:73
↓ 11 callersMethod__init__
(self, aug_params=None, root='/data2/cjd/StereoDatasets/eth3d', split='training')
core/stereo_datasets.py:189
↓ 10 callersFunctionconv2d
(in_channels, out_channels, kernel_size=3, stride=1, dilation=1, groups=1)
core/refinement.py:8
↓ 9 callersMethod__init__
(self)
core/refinement.py:483
↓ 8 callersFunction_make_fusion_block
(features, use_bn, size=None)
Depth-Anything-V2-list3/depth_anything_v2/dpt.py:12
↓ 8 callersMethodfreeze_bn
(self)
core/monster.py:333
↓ 6 callersMethod__init__
(self, in_channels, out_channels, deconv=False, is_3d=False, concat=True, keep_concat=True, bn=True, relu=True
core/submodule.py:40
↓ 6 callersMethodconstrain_to_multiple_of
(self, x, min_val=0, max_val=None)
Depth-Anything-V2-list3/depth_anything_v2/util/transform.py:51
↓ 6 callersMethodget_intermediate_layers
( self, x: torch.Tensor, n: Union[int, Sequence] = 1, # Layers or n last layers to ta
Depth-Anything-V2-list3/depth_anything_v2/dinov2.py:297
↓ 5 callersMethod__init__
(self)
core/extractor.py:328
↓ 5 callersMethod_make_layer
(self, dim, stride=1)
core/extractor.py:271
↓ 5 callersMethodpadding
(self, data, pad_len)
core/dataset/augmentor_mix_dataset.py:326
↓ 4 callersFunctionRGB_noise_aug
(image, sigma, rng)
core/dataset/transform.py:313
↓ 4 callersMethod__init__
( self, in_channels, features=256, use_bn=False, out_channels=[256
Depth-Anything-V2-list3/depth_anything_v2/dpt.py:39
↓ 4 callersFunctiondisp_warp
Warping by disparity Args: img: [B, 3, H, W] disp: [B, 1, H, W], positive padding_mode: 'zeros' or 'border' Returns:
core/warp.py:53
↓ 4 callersFunctionimage_blur_all
全局高斯模糊 kernel_size：5,5 - 15,15
core/dataset/transform.py:423
↓ 4 callersFunctionimage_blur_mask
在mask区域加入高斯模糊 brightness: -40-40
core/dataset/transform.py:398
↓ 4 callersMethodload_pkl
(self, pkl_path)
core/dataset/mix_dataset.py:203
↓ 4 callersFunctionmask_ge
(shape, rng, weights=[0.5, 0.5])
core/dataset/transform.py:432
↓ 4 callersMethodpadding
(self, data, pad_len)
core/dataset/augmentor_mix_dataset.py:67
↓ 4 callersMethodprepare_tokens_with_masks
(self, x, masks=None)
Depth-Anything-V2-list3/depth_anything_v2/dinov2.py:212
↓ 4 callersFunctionwhite_balance_augmentation
(img, ratio)
core/dataset/transform.py:303
↓ 3 callersMethod__init__
(self, args)
core/monster.py:221
↓ 3 callersMethod_make_layer
(self, dim, stride=1)
core/extractor.py:168
↓ 3 callersFunctiondefault_conv
(in_channels, out_channels, kernel_size, bias=True)
core/refinement.py:360
↓ 3 callersMethodget_disp
(self, path, input_h, input_w, dataset_name)
core/dataset/mix_dataset.py:213
↓ 3 callersFunctiongray_2_colormap_np
(img, cmap = 'rainbow', max = None)
train_eth3d.py:39
↓ 3 callersFunctiongray_2_colormap_np
(img, cmap = 'rainbow', max = None)
train_kitti.py:23
↓ 3 callersFunctiongray_2_colormap_np
(img, cmap = 'rainbow', max = None)
train_middlebury.py:38
↓ 3 callersFunctiongray_2_colormap_np
(img, cmap = 'rainbow', max = None)
train_sceneflow.py:25
↓ 3 callersFunctionreadPFM
(file)
core/utils/frame_utils.py:34
↓ 2 callersMethod__len__
(self)
core/dataset/mix_dataset.py:428
↓ 2 callersMethod_add_things
Add FlyingThings3D data
core/stereo_datasets.py:142
↓ 2 callersFunction_make_scratch
(in_shape, out_shape, groups=1, expand=False)
Depth-Anything-V2-list3/depth_anything_v2/util/blocks.py:4
↓ 2 callersFunctionbilinear_sampler
Wrapper for grid_sample, uses pixel coordinates
core/utils/utils.py:59
↓ 2 callersMethodchromatic_augmentation
(self, img)
core/dataset/augmentor_mix_dataset.py:46
↓ 2 callersMethodchromatic_augmentation
(self, img)
core/dataset/augmentor_mix_dataset.py:305
↓ 2 callersMethodchromatic_augmentation
(self, img)
core/dataset/pinhole_augmentor_mix_dataset.py:51
↓ 2 callersFunctionclip
(img: np.ndarray, dtype: np.dtype, maxval: float)
core/dataset/transform.py:93
↓ 2 callersFunctioncontext_upsample
(disp_low, up_weights)
core/submodule.py:241
↓ 2 callersFunctioncorrelation
(fea1, fea2)
core/submodule.py:189
↓ 2 callersFunctiondrop_add_residual_stochastic_depth
( x: Tensor, residual_func: Callable[[Tensor], Tensor], sample_drop_ratio: float = 0.0, )
Depth-Anything-V2-list3/depth_anything_v2/dinov2_layers/block.py:110
↓ 2 callersFunctiondrop_add_residual_stochastic_depth_list
( x_list: List[Tensor], residual_func: Callable[[Tensor, Any], Tensor], sample_drop_ratio: float =
Depth-Anything-V2-list3/depth_anything_v2/dinov2_layers/block.py:181
↓ 2 callersFunctionget_attn_bias_and_cat
this will perform the index select, cat the tensors, and provide the attn_bias from cache
Depth-Anything-V2-list3/depth_anything_v2/dinov2_layers/block.py:157
↓ 2 callersFunctiongroupwise_correlation
(fea1, fea2, num_groups)
core/submodule.py:151
↓ 2 callersFunctioninterp
(x, sample_grid, padding_mode)
core/warp.py:40
↓ 2 callersFunctionload_image
(imfile)
save_pfm_eth.py:20
↓ 2 callersFunctionload_image
(imfile)
save_disp.py:51
↓ 2 callersFunctionload_image
(imfile)
save_pfm.py:20
↓ 2 callersFunctionload_image
(imfile)
demo_video.py:77
↓ 2 callersFunctionmake_2tuple
(x)
Depth-Anything-V2-list3/depth_anything_v2/dinov2_layers/patch_embed.py:17
↓ 2 callersFunctionmake_odd_val
(v1, v2)
core/dataset/transform.py:213
↓ 2 callersFunctionmotion_blur_v2
(image, kernel)
core/dataset/transform.py:164
↓ 2 callersFunctionnorm_correlation
(fea1, fea2)
core/submodule.py:174
↓ 2 callersFunctionrandom_gamma
(image, gamma)
core/dataset/transform.py:79
↓ 2 callersFunctionrandom_hue_saturation
(image, hue_shift, sat_shift, val_shift)
core/dataset/transform.py:89
↓ 2 callersMethodupsample_disp
(self, disp, mask_feat_4, stem_2x)
core/monster.py:340
↓ 1 callersFunctionDINOv2
(model_name)
Depth-Anything-V2-list3/depth_anything_v2/dinov2.py:398
↓ 1 callersFunctionPinholeEulerAnglesToRotationMatrix
(theta)
core/utils/geometry.py:29
↓ 1 callersMethod__init__
( self, image_height=384, image_width=512, max_disp=150,
core/dataset/augmentor_mix_dataset.py:8
↓ 1 callersMethod__init__
Init. Args: features (int): number of features
Depth-Anything-V2-list3/depth_anything_v2/util/blocks.py:33
↓ 1 callersMethod__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
Depth-Anything-V2-list3/depth_anything_v2/dinov2_layers/swiglu_ffn.py:14
↓ 1 callersMethod_add_driving
Add FlyingThings3D data
core/stereo_datasets.py:173
↓ 1 callersMethod_add_monkaa
Add FlyingThings3D data
core/stereo_datasets.py:158
↓ 1 callersMethod_get_intermediate_layers_chunked
(self, x, n=1)
Depth-Anything-V2-list3/depth_anything_v2/dinov2.py:283
↓ 1 callersMethod_get_intermediate_layers_not_chunked
(self, x, n=1)
Depth-Anything-V2-list3/depth_anything_v2/dinov2.py:271
↓ 1 callersFunction_maybe_process_in_chunks
( process_fn: Callable[Concatenate[np.ndarray, P], np.ndarray], **kwargs )
core/dataset/transform.py:172
↓ 1 callersFunction_shift_hsv_non_uint8
(img, hue_shift, sat_shift, val_shift)
core/dataset/transform.py:120
↓ 1 callersFunction_shift_hsv_uint8
(img, hue_shift, sat_shift, val_shift)
core/dataset/transform.py:96
↓ 1 callersFunctionadd_haze_v1
添加雾霾 t : 透视率 0~1 A : 大气光照
core/dataset/transform.py:324
↓ 1 callersFunctionadd_residual
(x, brange, residual, residual_scale_factor, scaling_vector=None)
Depth-Anything-V2-list3/depth_anything_v2/dinov2_layers/block.py:142
↓ 1 callersFunctionbuild_gwc_volume
(refimg_fea, targetimg_fea, maxdisp, num_groups)
core/submodule.py:159
↓ 1 callersMethodcolor_transform
Photometric augmentation
core/utils/augmentor.py:82
↓ 1 callersMethodcolor_transform
(self, img1, img2)
core/utils/augmentor.py:205
↓ 1 callersFunctioncompute_scale_shift
计算 monocular depth 和 ground truth depth 之间的 scale 和 shift. 参数: monocular_depth (torch.Tensor): 单目深度图，形状为 (H, W) 或 (N, H, W) gt_d
core/monster.py:26
↓ 1 callersMethodcorr
(fmap1, fmap2)
core/geometry.py:62
↓ 1 callersFunctioncount_parameters
(model)
evaluate_stereo.py:47
↓ 1 callersFunctiondecompress
(data, dtype)
core/utils/file_io.py:29
↓ 1 callersFunctiondecompress
(data, dtype)
core/dataset/mix_dataset.py:436
↓ 1 callersFunctiondemo
(args)
save_pfm_eth.py:25
↓ 1 callersFunctiondemo
(args)
save_disp.py:56
↓ 1 callersFunctiondemo
(args)
save_pfm.py:25
↓ 1 callersFunctiondemo
(args)
demo_video.py:82
↓ 1 callersFunctiondisp2depth
(disp)
core/dataset/transform.py:336
↓ 1 callersFunctiondisparity_regression
(x, maxdisp)
core/submodule.py:219
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0.0, training: bool = False)
Depth-Anything-V2-list3/depth_anything_v2/dinov2_layers/drop_path.py:15
↓ 1 callersFunctionensure_dir
(path)
core/utils/file_io.py:9
↓ 1 callersMethoderaser_transform
Occlusion augmentation
core/utils/augmentor.py:98
↓ 1 callersMethoderaser_transform
(self, img1, img2)
core/utils/augmentor.py:211
↓ 1 callersFunctioneulerAnglesToRotationMatrix
(theta)
core/utils/geometry.py:4
↓ 1 callersFunctionfetch_optimizer
Create the optimizer and learning rate scheduler
train_eth3d.py:93
↓ 1 callersFunctionfetch_optimizer
Create the optimizer and learning rate scheduler
train_kitti.py:77
↓ 1 callersFunctionfetch_optimizer
Create the optimizer and learning rate scheduler
train_middlebury.py:92
↓ 1 callersFunctionfetch_optimizer
Create the optimizer and learning rate scheduler
train_sceneflow.py:79
↓ 1 callersMethodforward
(self, x)
Depth-Anything-V2-list3/depth_anything_v2/dpt.py:335
↓ 1 callersMethodforward
(self, x)
Depth-Anything-V2-list3/depth_anything_v2/dpt.py:420
next →1–100 of 359, ranked by callers