MCPcopy Create free account

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

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

↓ 14 callersFunctionwarp
(tenInput, tenFlow)
model/warplayer.py:8
↓ 12 callersFunctionconv
(in_planes, out_planes, kernel_size=3, stride=1, padding=1, dilation=1)
model/IFNet.py:13
↓ 8 callersFunctionplot_images
Plot a set of images horizontally. Args: imgs: a list of NumPy or PyTorch images, RGB (H, W, 3) or mono (H, W). titles: a list of
model/gluestick/drawing.py:7
↓ 6 callersMethod__init__
(self, feature_dim, layer_type, skip_init)
model/gluestick/models/gluestick.py:390
↓ 6 callersFunctionbatch_dog
(img, t=1.0, sigma=1.0, k=1.6, epsilon=0.01, kernel_factor=4, clip=True)
util/utils.py:349
↓ 6 callersFunctionbatch_edt
(img, block=1024)
util/metrics.py:51
↓ 6 callersFunctionget_model
(name)
model/gluestick/__init__.py:25
↓ 6 callersFunctionrgb2gray
(rgb)
util/metrics.py:240
↓ 5 callersMethodsave
(self, trainer, epoch, is_best)
logger/logger.py:55
↓ 4 callersFunctionMLP
(channels, do_bn=True)
model/gluestick/models/gluestick.py:289
↓ 4 callersMethod__init__
(self)
model/IFNet.py:62
↓ 4 callersFunctionbatch_edt
(img, block=1024)
util/sketcher.py:52
↓ 4 callersFunctionbatch_to_np
(batch)
model/gluestick/__init__.py:52
↓ 4 callersFunctiondeconv
(in_planes, out_planes, kernel_size=4, stride=2, padding=1)
model/tpsinbet.py:123
↓ 4 callersFunctiondeconv
(in_planes, out_planes, kernel_size=4, stride=2, padding=1)
model/IFNet.py:20
↓ 4 callersFunctionis_inside_img
(pts: torch.Tensor, img_shape: Tuple[int, int])
model/gluestick/geometry.py:88
↓ 4 callersFunctionnormalize_keypoints
(kpts, shape_or_size)
model/gluestick/models/gluestick.py:302
↓ 4 callersFunctionnumpy_image_to_torch
Normalize the image tensor and reorder the dimensions.
model/gluestick/__init__.py:30
↓ 4 callersFunctionrgb2sketch
(img, black_threshold)
util/metrics.py:234
↓ 3 callersMethod__init__
(self, indim=8, outdim=1, c=16)
model/tpsinbet.py:165
↓ 3 callersMethod__init__
(self, window_size=11, **kwargs)
util/metrics.py:14
↓ 3 callersFunctionbatch_chamfer_distance_p
(gt, pred, block=1024, return_more=False)
util/metrics.py:98
↓ 3 callersFunctionbatch_chamfer_distance_t
(gt, pred, block=1024, return_more=False)
util/metrics.py:88
↓ 3 callersMethodfit
(c, lambd=0., reduced=False)
util/utils.py:18
↓ 3 callersFunctionmax_pool
(x)
model/gluestick/models/superpoint.py:26
↓ 3 callersFunctionto_homogeneous
Convert N-dimensional points to homogeneous coordinates. Args: points: torch.Tensor or numpy.ndarray with size (..., N). Returns:
model/gluestick/geometry.py:7
↓ 2 callersMethod_get_line_matches
(self, ldesc0, ldesc1, lines_junc_idx0, lines_junc_idx1, final_proj)
model/gluestick/models/gluestick.py:252
↓ 2 callersMethod_get_matches
(self, scores_mat)
model/gluestick/models/gluestick.py:237
↓ 2 callersFunctionarange_like
(x, dim)
model/gluestick/models/gluestick.py:557
↓ 2 callersFunctionconv
(in_planes, out_planes, kernel_size=3, stride=1, padding=1, dilation=1)
model/tpsinbet.py:116
↓ 2 callersMethodd
(a, b)
util/utils.py:40
↓ 2 callersMethoddetect_lsd_lines
(self, x, max_n_lines=None)
model/gluestick/models/wireframe.py:108
↓ 2 callersMethodget_endpoint_attention
(self, ldesc, line_enc, lines_junc_idx)
model/gluestick/models/gluestick.py:432
↓ 2 callersMethodget_endpoint_update
(self, ldesc, line_enc, lines_junc_idx)
model/gluestick/models/gluestick.py:419
↓ 2 callersFunctionimg_open_torch
(img_pth, size=None, gray=True)
demo.py:54
↓ 2 callersMethodimg_open_torch
(self, img_pth, size, gray)
datasets/ml240.py:74
↓ 2 callersFunctionlog_double_softmax
(scores, bin_score)
model/gluestick/models/gluestick.py:543
↓ 2 callersFunctionnorm_flow
(flow, h, w)
generate_matches.py:42
↓ 2 callersFunctionnorm_flow
(flow, h, w)
demo.py:47
↓ 2 callersFunctionnormalize_points
(points, image_height, image_width)
util/metrics.py:279
↓ 2 callersFunctionplot_color_line_matches
Plot line matches for existing images with multiple colors. Args: lines: list of ndarrays of size (N, 2, 2). correct_matches: bool
model/gluestick/drawing.py:133
↓ 2 callersFunctionplot_keypoints
Plot keypoints for existing images. Args: kpts: list of ndarrays of size (N, 2). colors: string, or list of list of tuples (one fo
model/gluestick/drawing.py:42
↓ 2 callersFunctionplot_lines
Plot lines and endpoints for existing images. Args: lines: list of ndarrays of size (N, 2, 2). colors: string, or list of list of
model/gluestick/drawing.py:98
↓ 2 callersFunctionplot_matches
Plot matches for a pair of existing images. Args: kpts0, kpts1: corresponding keypoints of size (N, 2). color: color of each match
model/gluestick/drawing.py:56
↓ 2 callersFunctionsample_descriptors
(keypoints, descriptors, s)
model/gluestick/models/superpoint.py:54
↓ 2 callersFunctiontps
Evaluate the thin-plate-spline (TPS) surface at xy locations arranged in a grid. The TPS surface is a minimum bend interpolation surface defined b
util/utils.py:125
↓ 2 callersMethodu
(r)
util/utils.py:44
↓ 2 callersMethodupdate
(self, preds: torch.Tensor, target: torch.Tensor)
util/metrics.py:20
↓ 2 callersMethodwarp
(self, img, flow, size)
model/tpsinbet.py:27
↓ 1 callersMethod_forward
To be implemented by the child class.
model/gluestick/models/base_model.py:114
↓ 1 callersMethod_init
To be implemented by the child class.
model/gluestick/models/base_model.py:109
↓ 1 callersFunctionattention
(query, key, value)
model/gluestick/models/gluestick.py:347
↓ 1 callersFunctionbatch_chamfer_distance
(gt, pred, block=1024, return_more=False)
util/metrics.py:83
↓ 1 callersFunctionbatch_chamfer_distance_p
(gt, pred, block=1024, return_more=False)
util/sketcher.py:173
↓ 1 callersFunctionbatch_chamfer_distance_t
(gt, pred, block=1024, return_more=False)
util/sketcher.py:162
↓ 1 callersFunctionbatch_chamfer_distance_w
(gt, pred, block=1024, return_more=False)
util/metrics.py:108
↓ 1 callersFunctionbatch_hausdorff_distance
(gt, pred, block=1024, return_more=False)
util/metrics.py:146
↓ 1 callersFunctioncupy_launch
(func, kernel)
util/sketcher.py:11
↓ 1 callersFunctiondistance_transform
r"""Approximates the Manhattan distance transform of images using cascaded convolution operations. The value at each pixel in the output represen
util/utils.py:256
↓ 1 callersMethodendpoints_pooling
(segs, all_descriptors, img_shape)
model/gluestick/models/wireframe.py:256
↓ 1 callersFunctionfrom_homogeneous
Remove the homogeneous dimension of N-dimensional points. Args: points: torch.Tensor or numpy.ndarray with size (..., N+1). Returns:
model/gluestick/geometry.py:24
↓ 1 callersFunctionget_class
Get the class object which inherits from BaseClass and is defined in the module named mod_name, child of base_path.
model/gluestick/__init__.py:9
↓ 1 callersFunctionget_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:107
↓ 1 callersMethodget_video_idx
(self, idx)
datasets/ml240.py:65
↓ 1 callersFunctionimg2gif
(imgs_pth, save_pth, dur=0.1)
demo.py:72
↓ 1 callersFunctionlines_to_wireframe
Given a set of lines, their score and dense descriptors, merge close-by endpoints and compute a wireframe defined by its junctions an
model/gluestick/models/wireframe.py:11
↓ 1 callersFunctionmain
()
generate_matches.py:50
↓ 1 callersFunctionmain
()
demo.py:78
↓ 1 callersFunctionmain
()
model/gluestick/run.py:14
↓ 1 callersFunctionmain
()
model/gluestick/genflow.py:28
↓ 1 callersFunctionmap_tensor
(input_, func)
model/gluestick/__init__.py:41
↓ 1 callersFunctionmatch2flow
(linematch0, linematch1, h, w)
model/gluestick/genflow.py:14
↓ 1 callersFunctionnp2Tensor
(img, rgb_range=1, n_colors=1)
demo.py:40
↓ 1 callersFunctionnp2Tensor
(img, rgb_range=1, n_colors=1)
datasets/ml240.py:121
↓ 1 callersFunctionremove_borders
(keypoints, scores, b, h, w)
model/gluestick/models/superpoint.py:40
↓ 1 callersFunctionseg_equation
(segs)
model/gluestick/geometry.py:77
↓ 1 callersFunctionshrink_segs_to_img
Shrink an array of segments to fit inside the image. :param segs: The tensor of segments with shape (N, 2, 2) :param img_shape: The image
model/gluestick/geometry.py:93
↓ 1 callersFunctionsimple_nms
Perform non maximum suppression on the heatmap using max-pooling. This method does not suppress contiguous points that have the same score. Ar
model/gluestick/models/superpoint.py:18
↓ 1 callersFunctionskew_symmetric
Create a skew-symmetric matrix from a (batched) vector of size (..., 3).
model/gluestick/geometry.py:34
↓ 1 callersFunctiontop_k_keypoints
(keypoints, scores, k)
model/gluestick/models/superpoint.py:47
↓ 1 callersFunctiontps_theta_from_points
(c_src, c_dst, reduced=False)
util/utils.py:79
↓ 1 callersFunctionwarp_lines_torch
:param lines: A tensor of shape (B, N, 2, 2) where B is the batch size, N the number of lines. :param H: The homography used to convert the l
model/gluestick/geometry.py:150
↓ 1 callersFunctionwarp_points_torch
Warp a list of points with the INVERSE of the given homography. The inverse is used to be coherent with tf.contrib.image.transform Argume
model/gluestick/geometry.py:51
FunctionT_to_E
Convert batched poses (..., 4, 4) to batched essential matrices.
model/gluestick/geometry.py:46
Method__getitem__
(self, idx)
datasets/ml240.py:85
Method__init__
(self, args)
logger/logger.py:15
Method__init__
(self, args)
model/tpsinbet.py:12
Method__init__
(self, in_planes, out_planes, stride=2)
model/tpsinbet.py:130
Method__init__
(self, indim=1, c=16)
model/tpsinbet.py:141
Method__init__
(self, in_planes, out_planes, stride=2)
model/IFNet.py:27
Method__init__
(self)
model/IFNet.py:39
Method__init__
(self, in_planes, c=64)
model/IFNet.py:88
Method__init__
(self)
model/IFNet.py:121
Method__init__
Perform some logic and call the _init method of the child model.
model/gluestick/models/base_model.py:60
Method__init__
(self, feature_dim, layers)
model/gluestick/models/gluestick.py:317
Method__init__
(self, feature_dim, layers)
model/gluestick/models/gluestick.py:328
Method__init__
(self, h, d_model)
model/gluestick/models/gluestick.py:355
Method__init__
(self, num_dim, num_heads, skip_init=False)
model/gluestick/models/gluestick.py:374
Method__init__
(self, feature_dim, line_attention=False)
model/gluestick/models/gluestick.py:410
Method__init__
(self, feature_dim, layer_types, checkpointed=False, skip=False, inter_supervision=None, num_
model/gluestick/models/gluestick.py:498
next →1–100 of 191, ranked by callers