MCPcopy Create free account

hub / github.com/drinkingcoder/NeuralMarker / functions

Functions183 in github.com/drinkingcoder/NeuralMarker

⨍ Functions 183◇ Types & classes 40
Method__exit__
(self, *args)
core/raft.py:21
Method__getitem__
(self, index)
core/datasets.py:43
Method__getitem__
(self, index)
core/datasets.py:201
Method__getitem__
(self, index)
core/datasets.py:218
Method__init__
(self, args)
flow_estimator.py:9
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
core/extractor.py:7
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
core/extractor.py:61
Method__init__
(self, output_dim=128, norm_fn='batch', dropout=0.0)
core/extractor.py:196
Method__init__
(self, pretrained=True)
core/encoders.py:76
Method__init__
(self, args)
core/datasets.py:33
Method__init__
(self, args, split='train')
core/datasets.py:132
Method__init__
(self, args, split='train')
core/datasets.py:188
Method__init__
(self, args, split)
core/datasets.py:208
Method__init__
(self, enabled)
core/biraft.py:17
Method__init__
(self, args)
core/biraft.py:28
Method__init__
(self, fmap1, fmap2, num_levels=4, radius=4)
core/corr.py:13
Method__init__
(self, fmap1, fmap2, num_levels=4, radius=4)
core/corr.py:65
Method__init__
(self, input_dim=128, hidden_dim=256)
core/update.py:7
Method__init__
(self, input_dim=128, hidden_dim=256)
core/update.py:17
Method__init__
(self, hidden_dim=128, input_dim=192+128)
core/update.py:44
Method__init__
(self, args)
core/update.py:73
Method__init__
(self, args)
core/update.py:90
Method__init__
(self, args, hidden_dim=96)
core/update.py:110
Method__init__
(self, args, hidden_dim=128, input_dim=128)
core/update.py:125
Method__init__
(self, enabled)
core/raft.py:17
Method__init__
(self, args)
core/raft.py:26
Method__init__
(self, args)
core/utils/augmentor.py:7
Method__init__
(self, dims, mode='sintel')
core/utils/utils.py:9
Method__init__
(self,)
core/utils/forward_warp.py:10
Method__init__
(self, tps_grid_size=3, tps_reg_factor=0, out_h=240, out_w=240, offset_factor=1.0,
core/utils/transformation.py:18
Method__init__
(self, geometric_model='affine', tps_grid_size=3, tps_reg_factor=0, out_h=240, out_w=240, off
core/utils/transformation.py:80
Method__init__
(self, use_cuda=True, geometric_model='affine', crop_factor=9 / 16, output_size=(240, 240), p
core/utils/transformation.py:152
Method__init__
(self, use_cuda=True, crop_factor=9 / 16, output_size=(240, 240), padding_factor=0.5)
core/utils/transformation.py:234
Method__init__
(self, use_cuda=True, crop_factor=9 / 16, output_size=(240, 240), padding_factor=0.5)
core/utils/transformation.py:276
Method__init__
(self, use_cuda=True, crop_factor=9 / 16, output_size=(240, 240), padding_factor=0.5)
core/utils/transformation.py:336
Method__init__
(self, out_h=240, out_w=240, out_ch=3, use_cuda=True)
core/utils/transformation.py:386
Method__init__
(self, out_h=240, out_w=240, use_cuda=True)
core/utils/transformation.py:403
Method__init__
(self, out_h=240, out_w=240, use_cuda=True)
core/utils/transformation.py:443
Method__init__
(self, out_h=240, out_w=240, use_regular_grid=True, grid_size=3, reg_factor=0, use_cuda=True)
core/utils/transformation.py:524
Method__len__
(self)
core/datasets.py:128
Method__len__
(self)
core/datasets.py:234
Method__len___
(self)
core/datasets.py:204
Functioncompute_epipolar_distance_loss
(args, flow_ab, Fm)
core/loss.py:48
Methodcompute_params
(self, layer=2)
core/encoders.py:50
Functionflow_to_image
Expects a two dimensional flow image of shape. Args: flow_uv (np.ndarray): Flow UV image of shape [H,W,2] clip_flow (float,
core/utils/flow_viz.py:109
Methodforward
(self, x)
core/extractor.py:48
Methodforward
(self, x)
core/extractor.py:107
Methodforward
(self, x)
core/extractor.py:168
Methodforward
(self, x)
core/extractor.py:244
Methodforward
(self, x, data=None, layer=2)
core/encoders.py:38
Methodforward
(self, x, data=None, layer=2)
core/encoders.py:80
Methodforward
Estimate optical flow between pair of frames
core/biraft.py:92
Methodforward
(self, x)
core/update.py:13
Methodforward
(self, x)
core/update.py:23
Methodforward
(self, h, x)
core/update.py:33
Methodforward
(self, h, x)
core/update.py:55
Methodforward
(self, flow, corr)
core/update.py:81
Methodforward
(self, flow, corr)
core/update.py:99
Methodforward
(self, net, inp, corr, flow)
core/update.py:116
Methodforward
(self, net, inp, corr, flow, upsample=True)
core/update.py:140
Methodforward
-img: image (N, C, H, W) -flo: optical flow (N, 2, H, W) elements of flo is in [0, H] and [0, W] for dx, dy
core/utils/forward_warp.py:13
Methodforward
(self, theta)
core/utils/transformation.py:392
Methodforward
(self, theta)
core/utils/transformation.py:421
Methodforward
(self, theta)
core/utils/transformation.py:461
Methodforward
(self, theta)
core/utils/transformation.py:561
Functionforward_interpolate
(flow)
core/utils/utils.py:110
Methodfreeze_bn
(self)
core/biraft.py:65
Methodfreeze_bn
(self)
core/raft.py:63
Functionget_demo_harsh_lighting_args
()
config.py:91
Functionget_gt_correspondence_mask
Computes the mask of valid flows (that do not match to a pixel outside of the image).
core/utils/utils.py:316
Functionimage_editing
(data_root: str, marker_path: str, scene_path:str, src
harsh_lighting_utils.py:75
Functionimage_forward_warp
Input: image: HxWx3 numpy flow: HxWx2 torch.Tensor Output: outImg: HxWx3 numpy
core/utils/utils.py:366
Functionpytorch_foward_interpolate
(flow)
core/utils/utils.py:26
FunctionreadDispKITTI
(filename)
core/utils/frame_utils.py:109
FunctionreadFlowKITTI
(filename)
core/utils/frame_utils.py:102
Functionread_gen
(file_name, pil=False)
core/utils/frame_utils.py:123
Functionreverse_mapping
Input: flow: B x 2 x H x W torch.Tensor.cuda, represent the flow from A to B grid_map: B x H x W x 2 torch.Tensor.cuda, grid mapp
core/utils/utils.py:207
Functionreverse_mask
Input: mask: B x 1 x H x W torch.Tensor.cuda, represent the mask of A grid_map: B x H x W x 2 torch.Tensor.cuda, grid mapping fro
core/utils/utils.py:222
Functionshow_input_images
(scene_images, target_images)
eval_utils.py:6
Functionsrgb_to_rgb
(srgb)
harsh_lighting_utils.py:11
Functiontrain
(rank, world_size, args)
train.py:70
Functionvisualization
(args, model, step=None, split=None)
evaluation_FM.py:49
FunctionwriteFlowKITTI
(filename, uv)
core/utils/frame_utils.py:116
← previous101–183 of 183, ranked by callers