MCPcopy Create free account

hub / github.com/autonomousvision/unimatch / functions

Functions377 in github.com/autonomousvision/unimatch

⨍ Functions 377◇ Types & classes 110
↓ 1 callersFunctioncompute_flow_with_depth_pose
(depth_ref, intrinsics, extrinsics_ref=None, extrinsics_tgt=None, extrinsics_
unimatch/geometry.py:175
↓ 1 callersFunctioncompute_out_of_boundary_mask
(flow, downsample_factor=None)
utils/utils.py:79
↓ 1 callersFunctioncorrelation_softmax_depth
(feature0, feature1, intrinsics, pose,
unimatch/matching.py:203
↓ 1 callersFunctioncreate_eth3d_submission
create submission for the eth3d stereo leaderboard
evaluate_stereo.py:95
↓ 1 callersFunctioncreate_kitti_submission
create submission for the KITTI leaderboard
evaluate_stereo.py:29
↓ 1 callersFunctioncreate_kitti_submission
Create submission for the KITTI leaderboard
evaluate_flow.py:91
↓ 1 callersFunctioncreate_middlebury_submission
create submission for the Middlebury leaderboard
evaluate_stereo.py:199
↓ 1 callersFunctioncreate_sintel_submission
Create submission for the Sintel leaderboard
evaluate_flow.py:20
↓ 1 callersFunctiondepth_grad_loss_func
(depth_preds, depth_gt, valid, inverse_depth_loss=True, gamm
loss/depth_loss.py:58
↓ 1 callersFunctiondepth_loss_func
loss function defined over multiple depth predictions
loss/depth_loss.py:72
↓ 1 callersFunctiondisp_error_img
(D_est_tensor, D_gt_tensor, abs_thres=3., rel_thres=0.05, dilate_radius=1)
utils/visualization.py:35
↓ 1 callersFunctiondump_example
(dataset_name)
dataloader/depth/prepare_demon_train.py:21
↓ 1 callersFunctiondump_example
(dataset_name)
dataloader/depth/prepare_demon_test.py:22
↓ 1 callersMethoderaser_transform
Occlusion augmentation
dataloader/flow/transforms.py:51
↓ 1 callersMethoderaser_transform
(self, img1, img2)
dataloader/flow/transforms.py:185
↓ 1 callersMethodextract_feature
(self, img0, img1)
unimatch/unimatch_depthsplat.py:217
↓ 1 callersMethodextract_feature
(self, img0, img1)
unimatch/unimatch.py:64
↓ 1 callersFunctionextract_video
(video_name)
utils/file_io.py:203
↓ 1 callersFunctionflow_compute_color
Applies the flow color wheel to (possibly clipped) flow components u and v. According to the C++ source code of Daniel Scharstein Accordi
utils/flow_viz.py:69
↓ 1 callersFunctionflow_loss_func
(flow_preds, flow_gt, valid, gamma=0.9, max_flow=400,
loss/flow_loss.py:4
↓ 1 callersFunctionforward_backward_consistency_check
(fwd_flow, bwd_flow, alpha=0.01,
unimatch/geometry.py:75
↓ 1 callersMethodforward_local_window_attn
(self, feature0, flow, local_window_radius=1,
unimatch/attention.py:217
↓ 1 callersFunctiongen_error_colormap
()
utils/visualization.py:19
↓ 1 callersFunctiongenerate_shift_window_attn_mask
(input_resolution, window_size_h, window_size_w, shift_size_h, shift_size
unimatch/utils.py:84
↓ 1 callersFunctiongenerate_shift_window_attn_mask_1d
(input_w, window_size_w, shift_size_w, device=torch.device('cuda'))
unimatch/utils.py:199
↓ 1 callersFunctionget_args_parser
()
main_stereo.py:28
↓ 1 callersFunctionget_args_parser
()
main_depth.py:22
↓ 1 callersFunctionget_args_parser
()
main_flow.py:23
↓ 1 callersFunctionget_depth_grad_loss
(depth_pred, depth_gt, valid, inverse_depth_loss=True)
loss/depth_loss.py:27
↓ 1 callersFunctionglobal_correlation_softmax
(feature0, feature1, pred_bidir_flow=False, )
unimatch/matching.py:7
↓ 1 callersFunctionglobal_correlation_softmax_stereo
(feature0, feature1, )
unimatch/matching.py:126
↓ 1 callersFunctioninference_depth
(model, inference_dir=None, output_path='output',
evaluate_depth.py:297
↓ 1 callersFunctioninference_flow
Inference on a directory or a video
evaluate_flow.py:642
↓ 1 callersFunctioninference_stereo
(model, inference_dir=None, inference_dir_left=None,
evaluate_stereo.py:712
↓ 1 callersFunctionlocal_correlation_softmax
(feature0, feature1, local_radius, padding_mode='zeros',
unimatch/matching.py:39
↓ 1 callersFunctionlocal_correlation_softmax_stereo
(feature0, feature1, local_radius, )
unimatch/matching.py:154
↓ 1 callersMethodlr_summary
(self)
utils/logger.py:35
↓ 1 callersFunctionmain
(args)
main_stereo.py:125
↓ 1 callersFunctionmain
(args)
main_depth.py:116
↓ 1 callersFunctionmain
(args)
main_flow.py:137
↓ 1 callersFunctionmake_color_wheel
Generate color wheel according Middlebury color code :return: Color wheel
utils/flow_viz.py:145
↓ 1 callersFunctionmake_colorwheel
Generates a color wheel for optical flow visualization as presented in: Baker et al. "A Database and Evaluation Methodology for Optical F
utils/flow_viz.py:23
↓ 1 callersFunctionmerge_splits_1d
(splits, h, num_splits=2, )
unimatch/utils.py:171
↓ 1 callersFunctionnormalize_img
(img0, img1)
unimatch/utils.py:23
↓ 1 callersFunctionpreparedata
()
dataloader/depth/prepare_demon_train.py:78
↓ 1 callersFunctionpreparedata
()
dataloader/depth/prepare_demon_test.py:73
↓ 1 callersMethodprint_training_status
(self, mode='train', is_depth=False)
utils/logger.py:24
↓ 1 callersFunctionreadFlow
Read .flo file in Middlebury format
utils/frame_utils.py:10
↓ 1 callersFunctionreadPFM
(file)
utils/frame_utils.py:32
↓ 1 callersFunctionread_disp
(filename, subset=False, vkitti2=False, sintel=False, tartanair=False, instereo2k=False, crester
utils/file_io.py:20
↓ 1 callersMethodresize_sparse_flow_map
(self, flow, valid, fx=1.0, fy=1.0)
dataloader/flow/transforms.py:198
↓ 1 callersFunctionsave_images
(logger, mode_tag, images_dict, global_step)
utils/visualization.py:62
↓ 1 callersFunctionsingle_head_split_window_attention_1d
(q, k, v, relative_position_bias=None,
unimatch/attention.py:107
↓ 1 callersMethodspatial_transform
(self, img1, img2, flow, valid)
dataloader/flow/transforms.py:232
↓ 1 callersFunctiontensor2numpy
(var_dict)
utils/visualization.py:80
↓ 1 callersFunctionwindow_partition_1d
Args: x: (B, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, C)
unimatch/utils.py:185
↓ 1 callersMethodwrite_dict
(self, results)
utils/logger.py:97
↓ 1 callersFunctionzero_module
Zero out the parameters of a module and return it.
unimatch/ldm_unet/attention.py:65
Method__call__
(self, img1, img2, flow, occlusion=None)
dataloader/flow/transforms.py:131
Method__call__
(self, img1, img2, flow, valid)
dataloader/flow/transforms.py:273
Method__call__
(self, sample)
dataloader/stereo/transforms.py:14
Method__call__
(self, sample)
dataloader/stereo/transforms.py:26
Method__call__
(self, sample)
dataloader/stereo/transforms.py:54
Method__call__
(self, sample)
dataloader/stereo/transforms.py:71
Method__call__
(self, sample)
dataloader/stereo/transforms.py:123
Method__call__
(self, sample)
dataloader/stereo/transforms.py:135
Method__call__
(self, sample)
dataloader/stereo/transforms.py:144
Method__call__
(self, sample)
dataloader/stereo/transforms.py:161
Method__call__
(self, sample)
dataloader/stereo/transforms.py:183
Method__call__
(self, sample)
dataloader/stereo/transforms.py:205
Method__call__
(self, sample)
dataloader/stereo/transforms.py:227
Method__call__
(self, sample)
dataloader/stereo/transforms.py:249
Method__call__
(self, sample)
dataloader/stereo/transforms.py:271
Method__call__
(self, sample)
dataloader/stereo/transforms.py:308
Method__call__
(self, sample)
dataloader/stereo/transforms.py:345
Method__call__
(self, sample)
dataloader/stereo/transforms.py:364
Method__call__
(self, sample)
dataloader/stereo/transforms.py:382
Method__call__
(self, sample)
dataloader/stereo/transforms.py:415
Method__call__
(self, sample)
dataloader/depth/augmentation.py:13
Method__call__
(self, sample)
dataloader/depth/augmentation.py:22
Method__call__
(self, sample)
dataloader/depth/augmentation.py:43
Method__call__
(self, sample)
dataloader/depth/augmentation.py:70
Method__call__
(self, sample)
dataloader/depth/augmentation.py:99
Method__call__
(self, sample)
dataloader/depth/augmentation.py:135
Method__call__
(self, sample)
dataloader/depth/augmentation.py:178
Method__call__
(self, sample)
dataloader/depth/augmentation.py:187
Method__call__
(self, sample)
dataloader/depth/augmentation.py:201
Method__call__
(self, sample)
dataloader/depth/augmentation.py:219
Method__call__
(self, sample)
dataloader/depth/augmentation.py:237
Method__call__
(self, sample)
dataloader/depth/augmentation.py:255
Method__call__
(self, sample)
dataloader/depth/augmentation.py:273
Method__getitem__
(self, index)
dataloader/flow/datasets.py:43
Method__getitem__
(self, index)
dataloader/stereo/datasets.py:46
Method__getitem__
(self, i)
dataloader/depth/datasets.py:63
Method__getitem__
(self, i)
dataloader/depth/datasets.py:177
Method__init__
(self, dims, mode='sintel', padding_factor=8)
utils/utils.py:9
Method__init__
(self, lr_scheduler, summary_writer, summary_freq=100, s
utils/logger.py:8
Method__init__
(self, crop_size, min_scale=-0.2, max_scale=0.5, do_flip=True, no_eraser_aug=True,
dataloader/flow/transforms.py:8
Method__init__
(self, crop_size, min_scale=-0.2, max_scale=0.5, do_flip=False, no_eraser_aug=True,
dataloader/flow/transforms.py:153
Method__init__
(self, aug_params=None, sparse=False, load_occlusion=False, vkitti2=False,
dataloader/flow/datasets.py:20
← previousnext →101–200 of 377, ranked by callers