MCPcopy Create free account

hub / github.com/Parskatt/DKM / functions

Functions179 in github.com/Parskatt/DKM

⨍ Functions 179◇ Types & classes 43
Method__call__
(self, x, y, eps=1e-6)
dkm/models/deprecated/corr_channels.py:11
Method__getitem__
(self, pair_idx)
dkm/datasets/scannet.py:78
Method__getitem__
(self, pair_idx)
dkm/datasets/megadepth.py:78
Method__init__
(self, *transforms, align_corners=True)
dkm/utils/transforms.py:9
Method__init__
(self)
dkm/utils/utils.py:145
Method__init__
(self)
dkm/utils/utils.py:168
Method__init__
(self, size, mode=InterpolationMode.BICUBIC)
dkm/utils/utils.py:179
Method__init__
(self, mean, std)
dkm/utils/utils.py:191
Method__init__
(self, transforms)
dkm/utils/utils.py:204
Method__init__
(self, dir=None, name="tmp")
dkm/checkpointing/checkpoint.py:9
Method__init__
(self, data_root, scene_info, ht = 384, wt = 512, min_overlap=0., shake_t = 0, rot_prob=0.)
dkm/datasets/scannet.py:23
Method__init__
(self, data_root = 'data/scannet')
dkm/datasets/scannet.py:127
Method__init__
( self, data_root, scene_info, ht=384, wt=512, min_overlap=0.0
dkm/datasets/megadepth.py:16
Method__init__
(self, data_root="data/megadepth")
dkm/datasets/megadepth.py:137
Method__init__
(self, pretrained=False, high_res = False, weights = None, dilation = None, freeze_bn = True, anti_aliased = F
dkm/models/encoders.py:31
Method__init__
(self, pretrained=False, high_res = False, weights = None)
dkm/models/encoders.py:75
Method__init__
(self, pretrained=False, high_res = False, weights = None)
dkm/models/encoders.py:113
Method__init__
( self, in_dim=6, hidden_dim=16, out_dim=2, dw=False, kernel_s
dkm/models/dkm.py:14
Method__init__
(self, T, learn_temperature=False)
dkm/models/dkm.py:129
Method__init__
(self, in_channels, out_channels)
dkm/models/dkm.py:150
Method__init__
(self, in_channels, out_channels, kernel_size=3)
dkm/models/dkm.py:176
Method__init__
( self, internal_dim, feat_input_modules, pred_input_modules, rrb_d_di
dkm/models/dkm.py:208
Method__init__
(self, resnet)
dkm/models/dkm.py:376
Method__init__
( self, embedding_decoder, gps, proj, conv_refiner, transformers = None, detach=False, scales="all", p
dkm/models/dkm.py:406
Method__init__
( self, encoder, decoder, h=384, w=512, use_contrastive_loss =
dkm/models/dkm.py:540
Method__init__
(self)
dkm/models/deprecated/local_corr.py:595
Method__init__
(self)
dkm/models/deprecated/corr_channels.py:8
Method__init__
( self, robust=True, center_coords=False, scale_normalize=False, ce_we
dkm/losses/depth_match_regression_loss.py:9
Method__init__
(self, data_root="data/megadepth", scene_names = None)
dkm/benchmarks/megadepth1500_benchmark.py:9
Method__init__
(self, data_root="data/megadepth", h = 384, w = 512, num_samples = 2000, device=None)
dkm/benchmarks/megadepth_dense_benchmark.py:10
Method__init__
(self, data_root="data/scannet")
dkm/benchmarks/scannet_benchmark.py:10
Method__init__
(self, dataset_path)
dkm/benchmarks/hpatches_dense_benchmark.py:10
Method__init__
(self, dataset_path)
dkm/benchmarks/hpatches_sequences_homog_benchmark.py:14
Method__init__
(self, dataset_path)
dkm/benchmarks/deprecated/hpatches_sequences_dense_benchmark.py:15
Method__init__
(self, data_root="data/yfcc100m_test")
dkm/benchmarks/deprecated/yfcc100m_benchmark.py:11
Method__len__
(self)
dkm/datasets/scannet.py:51
Method__len__
(self)
dkm/datasets/megadepth.py:63
Method__repr__
(self)
dkm/utils/utils.py:140
Method__repr__
(self)
dkm/utils/utils.py:151
Method__repr__
(self)
dkm/utils/utils.py:161
Method__repr__
(self)
dkm/utils/utils.py:174
Method__repr__
(self)
dkm/utils/utils.py:186
Method__repr__
(self)
dkm/utils/utils.py:199
Method__repr__
(self)
dkm/utils/utils.py:212
Methodbackward
(self, gradOutput)
dkm/models/deprecated/local_corr.py:486
Methodbenchmark
(self, model, ransac_thr = 3.,hp240=False)
dkm/benchmarks/hpatches_dense_benchmark.py:18
Methodbenchmark
(self, model)
dkm/benchmarks/deprecated/hpatches_sequences_dense_benchmark.py:53
Methodbenchmark
(self, model, r=2)
dkm/benchmarks/deprecated/yfcc100m_benchmark.py:20
Functionfast_kde
(x, std = 0.1, kernel_size = 9, dilation = 3, padding = 9//2, stride = 1)
dkm/utils/kde.py:5
Methodforward
(self, x)
dkm/models/encoders.py:10
Methodforward
(self, x)
dkm/models/encoders.py:45
Methodforward
(self, x)
dkm/models/encoders.py:83
Methodforward
(self, x)
dkm/models/encoders.py:121
Methodforward
Computes the relative refining displacement in pixels for a given image x,y and a coarse flow-field between them Args: x ([type])
dkm/models/dkm.py:77
Methodforward
(self, x)
dkm/models/dkm.py:162
Methodforward
(self, x)
dkm/models/dkm.py:198
Methodforward
(self, embeddings, feats, context, key)
dkm/models/dkm.py:247
Methodforward
(self, x, y, **kwargs)
dkm/models/dkm.py:342
Methodforward
(self, x)
dkm/models/dkm.py:379
Methodforward
(self, f1, f2, upsample = False, dense_flow = None, dense_certainty = None)
dkm/models/dkm.py:456
Methodforward
(self, first, second)
dkm/models/deprecated/local_corr.py:266
Methodforward
(self, input, second)
dkm/models/deprecated/local_corr.py:420
Methodforward
(self, tensorFirst, tensorSecond)
dkm/models/deprecated/local_corr.py:586
Methodforward
(self, tensorFirst, tensorSecond)
dkm/models/deprecated/local_corr.py:598
Methodforward
Computes the relative refining displacement in pixels for a given image x,y and a coarse flow-field between them Args: x ([type])
dkm/models/deprecated/local_corr.py:603
Methodforward
(self, x, y, **kwargs)
dkm/models/deprecated/corr_channels.py:28
Methodforward
[summary] Args: out ([type]): [description] batch ([type]): [description] Returns: [type]: [desc
dkm/losses/depth_match_regression_loss.py:82
Functionrotate_pose_inplane
(i_T_w, rot)
dkm/utils/utils.py:44
Functionscale_intrinsics
(K, scales)
dkm/utils/utils.py:60
Functiontest_hpatches
(model)
experiments/dkm/test_DKMv3_outdoor.py:46
Functiontest_mega_8_scenes
(model)
experiments/dkm/test_DKMv3_outdoor.py:17
Functiontest_st_pauls
(model)
experiments/dkm/test_DKMv3_outdoor.py:60
Functionto_cpu
(batch)
dkm/utils/utils.py:324
Functionto_homogeneous
(points)
dkm/utils/utils.py:65
Methodtrain
(self, mode=True)
dkm/models/encoders.py:63
Methodtrain
(self, mode=True)
dkm/models/encoders.py:104
Methodtrain
(self, mode=True)
dkm/models/encoders.py:142
Functiontrain_k_epochs
( start_epoch, end_epoch, dataloader, model, objective, optimizer, lr_scheduler )
dkm/train/train.py:56
Methodupsample_preds
(self, flow, certainty, query, support)
dkm/models/dkm.py:420
← previous101–179 of 179, ranked by callers