MCPcopy Create free account

hub / github.com/Tian-one/tps-inbetween / functions

Functions191 in github.com/Tian-one/tps-inbetween

⨍ Functions 191◇ Types & classes 31↳ Endpoints 1
Method__init__
(self, data_path, size=(512 ,512), xN=2, gray=True, match_dir=None, pre_process=False, train=False)
datasets/ml240.py:16
Method__init__
(self, data_range=1.0, **kwargs)
util/metrics.py:33
Method__init__
( self, block=1024, convert_dog=True, k=1.6, epsilon=0.01, kernel_factor=4, clip=False,
util/metrics.py:165
Method__init__
( self, block=1024, convert_dog=True, t=2.0, sigma=1.0, k=1.6, epsilon=0.01, kernel_fa
util/metrics.py:206
Method__len__
(self)
datasets/ml240.py:59
Method__prepare__
(name, bases, **kwds)
model/gluestick/models/base_model.py:13
Method_forward
(self, data)
model/gluestick/models/wireframe.py:141
Method_forward
(self, data)
model/gluestick/models/two_view_pipeline.py:116
Method_forward
(self, data)
model/gluestick/models/gluestick.py:96
Method_forward
(self, data)
model/gluestick/models/superpoint.py:111
Method_init
(self, conf)
model/gluestick/models/wireframe.py:104
Method_init
(self, conf)
model/gluestick/models/two_view_pipeline.py:91
Method_init
(self, conf)
model/gluestick/models/gluestick.py:43
Method_init
(self, conf)
model/gluestick/models/superpoint.py:85
Functionbatch_chamfer_distance
(gt, pred, block=1024, return_more=False)
util/sketcher.py:157
Functionbatch_chamfer_distance_p_w
(gt, pred, block=1024, return_more=False)
util/metrics.py:131
Functionbatch_chamfer_distance_t_w
(gt, pred, block=1024, return_more=False)
util/metrics.py:118
Functionbatch_dog
(img, t=1.0, sigma=1.0, k=1.6, epsilon=0.01, kernel_factor=4, clip=True)
util/sketcher.py:126
Functionbatch_hausdorff_distance
(gt, pred, block=1024, return_more=False)
util/sketcher.py:185
Functionbinarization
(img, th=0.95)
demo.py:66
Functioncd_score
(img1, img2)
util/metrics.py:246
Functioncd_score_w
(img1, img2)
util/metrics.py:263
Methodcompute
(self)
util/metrics.py:27
Methodcompute
(self)
util/metrics.py:44
Methodcompute
(self)
util/metrics.py:183
Methodcompute
(self)
util/metrics.py:229
Functioncompute_dtm
compute the distance transform map of foreground in binary mask input: segmentation, shape = (batch_size, x, y, z) output: the foreground
util/utils.py:327
Functiondistance_transform_norm
(image: torch.Tensor, kernel_size: int = 3, h: float = 0.35, norm_factor: float=15)
util/utils.py:320
Methoddone
(self)
logger/logger.py:134
Functionearth_mover_distance
(img1, img2)
util/metrics.py:284
Methodend_log
(self, n_div, train=True)
logger/logger.py:104
Functionflow_vis
(flow, radius=0.2)
generate_matches.py:21
Functionflow_vis
(flow, radius=0.2)
demo.py:19
Methodforward
(self, x0, x1, matches_path=None, middle=False, aug=False)
model/tpsinbet.py:39
Methodforward
(self, x)
model/tpsinbet.py:135
Methodforward
(self, x, flow)
model/tpsinbet.py:148
Methodforward
(self, img0, img1, flow0, flow1, warped_img0, warped_img1, mask, c0, c1)
model/tpsinbet.py:177
Methodforward
(self, x)
model/IFNet.py:32
Methodforward
(self, x, flow)
model/IFNet.py:46
Methodforward
(self, img0, img1, warped_img0, warped_img1, mask, flow, c0, c1)
model/IFNet.py:74
Methodforward
(self, x, flow, scale)
model/IFNet.py:106
Methodforward
(self, I0t, I1t, scale=[4,2,1], timestep=0.5)
model/IFNet.py:130
Methodforward
Check the data and call the _forward method of the child model.
model/gluestick/models/base_model.py:97
Methodforward
(self, kpts, scores)
model/gluestick/models/gluestick.py:322
Methodforward
(self, endpoints, scores)
model/gluestick/models/gluestick.py:333
Methodforward
(self, query, key, value)
model/gluestick/models/gluestick.py:364
Methodforward
(self, x, source)
model/gluestick/models/gluestick.py:384
Methodforward
(self, desc0, desc1)
model/gluestick/models/gluestick.py:396
Methodforward
(self, ldesc0, ldesc1, line_enc0, line_enc1, lines_junc_idx0, lines_junc_idx1)
model/gluestick/models/gluestick.py:459
Methodforward
(self, desc0, desc1, line_enc0, line_enc1, lines_junc_idx0, lines_junc_idx1)
model/gluestick/models/gluestick.py:513
Methodfreeze_bn
(module)
model/gluestick/models/base_model.py:89
Functionget_val_loader
(data_path, batch_size, shuffle, match_dir=None, img_size=(512, 512), num_workers=0, xN=6, train=False, gray=T
datasets/ml240.py:114
Functiongrid_from_flow
(flow, size)
util/utils.py:203
Functionkeep_best_kp_subset
Keep the top num_selected best keypoints.
model/gluestick/models/two_view_pipeline.py:45
Functionkeep_quadrant_kp_subset
Keep only keypoints in one of the four quadrant of the image.
model/gluestick/models/two_view_pipeline.py:20
Functionkeep_random_kp_subset
Keep a random subset of keypoints.
model/gluestick/models/two_view_pipeline.py:35
Methodloss
(self, pred, data)
model/gluestick/models/wireframe.py:270
Methodloss
(self, pred, data)
model/gluestick/models/two_view_pipeline.py:154
Methodloss
To be implemented by the child class.
model/gluestick/models/base_model.py:119
Methodloss
(self, pred, data)
model/gluestick/models/gluestick.py:282
Methodloss
(self, pred, data)
model/gluestick/models/superpoint.py:220
Functionmake_model
(args)
model/tpsinbet.py:112
Methodmetrics
(self, pred, data)
model/gluestick/models/wireframe.py:273
Methodmetrics
(self, pred, data)
model/gluestick/models/two_view_pipeline.py:167
Methodmetrics
To be implemented by the child class.
model/gluestick/models/base_model.py:124
Methodmetrics
(self, pred, data)
model/gluestick/models/gluestick.py:285
Methodmetrics
(self, pred, data)
model/gluestick/models/superpoint.py:223
Functionnorm_match_torch
match shape: (B, N, 2)
util/utils.py:246
Methodplot_loss_log
(self, epoch)
logger/logger.py:110
Methodplot_psnr_log
(self, epoch)
logger/logger.py:122
Methodprocess_siamese
(data, i)
model/gluestick/models/two_view_pipeline.py:118
Methodrecursive_key_check
(expected, given)
model/gluestick/models/base_model.py:99
Methodreport_log
(self, item, train=True)
logger/logger.py:98
Methodsave_images
(self, filename, save_list, epoch, exposure)
logger/logger.py:61
Methodstart_log
(self, train=True)
logger/logger.py:92
Functiontensor2numpy
(tensor, rgb_range=1.)
generate_matches.py:36
Functiontensor2numpy
(tensor, rgb_range=1.)
demo.py:34
Functiontps_grid
Compute a thin-plate-spline grid from parameters for sampling. Params ------ theta: Nx(T+3)x2 tensor Batch size N, T+3 model
util/utils.py:174
Functiontps_grid_to_remap
Convert a dense grid to OpenCV's remap compatible maps. Params ------ grid : HxWx2 array Normalized flow field coordinates as
util/utils.py:103
Functiontps_sparse
(theta, ctrl, xy)
util/utils.py:214
Functiontps_theta_from_points_torch
(c_src, c_dst, reduced=False)
util/utils.py:92
Methodtrain
(self, mode=True)
model/gluestick/models/base_model.py:86
Functionuniform_grid
Uniform grid coordinates. Params ------ shape : tuple HxW defining the number of height and width dimension of the grid
util/utils.py:58
Methodupdate
(self, preds: torch.Tensor, target: torch.Tensor)
util/metrics.py:39
Methodupdate
(self, preds: torch.Tensor, target: torch.Tensor)
util/metrics.py:177
Methodupdate
(self, preds: torch.Tensor, target: torch.Tensor)
util/metrics.py:187
Methodupdate
(self, preds: torch.Tensor, target: torch.Tensor)
util/metrics.py:196
Methodupdate
(self, preds: torch.Tensor, target: torch.Tensor)
util/metrics.py:221
Methodwarp_zero
(self, img, flow, size)
model/tpsinbet.py:33
Methodwrite_log
(self, log)
logger/logger.py:51
Methodz
(x, c, theta)
util/utils.py:48
← previous101–191 of 191, ranked by callers