MCPcopy Create free account

hub / github.com/M2219/ESMStereo / functions

Functions258 in github.com/M2219/ESMStereo

↓ 18 callersMethodmean
(self)
utils/experiment.py:138
↓ 12 callersMethodupdate
(self, val, n=1)
train_kitti.py:249
↓ 10 callersMethod__init__
(self, chans: List[int], vol_size: int)
models/ESMStereo_confidence.py:80
↓ 9 callersFunctiontensor2float
(vars)
utils/experiment.py:40
↓ 9 callersFunctiontensor2numpy
(vars)
utils/experiment.py:50
↓ 8 callersMethod__init__
(self, chans: List[int], vol_size: int)
models/ESMStereo.py:80
↓ 8 callersMethod__init__
(self, chans: List[int], vol_size: int)
models/ESMStereo_trt.py:80
↓ 7 callersMethod__init__
(self, dim: int, kernel_size: int, mlp_ratio: int=2)
models/shufflemixer.py:98
↓ 7 callersMethodupdate
(self, val, n=1)
train_sceneflow.py:289
↓ 6 callersFunctionThres_metric
(D_est, D_gt, mask, thres)
utils/metrics.py:62
↓ 6 callersFunctiondisparity_regression
(x, maxdisp)
models/submodule.py:211
↓ 6 callersFunctionsave_scalars
( logger: SummaryWriter, mode_tag: str, scalar_dict: Dict[str, Union[float, List[float]]], glo
utils/experiment.py:64
↓ 5 callersFunctionpfm_imread
(filename)
datasets/data_io.py:26
↓ 4 callersFunctionD1_metric
(D_est, D_gt, mask)
utils/metrics.py:44
↓ 4 callersFunctionEPE_metric
(D_est, D_gt, mask)
utils/metrics.py:72
↓ 4 callersFunctionget_transform
()
datasets/data_io.py:7
↓ 3 callersFunctionbuild_gwc_volume
(refimg_fea: torch.Tensor, targetimg_fea: torch.Tensor, maxdisp: int, num_groups: int)
models/submodule.py:151
↓ 3 callersFunctionbuild_norm_correlation_volume
(refimg_fea, targetimg_fea, maxdisp)
models/submodule.py:191
↓ 3 callersFunctionhorizontal_flip
(img)
datasets/middlebury_loader.py:134
↓ 3 callersFunctionread_all_lines
(filename: str)
datasets/data_io.py:20
↓ 3 callersFunctionregression_topk
(cost: torch.Tensor, disparity_samples: torch.Tensor, k: int)
models/submodule.py:218
↓ 3 callersFunctionvis
(D_est_np, D_gt_np, abs_thres=3., rel_thres=0.05, dilate_radius=1)
utils/visualization.py:20
↓ 2 callersMethod__init__
(self, in_channels: int, out_channels: int, deconv: bool=False, is_3d: bool=False, concat: bool=True, keep_con
models/submodule.py:66
↓ 2 callersFunctionadjust_learning_rate
( optimizer: Adam, epoch: int, base_lr: float, lrepochs: str )
utils/experiment.py:103
↓ 2 callersFunctioncheck_allfloat
(vars)
utils/experiment.py:60
↓ 2 callersFunctiongroupwise_correlation
(fea1: torch.Tensor, fea2: torch.Tensor, num_groups: int)
models/submodule.py:143
↓ 2 callersFunctiongroupwise_correlation_norm
(fea1, fea2, num_groups)
models/submodule.py:163
↓ 2 callersFunctiongroupwise_difference
(fea1, fea2, num_groups)
models/submodule.py:105
↓ 2 callersMethodload_image
(self, filename)
datasets/kitti_dataset.py:39
↓ 2 callersMethodload_image
(self, filename)
datasets/sceneflow_dataset.py:33
↓ 2 callersMethodmean
(self)
utils/experiment.py:164
↓ 2 callersFunctionmodel_loss_test
(disp_ests, disp_gts,img_masks)
models/loss.py:24
↓ 2 callersFunctionmodel_loss_train
(disp_ests, disp_gts, img_masks, cv_scale)
models/loss.py:3
↓ 2 callersFunctionnorm_correlation
(fea1, fea2)
models/submodule.py:187
↓ 2 callersFunctionpreprocess_image
virtual_kitti_publisher/src/virtual_kitti_publisher_cuda_node.cpp:239
↓ 2 callersFunctionpreprocess_image
kitti_publisher_ess/src/kitti_publisher_ess_cuda_node.cpp:247
↓ 2 callersFunctionpreprocess_image
kitti_publisher/src/kitti_publisher_cuda_node.cpp:141
↓ 2 callersFunctionpreprocess_image
kitti_publisher_conf/src/kitti_publisher_conf_cuda_node.cpp:285
↓ 2 callersMethodupdate
(self, x: Dict[str, Union[float, List[float]]])
utils/experiment.py:147
↓ 1 callersMethodL2normalize
(self, x)
models/ESMStereo_confidence.py:645
↓ 1 callersFunctioncheck_shape_for_metric_computation
(*vars)
utils/metrics.py:10
↓ 1 callersFunctioncomputeEPE
virtual_kitti_publisher/src/virtual_kitti_publisher_cuda_node.cpp:80
↓ 1 callersFunctioncomputeEPE
kitti_publisher_ess/src/kitti_publisher_ess_cuda_node.cpp:92
↓ 1 callersFunctioncomputeEPE
kitti_publisher_conf/src/kitti_publisher_conf_cuda_node.cpp:55
↓ 1 callersFunctioncount_parameters
(model)
onnx_transformed_confidence.py:11
↓ 1 callersFunctioncount_parameters
(model)
onnx_transformed.py:11
↓ 1 callersFunctioncount_parameters
(model)
train_sceneflow.py:69
↓ 1 callersFunctiondepthToDisparity
virtual_kitti_publisher/src/virtual_kitti_publisher_cuda_node.cpp:55
↓ 1 callersFunctiondepthToDisparity
kitti_publisher_ess/src/kitti_publisher_ess_cuda_node.cpp:68
↓ 1 callersFunctiongen_error_colormap
()
utils/visualization.py:4
↓ 1 callersFunctiongen_error_colormap
kitti_publisher_conf/src/kitti_publisher_conf_cuda_node.cpp:69
↓ 1 callersFunctioninitializeTensorRT
virtual_kitti_publisher/src/virtual_kitti_publisher_cuda_node.cpp:309
↓ 1 callersFunctioninitializeTensorRT
kitti_publisher_ess/src/kitti_publisher_ess_cuda_node.cpp:317
↓ 1 callersFunctioninitializeTensorRT
kitti_publisher/src/kitti_publisher_cuda_node.cpp:216
↓ 1 callersFunctioninitializeTensorRT
kitti_publisher_conf/src/kitti_publisher_conf_cuda_node.cpp:360
↓ 1 callersFunctionloadEngine
virtual_kitti_publisher/src/virtual_kitti_publisher_cuda_node.cpp:275
↓ 1 callersFunctionloadEngine
kitti_publisher_ess/src/kitti_publisher_ess_cuda_node.cpp:283
↓ 1 callersFunctionloadEngine
kitti_publisher/src/kitti_publisher_cuda_node.cpp:182
↓ 1 callersFunctionloadEngine
kitti_publisher_conf/src/kitti_publisher_conf_cuda_node.cpp:326
↓ 1 callersFunctionloadPluginLibrary
kitti_publisher_ess/src/kitti_publisher_ess_cuda_node.cpp:56
↓ 1 callersMethodload_disp
(self, filename)
datasets/kitti_dataset.py:42
↓ 1 callersMethodload_disp
(self, filename)
datasets/sceneflow_dataset.py:36
↓ 1 callersFunctionload_path
(list_filename)
save_vid.py:68
↓ 1 callersMethodload_path
(self, list_filename: str)
datasets/kitti_dataset.py:28
↓ 1 callersMethodload_path
(self, list_filename: str)
datasets/sceneflow_dataset.py:25
↓ 1 callersFunctionmeasure_performance
(dummy_input1, dummy_input2)
train_sceneflow.py:255
↓ 1 callersMethodreset
(self)
train_kitti.py:243
↓ 1 callersMethodreset
(self)
train_sceneflow.py:283
↓ 1 callersFunctiontest
()
save_disp.py:64
↓ 1 callersFunctiontest
()
save_vid.py:77
↓ 1 callersFunctiontest_sample
(sample, compute_metrics=True)
train_kitti.py:212
↓ 1 callersFunctiontest_sample
(sample)
save_disp.py:100
↓ 1 callersFunctiontest_sample
(im_left, im_right)
save_vid.py:140
↓ 1 callersFunctiontest_sample
(sample, compute_metrics=True)
train_sceneflow.py:230
↓ 1 callersFunctiontest_trainset
()
test_mid.py:64
↓ 1 callersFunctionto_3d
(x: torch.Tensor)
models/shufflemixer.py:40
↓ 1 callersFunctionto_4d
(x: torch.Tensor, h: int, w: int)
models/shufflemixer.py:43
↓ 1 callersFunctiontrain
()
train_kitti.py:101
↓ 1 callersFunctiontrain
()
train_sceneflow.py:116
↓ 1 callersFunctiontrain_sample
(sample, compute_metrics=False)
train_kitti.py:179
↓ 1 callersFunctiontrain_sample
(sample, compute_metrics=False)
train_sceneflow.py:196
↓ 1 callersFunctionvis
kitti_publisher_conf/src/kitti_publisher_conf_cuda_node.cpp:94
↓ 1 callersFunctionvisualize_and_record_disparity
virtual_kitti_publisher/src/virtual_kitti_publisher_cuda_node.cpp:131
↓ 1 callersFunctionvisualize_and_record_disparity
kitti_publisher_ess/src/kitti_publisher_ess_cuda_node.cpp:142
↓ 1 callersFunctionvisualize_and_record_disparity
kitti_publisher/src/kitti_publisher_cuda_node.cpp:53
↓ 1 callersFunctionvisualize_and_record_disparity
kitti_publisher_conf/src/kitti_publisher_conf_cuda_node.cpp:153
↓ 1 callersMethodweight_init
(self)
models/ESMStereo.py:25
↓ 1 callersMethodweight_init
(self)
models/ESMStereo_confidence.py:25
↓ 1 callersMethodweight_init
(self)
models/ESMStereo_trt.py:25
FunctionD1_metric_thres
(D_est, D_gt, mask, thres)
utils/metrics.py:53
MethodKittiImagePublisher
virtual_kitti_publisher/src/virtual_kitti_publisher_cuda_node.cpp:360
MethodKittiImagePublisher
kitti_publisher_ess/src/kitti_publisher_ess_cuda_node.cpp:376
MethodKittiImagePublisher
kitti_publisher/src/kitti_publisher_cuda_node.cpp:273
MethodKittiImagePublisher
kitti_publisher_conf/src/kitti_publisher_conf_cuda_node.cpp:418
Method__call__
(self, input, target)
datasets/flow_transforms.py:16
Method__call__
(self, inputs, target)
datasets/flow_transforms.py:35
Method__call__
(self, inputs, target)
datasets/flow_transforms.py:62
Method__call__
(self, inputs, target)
datasets/flow_transforms.py:87
Method__getitem__
(self, index)
datasets/KITTI2015loader.py:95
Method__getitem__
(self, index)
datasets/middlebury_loader.py:78
next →1–100 of 258, ranked by callers