MCPcopy Create free account

hub / github.com/Parskatt/DKM / functions

Functions179 in github.com/Parskatt/DKM

⨍ Functions 179◇ Types & classes 43
↓ 24 callersMethodreshape
(self, x)
dkm/models/dkm.py:313
↓ 14 callersMethodbuild_scenes
(self, split="train", min_overlap=0.0, **kwargs)
dkm/datasets/megadepth.py:144
↓ 10 callersMethodbenchmark
(self, model, model_name = None)
dkm/benchmarks/scannet_benchmark.py:13
↓ 10 callersFunctioncupy_kernel
(strFunction, objectVariables)
dkm/models/deprecated/local_corr.py:211
↓ 10 callersFunctioncupy_launch
(strFunction, strKernel)
dkm/models/deprecated/local_corr.py:257
↓ 9 callersMethodmatch
( self, im1_path, im2_path, *args, batched=False, device = Non
dkm/models/dkm.py:653
↓ 9 callersFunctiontrain_k_steps
( n_0, k, dataloader, model, objective, optimizer, lr_scheduler, progress_bar=True )
dkm/train/train.py:14
↓ 8 callersMethod__init__
( self, kernel, T=1, learn_temperature=False, only_attention=False,
dkm/models/dkm.py:260
↓ 6 callersFunctionDKMv2
(pretrained=True, version="outdoor", resolution = "low", **kwargs)
dkm/models/deprecated/build_model.py:153
↓ 6 callersMethodsample
( self, dense_matches, dense_certainty, num=10000, )
dkm/models/dkm.py:585
↓ 6 callersMethodtrain
(self, mode=True)
dkm/models/dkm.py:397
↓ 6 callersMethodweight_scenes
(self, concat_dataset, alpha=0.5)
dkm/datasets/megadepth.py:167
↓ 5 callersFunctionDKM
(pretrained=True, version="mega_synthetic", device=None)
dkm/models/deprecated/build_model.py:20
↓ 5 callersFunctionget_tuple_transform_ops
(resize=None, normalize=True, unscale=False)
dkm/utils/utils.py:112
↓ 4 callersFunctionDKMv3_outdoor
Loads DKMv3 outdoor weights, uses internal resolution of (540, 720) by default resolution can be changed by setting model.h_resized, model.w_
dkm/models/model_zoo/__init__.py:11
↓ 4 callersFunctionpose_auc
(errors, thresholds)
dkm/utils/utils.py:89
↓ 3 callersMethod__init__
(self, pretrained=False)
dkm/models/encoders.py:7
↓ 3 callersMethodbuild_scenes
(self, split = 'train', min_overlap=0., **kwargs)
dkm/datasets/scannet.py:132
↓ 3 callersFunctioncompute_pose_error
(T_0to1, R, t)
dkm/utils/utils.py:80
↓ 3 callersFunctionestimate_pose
(kpts0, kpts1, K0, K1, norm_thresh, conf=0.99999)
dkm/utils/utils.py:13
↓ 3 callersFunctionlocal_corr
(pretrained=True, version="mega_synthetic")
dkm/models/deprecated/build_model.py:304
↓ 3 callersMethodtrain
(self, mode=True)
dkm/models/encoders.py:23
↓ 2 callersFunctionDKMv3
(weights, h, w, symmetric = True, sample_mode= "threshold_balanced", device = None, **kwargs)
dkm/models/model_zoo/DKMv3.py:8
↓ 2 callersFunctionbaseline
(pretrained=True, version="mega_synthetic")
dkm/models/deprecated/build_model.py:547
↓ 2 callersMethodbenchmark
(self, model)
dkm/benchmarks/megadepth1500_benchmark.py:26
↓ 2 callersMethodbenchmark
(self, model, batch_size=8)
dkm/benchmarks/megadepth_dense_benchmark.py:49
↓ 2 callersFunctioncompute_relative_pose
(R1, t1, R2, t2)
dkm/utils/utils.py:336
↓ 2 callersFunctioncorr_channels
(pretrained=True, version="mega_synthetic")
dkm/models/deprecated/build_model.py:436
↓ 2 callersMethodcreate_block
( self, in_dim, out_dim, dw=False, kernel_size=5, )
dkm/models/dkm.py:52
↓ 2 callersMethodextract_backbone_features
(self, batch, batched = True, upsample = True)
dkm/models/dkm.py:574
↓ 2 callersMethodforward
(self, batch, batched = True, upsample = False)
dkm/models/dkm.py:620
↓ 2 callersMethodforward_symmetric
(self, batch, upsample = False, batched = True)
dkm/models/dkm.py:637
↓ 2 callersFunctionget_depth_tuple_transform_ops
(resize=None, normalize=True, unscale=False)
dkm/utils/utils.py:105
↓ 2 callersFunctionget_pose
(calib)
dkm/utils/utils.py:331
↓ 2 callersFunctionlinear
(pretrained=True, version="mega_synthetic")
dkm/models/deprecated/build_model.py:658
↓ 2 callersMethodload_depth
(self, depth_ref, crop=None)
dkm/datasets/scannet.py:45
↓ 2 callersMethodload_depth
(self, depth_ref, crop=None)
dkm/datasets/megadepth.py:59
↓ 2 callersMethodload_im
(self, im_ref, crop=None)
dkm/datasets/scannet.py:41
↓ 2 callersMethodload_im
(self, im_ref, crop=None)
dkm/datasets/megadepth.py:55
↓ 2 callersFunctionlocal_correlation
( feature0, feature1, local_radius, padding_mode="zeros", flow = None )
dkm/utils/local_correlation.py:5
↓ 2 callersMethodread_scannet_pose
Read ScanNet's Camera2World pose and transform it to World2Camera. Returns: pose_w2c (np.ndarray): (4, 4)
dkm/datasets/scannet.py:61
↓ 2 callersMethodreshape
(self, x)
dkm/models/deprecated/corr_channels.py:25
↓ 2 callersFunctionrotate_intrinsic
(K, n)
dkm/utils/utils.py:38
↓ 2 callersMethodscale_intrinsic
(self, K, wi, hi)
dkm/datasets/scannet.py:54
↓ 2 callersMethodscale_intrinsic
(self, K, wi, hi)
dkm/datasets/megadepth.py:66
↓ 2 callersFunctionto_cuda
(batch)
dkm/utils/utils.py:317
↓ 2 callersMethodto_pixel_coordinates
(self, matches, H_A, W_A, H_B, W_B)
dkm/models/dkm.py:647
↓ 2 callersFunctiontrain_step
(train_batch, model, objective, optimizer, **kwargs)
dkm/train/train.py:5
↓ 2 callersFunctionwarp_kpts
Warp kpts0 from I0 to I1 with depth, K and Rt Also check covisibility and depth consistency. Depth is consistent if relative error < 0.2 (hard
dkm/utils/utils.py:222
↓ 2 callersMethodweight_scenes
(self, concat_dataset, alpha=.5)
dkm/datasets/scannet.py:141
↓ 1 callersFunctionDKMv3_indoor
Loads DKMv3 indoor weights, uses internal resolution of (480, 640) by default Resolution can be changed by setting model.h_resized, model.w_r
dkm/models/model_zoo/__init__.py:27
↓ 1 callersFunctionFunctionCorrelation
(reference_features, query_features)
dkm/models/deprecated/local_corr.py:578
↓ 1 callersMethod__init__
(self)
dkm/models/deprecated/local_corr.py:583
↓ 1 callersMethod__init__
( self, )
dkm/models/deprecated/corr_channels.py:19
↓ 1 callersFunctionangle_error_mat
(R1, R2)
dkm/utils/utils.py:69
↓ 1 callersFunctionangle_error_vec
(v1, v2)
dkm/utils/utils.py:75
↓ 1 callersMethodapply_transform
(self, x, M, mode="bilinear")
dkm/utils/transforms.py:28
↓ 1 callersMethodbackward
(self, gradOutput)
dkm/models/deprecated/local_corr.py:329
↓ 1 callersMethodbenchmark
(self, model, model_name = None)
dkm/benchmarks/hpatches_sequences_homog_benchmark.py:56
↓ 1 callersMethodconvert_coordinates
(self, query_coords, query_to_support, wq, hq, wsup, hsup)
dkm/benchmarks/hpatches_sequences_homog_benchmark.py:32
↓ 1 callersMethodconvert_coordinates
(self, query_coords, query_to_support, wq, hq, wsup, hsup)
dkm/benchmarks/deprecated/hpatches_sequences_dense_benchmark.py:20
↓ 1 callersMethoddense_depth_loss
[summary] Args: dense_certainty ([type]): [description] prob ([type]): [description] eps ([type], optiona
dkm/losses/depth_match_regression_loss.py:61
↓ 1 callersMethodgenerate_parameters
(self, batch_shape: torch.Size)
dkm/utils/transforms.py:36
↓ 1 callersMethodgeometric_dist
[summary] Args: H ([type]): [description] scale ([type]): [description] Returns: [type]: [descri
dkm/losses/depth_match_regression_loss.py:28
↓ 1 callersMethodgeometric_dist
(self, depth1, depth2, T_1to2, K1, K2, dense_matches)
dkm/benchmarks/megadepth_dense_benchmark.py:20
↓ 1 callersMethodget_local_cov
(self, cov)
dkm/models/dkm.py:286
↓ 1 callersFunctionget_model
(pretrained_backbone=True, resolution = "low", **kwargs)
experiments/dkm/train_DKMv3_indoor.py:20
↓ 1 callersFunctionget_model
(pretrained_backbone=True, resolution = "low", **kwargs)
experiments/dkm/train_DKMv3_outdoor.py:19
↓ 1 callersMethodget_placeholder_flow
(self, b, h, w, device)
dkm/models/dkm.py:442
↓ 1 callersMethodget_pos_enc
(self, y)
dkm/models/dkm.py:326
↓ 1 callersMethodinside_image
(self,x,w,h)
dkm/benchmarks/hpatches_dense_benchmark.py:15
↓ 1 callersMethodinside_image
(self, x, w, h)
dkm/benchmarks/deprecated/hpatches_sequences_dense_benchmark.py:47
↓ 1 callersFunctionkde
(x, std = 0.1, device=None)
dkm/utils/kde.py:17
↓ 1 callersFunctionnumpy_to_pil
Args: x: Assumed to be of shape (h,w,c)
dkm/utils/utils.py:294
↓ 1 callersMethodproject_to_basis
(self, x)
dkm/models/dkm.py:316
↓ 1 callersMethodrand_shake
(self, *things)
dkm/datasets/megadepth.py:71
↓ 1 callersMethodrandom_perspective_generator
r"""Get parameters for ``perspective`` for a random perspective transform. Args: batch_size (int): the tensor batch size.
dkm/utils/transforms.py:50
↓ 1 callersMethodread_scannet_intrinsic
Read ScanNet's intrinsic matrix and return the 3x3 matrix.
dkm/datasets/scannet.py:72
↓ 1 callersFunctionrun
(gpus=1)
experiments/deprecated/train_mega.py:15
↓ 1 callersFunctionrun
(gpus=1)
experiments/deprecated/train_mega_synthetic.py:27
↓ 1 callersFunctionrun
(gpus=1)
experiments/deprecated/dkmv2/train_indoor.py:18
↓ 1 callersFunctionrun
(gpus=1)
experiments/deprecated/dkmv2/train_outdoor.py:15
↓ 1 callersMethodscales
(self)
dkm/models/dkm.py:244
↓ 1 callersFunctiontensor_to_pil
(x, unnormalize=False)
dkm/utils/utils.py:307
↓ 1 callersFunctiontest_mega1500
(model)
experiments/dkm/test_DKMv3_outdoor.py:3
↓ 1 callersFunctiontest_scannet
(model)
experiments/dkm/test_DKMv3_indoor.py:4
↓ 1 callersFunctiontrain
(args)
experiments/dkm/train_DKMv3_indoor.py:168
↓ 1 callersFunctiontrain
(args)
experiments/dkm/train_DKMv3_outdoor.py:171
↓ 1 callersFunctiontrain_epoch
( dataloader=None, model=None, objective=None, optimizer=None, lr_scheduler=None, epoc
dkm/train/train.py:32
FunctionFunctionCorrelationTranspose
(reference_features, query_features)
dkm/models/deprecated/local_corr.py:590
Method__call__
(self, x, mode="bilinear")
dkm/utils/transforms.py:13
Method__call__
(self, im)
dkm/utils/utils.py:132
Method__call__
(self, im_tuple)
dkm/utils/utils.py:148
Method__call__
(self, im)
dkm/utils/utils.py:158
Method__call__
(self, im_tuple)
dkm/utils/utils.py:171
Method__call__
(self, im_tuple)
dkm/utils/utils.py:183
Method__call__
(self, im_tuple)
dkm/utils/utils.py:196
Method__call__
(self, im_tuple)
dkm/utils/utils.py:207
Method__call__
( self, model, optimizer, lr_scheduler, n, )
dkm/checkpointing/checkpoint.py:14
Method__call__
(self, x, y, eps=1e-6)
dkm/models/dkm.py:137
next →1–100 of 179, ranked by callers