MCPcopy Create free account

hub / github.com/aojunzz/NM-sparsity / functions

Functions200 in github.com/aojunzz/NM-sparsity

⨍ Functions 200◇ Types & classes 55
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
RAFT/core/extractor.py:7
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
RAFT/core/extractor.py:61
Method__init__
(self, output_dim=128, norm_fn='batch', dropout=0.0)
RAFT/core/extractor.py:196
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
RAFT/core/sparse_extractor.py:12
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
RAFT/core/sparse_extractor.py:66
Method__init__
(self, output_dim=128, norm_fn='batch', dropout=0.0)
RAFT/core/sparse_extractor.py:201
Method__init__
(self, args)
RAFT/core/sparse_raft.py:25
Method__init__
(self, aug_params=None, sparse=False)
RAFT/core/datasets.py:19
Method__init__
(self, aug_params=None, split='training', root='datasets/Sintel', dstype='clean')
RAFT/core/datasets.py:103
Method__init__
(self, aug_params=None, split='train', root='xxxxx/FlyingChairs_release/data')
RAFT/core/datasets.py:122
Method__init__
(self, aug_params=None, root='datasets/FlyingThings3D', dstype='frames_cleanpass')
RAFT/core/datasets.py:138
Method__init__
(self, aug_params=None, split='training', root='datasets/KITTI')
RAFT/core/datasets.py:162
Method__init__
(self, optimizer, last_epoch=-1, verbose=False)
RAFT/core/lr_scheduler.py:14
Method__init__
(self, o)
RAFT/core/lr_scheduler.py:129
Method__init__
(self, optimizer, max_lr, total_steps=None,
RAFT/core/lr_scheduler.py:260
Method__init__
(self, input_dim=128, hidden_dim=256)
RAFT/core/sparse_update.py:11
Method__init__
(self, hidden_dim=128, input_dim=192+128)
RAFT/core/sparse_update.py:38
Method__init__
(self, args)
RAFT/core/sparse_update.py:67
Method__init__
(self, args)
RAFT/core/sparse_update.py:84
Method__init__
(self, args, hidden_dim=96)
RAFT/core/sparse_update.py:104
Method__init__
(self, args, hidden_dim=128, input_dim=128)
RAFT/core/sparse_update.py:119
Method__init__
(self, fmap1, fmap2, num_levels=4, radius=4)
RAFT/core/corr.py:13
Method__init__
(self, fmap1, fmap2, num_levels=4, radius=4)
RAFT/core/corr.py:64
Method__init__
(self, input_dim=128, hidden_dim=256)
RAFT/core/update.py:7
Method__init__
(self, hidden_dim=128, input_dim=192+128)
RAFT/core/update.py:34
Method__init__
(self, args)
RAFT/core/update.py:63
Method__init__
(self, args)
RAFT/core/update.py:80
Method__init__
(self, args, hidden_dim=96)
RAFT/core/update.py:100
Method__init__
(self, args, hidden_dim=128, input_dim=128)
RAFT/core/update.py:115
Method__init__
(self, enabled)
RAFT/core/raft.py:16
Method__init__
(self, args)
RAFT/core/raft.py:25
Method__init__
(self, crop_size, min_scale=-0.2, max_scale=0.5, do_flip=True)
RAFT/core/utils/augmentor.py:16
Method__init__
(self, crop_size, min_scale=-0.2, max_scale=0.5, do_flip=False)
RAFT/core/utils/augmentor.py:123
Method__init__
(self, dims, mode='sintel')
RAFT/core/utils/utils.py:9
Method__init__
(self)
classification/train_imagenet.py:281
Method__init__
(self, inplanes, planes, stride=1, downsample=None, N=2, M=4)
classification/models/resnet.py:30
Method__init__
(self, inplanes, planes, stride=1, downsample=None, N=2, M=4)
classification/models/resnet.py:62
Method__len__
(self)
devkit/dataset/imagenet_dataset.py:24
Method__len__
(self)
RAFT/core/datasets.py:98
Method__repr__
(self)
devkit/sparse_ops/syncbn_layer.py:92
Method__rmul__
(self, v)
RAFT/core/datasets.py:93
Method_annealing_cos
Cosine anneal from `start` to `end` as pct goes from 0.0 to 1.0.
RAFT/core/lr_scheduler.py:389
Method_annealing_linear
Linearly anneal from `start` to `end` as pct goes from 0.0 to 1.0.
RAFT/core/lr_scheduler.py:394
Methodbackward
(ctx, grad_output)
devkit/sparse_ops/sparse_ops.py:60
Methodbackward
(ctx, grad_outdata)
devkit/sparse_ops/syncbn_layer.py:49
Functioncorr_backward
RAFT/alt_cuda_corr/correlation.cpp:36
Functioncorr_forward
RAFT/alt_cuda_corr/correlation.cpp:23
Functioncreate_kitti_submission
Create submission for the Sintel leaderboard
RAFT/evaluate.py:54
Functioncreate_sintel_submission
Create submission for the Sintel leaderboard
RAFT/evaluate.py:22
Functionfetch_dataloader
Create the data loader for the corresponding trainign set
RAFT/core/datasets.py:199
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,
RAFT/core/utils/flow_viz.py:109
Methodforward
(ctx, weight, N, M, decay = 0.0002)
devkit/sparse_ops/sparse_ops.py:13
Methodforward
(ctx, weight, N, M, decay = 0.0002)
devkit/sparse_ops/sparse_ops.py:40
Methodforward
(self, x)
devkit/sparse_ops/sparse_ops.py:80
Methodforward
(self, x)
devkit/sparse_ops/sparse_ops.py:104
Methodforward
(ctx, in_data, scale_data, shift_data, running_mean, running_var, eps, momentum, training)
devkit/sparse_ops/syncbn_layer.py:11
Methodforward
(self, in_data)
devkit/sparse_ops/syncbn_layer.py:97
Methodforward
(self, x)
RAFT/core/extractor.py:48
Methodforward
(self, x)
RAFT/core/extractor.py:107
Methodforward
(self, x)
RAFT/core/extractor.py:168
Methodforward
(self, x)
RAFT/core/extractor.py:244
Methodforward
(self, x)
RAFT/core/sparse_extractor.py:53
Methodforward
(self, x)
RAFT/core/sparse_extractor.py:112
Methodforward
(self, x)
RAFT/core/sparse_extractor.py:173
Methodforward
(self, x)
RAFT/core/sparse_extractor.py:249
Methodforward
Estimate optical flow between pair of frames
RAFT/core/sparse_raft.py:86
Methodforward
(self, x)
RAFT/core/sparse_update.py:17
Methodforward
(self, h, x)
RAFT/core/sparse_update.py:27
Methodforward
(self, h, x)
RAFT/core/sparse_update.py:49
Methodforward
(self, flow, corr)
RAFT/core/sparse_update.py:75
Methodforward
(self, flow, corr)
RAFT/core/sparse_update.py:93
Methodforward
(self, net, inp, corr, flow)
RAFT/core/sparse_update.py:110
Methodforward
(self, net, inp, corr, flow, upsample=True)
RAFT/core/sparse_update.py:131
Methodforward
(self, x)
RAFT/core/update.py:13
Methodforward
(self, h, x)
RAFT/core/update.py:23
Methodforward
(self, h, x)
RAFT/core/update.py:45
Methodforward
(self, flow, corr)
RAFT/core/update.py:71
Methodforward
(self, flow, corr)
RAFT/core/update.py:89
Methodforward
(self, net, inp, corr, flow)
RAFT/core/update.py:106
Methodforward
(self, net, inp, corr, flow, upsample=True)
RAFT/core/update.py:127
Methodforward
(self, x)
classification/models/resnet.py:41
Methodforward
(self, x)
classification/models/resnet.py:76
Methodforward
(self, x)
classification/models/resnet.py:142
Methodfreeze_bn
(self)
RAFT/core/sparse_raft.py:58
Methodget_lr
(self)
RAFT/core/lr_scheduler.py:398
Functionload_state_epoch
(model_dir, model, epoch)
devkit/core/utils.py:45
FunctionreadDispKITTI
(filename)
RAFT/core/utils/frame_utils.py:109
FunctionreadFlowKITTI
(filename)
RAFT/core/utils/frame_utils.py:102
Functionread_gen
(file_name, pil=False)
RAFT/core/utils/frame_utils.py:123
Functionresnet101
(**kwargs)
classification/models/resnet.py:176
Functionresnet152
(**kwargs)
classification/models/resnet.py:181
Functionresnet18
(**kwargs)
classification/models/resnet.py:161
Functionresnet34
(**kwargs)
classification/models/resnet.py:166
Functionresnet50
(**kwargs)
classification/models/resnet.py:171
Methodupdate
(self, i, epoch)
devkit/core/lr_scheduler.py:66
Methodwith_counter
(method)
RAFT/core/lr_scheduler.py:37
Methodwrapper
(*args, **kwargs)
RAFT/core/lr_scheduler.py:51
FunctionwriteFlow
Write optical flow to file. If v is None, uv is assumed to contain both u and v channels, stacked in depth. Original code by Deqing
RAFT/core/utils/frame_utils.py:70
FunctionwriteFlowKITTI
(filename, uv)
RAFT/core/utils/frame_utils.py:116
Methodwrite_dict
(self, results)
RAFT/train.py:130
← previous101–200 of 200, ranked by callers