MCPcopy Create free account

hub / github.com/XiandaGuo/OpenStereo / functions

Functions2,487 in github.com/XiandaGuo/OpenStereo

↓ 1 callersMethodbuild_warmup
(self)
stereo/modeling/trainer_template.py:132
↓ 1 callersMethodcal_focal_loss
(self, input, output)
stereo/modeling/models/iinet/loss.py:160
↓ 1 callersMethodcal_l1_loss
(self, inputs: dict, outputs: dict, validmask_pyr: list)
stereo/modeling/models/iinet/loss.py:127
↓ 1 callersMethodcal_msgrad_loss
(self, inputs: dict, outputs: dict, validmask_pyr: list)
stereo/modeling/models/iinet/loss.py:140
↓ 1 callersMethodcal_normal_loss
(self, inputs: dict, outputs: dict, valid_mask: Tensor)
stereo/modeling/models/iinet/loss.py:79
↓ 1 callersMethodcalc_px_error
compute px error :param pred: disparity prediction [N,H,W] :param disp: ground truth disparity [N,H,W] :param loss_d
stereo/modeling/models/sttr/utilities/loss.py:32
↓ 1 callersMethodcall_checkpoint_bottleneck
(self, input)
stereo/modeling/models/sttr/utilities/densenet_in.py:54
↓ 1 callersFunctioncenter_crop
(layer, max_height, max_width)
stereo/modeling/models/sttr/utilities/misc.py:24
↓ 1 callersFunctionchannel_length
(x)
stereo/modeling/models/fadnet/submodule.py:667
↓ 1 callersFunctioncheck_forward_equal_with_pytorch_double
()
stereo/modeling/models/nmrf/ops/test.py:27
↓ 1 callersFunctioncheck_forward_equal_with_pytorch_float
()
stereo/modeling/models/nmrf/ops/test.py:54
↓ 1 callersFunctioncheck_gradient_numerical
(channels=4, grad_value=True, grad_sampling_loc=True
stereo/modeling/models/nmrf/ops/test.py:78
↓ 1 callersFunctioncheck_python
(minimum='3.7.0')
deploy/deploy_utils.py:50
↓ 1 callersFunctioncheck_shape_for_metric_computation
(*vars)
stereo/modeling/models/iinet/metrics.py:21
↓ 1 callersFunctioncheckpoint_filter_fn
(state_dict)
stereo/modeling/models/nmrf/backbone.py:162
↓ 1 callersFunctionclones
Produce N identical layers.
stereo/modeling/models/fadnet/submodule.py:12
↓ 1 callersFunctioncollect_torch_env
()
stereo/modeling/models/nmrf/utils/misc.py:268
↓ 1 callersMethodcompute
(self)
stereo/modeling/models/foundationstereo/dinov2/eval/metrics.py:111
↓ 1 callersMethodcompute_entropy_loss
compute binary entropy loss on occlusion mask :param occ_pred: occlusion prediction, [N,H,W] :param inputs: input data
stereo/modeling/models/sttr/utilities/loss.py:150
↓ 1 callersMethodcompute_epe
compute EPE :param pred: disparity prediction [N,H,W] :param disp: ground truth disparity [N,H,W] :param loss_dict:
stereo/modeling/models/sttr/utilities/loss.py:50
↓ 1 callersMethodcompute_iou
compute IOU on occlusion :param pred: occlusion prediction [N,H,W] :param occ_mask: ground truth occlusion mask [N,H,W]
stereo/modeling/models/sttr/utilities/loss.py:63
↓ 1 callersMethodcompute_left_occ_region
Compute occluded region on the left image border :param w: image width :param disp: left disparity :return: occ mask
stereo/datasets/sceneflow_dataset.py:151
↓ 1 callersMethodcompute_neighbors
(self, features_rank)
stereo/modeling/models/foundationstereo/dinov2/eval/knn.py:162
↓ 1 callersMethodcompute_right_occ_region
Compute occluded region on the right image border :param w: image width :param disp: right disparity :return: occ mas
stereo/datasets/sceneflow_dataset.py:164
↓ 1 callersMethodcompute_rr_loss
(self, outputs: dict, inputs: NestedTensor, invalid_mask: Tensor)
stereo/modeling/models/sttr/utilities/loss.py:88
↓ 1 callersFunctioncompute_scale_shift
计算 monocular depth 和 ground truth depth 之间的 scale 和 shift. 参数: monocular_depth (torch.Tensor): 单目深度图,形状为 (H, W) 或 (N, H, W) gt_d
stereo/modeling/models/monster/monster.py:24
↓ 1 callersFunctionconfig_loader
(path)
stereo/utils/common_utils.py:18
↓ 1 callersFunctioncontext_upsample
(disp_low, up_weights, scale_factor=4)
stereo/modeling/disp_refinement/disp_refinement.py:194
↓ 1 callersFunctioncontext_upsample
@disp_low: (b,1,h,w) 1/4 resolution @up_weights: (b,9,4*h,4*w) Image resolution
stereo/modeling/models/foundationstereo/core/submodule.py:490
↓ 1 callersMethodcorr
(fmap1, fmap2)
stereo/modeling/disp_refinement/gru_blocks.py:222
↓ 1 callersMethodcorr
(fmap1, fmap2)
stereo/modeling/models/stereobase/gru_blocks.py:222
↓ 1 callersMethodcorr
(fmap1, fmap2, normalize=True)
stereo/modeling/models/fast_foundationstereo/core/geometry.py:70
↓ 1 callersMethodcorr
(fmap1, fmap2)
stereo/modeling/models/igev/geometry.py:59
↓ 1 callersMethodcorr
(fmap1, fmap2)
stereo/modeling/models/foundationstereo/core/geometry.py:67
↓ 1 callersMethodcorr
(fmap1, fmap2)
stereo/modeling/models/monster/geometry.py:62
↓ 1 callersMethodcorr
(fmap1, fmap2)
stereo/modeling/models/igevpp/geometry.py:80
↓ 1 callersFunctioncorrelation_volume
(left_feature, right_feature, max_disp)
stereo/modeling/cost_volume/cost_volume.py:32
↓ 1 callersMethodcost_volume_construction
(self, left_feature, right_feature)
stereo/modeling/models/aanet/aanet.py:57
↓ 1 callersFunctioncreate_class_indices_mapping
(labels)
stereo/modeling/models/foundationstereo/dinov2/eval/knn.py:234
↓ 1 callersFunctioncreate_module_dict
(*, module, n_per_class_list, n_tries, nb_knn, train_features, train_labels)
stereo/modeling/models/foundationstereo/dinov2/eval/knn.py:198
↓ 1 callersMethodcrop_all
(self, inputs)
stereo/datasets/mono.py:184
↓ 1 callersFunctiondefault_setup
(args)
stereo/modeling/models/foundationstereo/dinov2/utils/config.py:49
↓ 1 callersFunctiondeform_inputs
(x, patch_size)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/adapter_modules.py:32
↓ 1 callersFunctiondeform_inputs_dn
(x)
stereo/modeling/models/nmrf/adaptor_modules.py:25
↓ 1 callersFunctiondepth_uint8_decoding
(depth_uint8, scale=1000)
stereo/datasets/foundationstereo.py:13
↓ 1 callersFunctiondetect_compute_compatibility
(CUDA_HOME, so_file)
stereo/modeling/models/nmrf/utils/misc.py:247
↓ 1 callersFunctiondice_loss
Calculate dice loss, which is proposed in `V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation <https://arxiv
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/losses/dice_loss.py:12
↓ 1 callersFunctiondisp_map
Based on color histogram, convert the gray disp into color disp map. The histogram consists of 7 bins, value of each is e.g. [114.0, 185.0, 1
stereo/utils/disp_color.py:36
↓ 1 callersFunctiondisp_warp
Warping by disparity Args: img: [B, 3, H, W] disp: [B, 1, H, W], positive padding_mode: 'zeros' or 'border' Returns:
stereo/modeling/disp_refinement/disp_refinement.py:113
↓ 1 callersFunctiondisp_warp
Warping by disparity Args: img: [B, 3, H, W] disp: [B, 1, H, W], positive padding_mode: 'zeros' or 'border' Returns:
stereo/modeling/models/aanet/submodule.py:801
↓ 1 callersMethoddisparity_computation
(self, aggregation)
stereo/modeling/models/aanet/aanet.py:66
↓ 1 callersMethoddisparity_read
(self,filename)
stereo/datasets/sintel_dataset.py:73
↓ 1 callersMethoddisparity_refinement
(self, left_img, right_img, disparity)
stereo/modeling/models/aanet/aanet.py:79
↓ 1 callersFunctiondisparity_regression
(prob, maxdisp, interval)
stereo/modeling/models/igev_rt/submodule.py:219
↓ 1 callersFunctiondisparity_regression
(x, maxdisp)
stereo/modeling/models/igev/submodule.py:230
↓ 1 callersFunctiondisparity_regression
(x, maxdisp)
stereo/modeling/models/foundationstereo/core/submodule.py:465
↓ 1 callersFunctiondisparity_regression
(x, maxdisp)
stereo/modeling/models/monster/submodule.py:219
↓ 1 callersFunctiondisparity_variance
(x, maxdisp, disparity)
stereo/modeling/models/cfnet/submodule.py:128
↓ 1 callersFunctiondisparity_variance_confidence
(x, disparity_samples, disparity)
stereo/modeling/models/cfnet/submodule.py:137
↓ 1 callersFunctiondo_train
(cfg, model, resume=False)
stereo/modeling/models/foundationstereo/dinov2/train/train.py:134
↓ 1 callersFunctiondownsample_disp
(disp, label, stride=8)
stereo/modeling/models/nmrf/utils/frame_utils.py:272
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0.0, training: bool = False)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/drop_path.py:13
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0.0, training: bool = False)
stereo/modeling/models/foundationstereo/dinov2/layers/drop_path.py:14
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0.0, training: bool = False)
stereo/modeling/models/monster/depth_anything_v2/dinov2_layers/drop_path.py:15
↓ 1 callersMethoddump
Returns: str: a yaml string representation of the config
stereo/modeling/models/nmrf/config/config.py:157
↓ 1 callersMethoddump_in_output_file
(self, iteration, iter_time, data_time)
stereo/modeling/models/foundationstereo/dinov2/logging/helpers.py:53
↓ 1 callersFunctioneval_knn
( model, train_dataset, val_dataset, accuracy_averaging, nb_knn, temperature, batc
stereo/modeling/models/foundationstereo/dinov2/eval/knn.py:245
↓ 1 callersFunctioneval_knn_with_model
( model, output_dir, train_dataset_str="ImageNet:split=TRAIN", val_dataset_str="ImageNet:split
stereo/modeling/models/foundationstereo/dinov2/eval/knn.py:318
↓ 1 callersFunctioneval_linear
( *, feature_model, linear_classifiers, train_data_loader, val_data_loader, metrics_fi
stereo/modeling/models/foundationstereo/dinov2/eval/linear.py:313
↓ 1 callersFunctioneval_log_regression
Implements the "standard" process for log regression evaluation: The value of C is chosen by training on train_dataset and evaluating on
stereo/modeling/models/foundationstereo/dinov2/eval/log_regression.py:252
↓ 1 callersFunctioneval_log_regression_with_model
( model, train_dataset_str="ImageNet:split=TRAIN", val_dataset_str="ImageNet:split=VAL", finet
stereo/modeling/models/foundationstereo/dinov2/eval/log_regression.py:362
↓ 1 callersMethodeval_one_epoch
(self, current_epoch)
stereo/modeling/trainer_template.py:260
↓ 1 callersMethodevaluate
(self)
stereo/modeling/models/nmrf/utils/evaluation.py:153
↓ 1 callersMethodevaluate
(self)
stereo/modeling/models/nmrf/utils/evaluation.py:385
↓ 1 callersFunctionexport_coreml
(model, im, file, int8, half, prefix=colorstr('CoreML:'))
deploy/export.py:264
↓ 1 callersFunctionexport_engine
(model, inputs, file, half, dynamic, simplify, optimize, workspace=4, verbose=False, prefix=colorstr('TensorRT
deploy/export.py:198
↓ 1 callersFunctionexport_onnx
(model, inputs, weights, opset, dynamic, simplify, prefix=colorstr('ONNX:'))
deploy/export.py:129
↓ 1 callersFunctionexport_openvino
(file, half, prefix=colorstr('OpenVINO:'))
deploy/export.py:183
↓ 1 callersFunctionexport_paddle
(model, inputs, file, prefix=colorstr('PaddlePaddle:'))
deploy/export.py:250
↓ 1 callersFunctionexport_torchscript
(model, inputs, file, optimize, prefix=colorstr('TorchScript:'))
deploy/export.py:114
↓ 1 callersMethodextract_feature
(self, img1, img2)
stereo/modeling/models/nmrf/NMRF.py:194
↓ 1 callersFunctionextract_features_with_dataloader
(model, data_loader, sample_count, gather_on_cpu=False)
stereo/modeling/models/foundationstereo/dinov2/eval/utils.py:113
↓ 1 callersFunctionfilter_train
(mapping, n_per_class, seed)
stereo/modeling/models/foundationstereo/dinov2/eval/knn.py:225
↓ 1 callersMethodfit
(self, train_features, train_labels)
stereo/modeling/models/foundationstereo/dinov2/eval/log_regression.py:137
↓ 1 callersMethodforward
(self, x)
stereo/modeling/models/nmrf/backbone.py:86
↓ 1 callersMethodforward
(self, x)
stereo/modeling/models/foundationstereo/core/extractor.py:346
↓ 1 callersMethodforward
(self, x: Tensor)
stereo/modeling/models/foundationstereo/dinov2/layers/block.py:89
↓ 1 callersMethodforward
(self, x: Tensor)
stereo/modeling/models/foundationstereo/dinov2/layers/attention.py:56
↓ 1 callersMethodforward
(self, x)
stereo/modeling/models/monster/depth_anything_v2/dpt.py:335
↓ 1 callersMethodforward
(self, x)
stereo/modeling/models/monster/depth_anything_v2/dpt.py:420
↓ 1 callersMethodforward
(self, x: Tensor)
stereo/modeling/models/monster/depth_anything_v2/dinov2_layers/block.py:82
↓ 1 callersMethodforward
(self, x: Tensor)
stereo/modeling/models/monster/depth_anything_v2/dinov2_layers/attention.py:49
↓ 1 callersMethodforward_backward
(self, images, teacher_temp)
stereo/modeling/models/foundationstereo/dinov2/train/ssl_meta_arch.py:132
↓ 1 callersMethodforward_features
(self, x)
stereo/modeling/models/foundationstereo/dinov2/eval/segmentation_m2f/models/backbones/vit.py:508
↓ 1 callersMethodforward_features
(self, x, masks=None)
stereo/modeling/models/monster/depth_anything_v2/dinov2.py:253
↓ 1 callersMethodforward_features_list
(self, x_list, masks_list)
stereo/modeling/models/foundationstereo/dinov2/models/vision_transformer.py:236
↓ 1 callersMethodforward_features_list
(self, x_list, masks_list)
stereo/modeling/models/monster/depth_anything_v2/dinov2.py:233
↓ 1 callersMethodforward_masked
( self, student_patch_tokens_masked, teacher_patch_tokens_masked, student_mask
stereo/modeling/models/foundationstereo/dinov2/loss/ibot_patch_loss.py:105
↓ 1 callersMethodforward_nested
x_list contains a list of tensors to nest together and run
stereo/modeling/models/foundationstereo/dinov2/layers/block.py:212
↓ 1 callersMethodforward_nested
x_list contains a list of tensors to nest together and run
stereo/modeling/models/monster/depth_anything_v2/dinov2_layers/block.py:205
← previousnext →601–700 of 2,487, ranked by callers