MCPcopy Create free account

hub / github.com/drinkingcoder/FlowFormer-Official / functions

Functions336 in github.com/drinkingcoder/FlowFormer-Official

⨍ Functions 336◇ Types & classes 109
Method__call__
(self, img1, img2, flow, valid)
core/utils/augmentor.py:274
Method__call__
(self, input, target)
core/utils/flow_transforms.py:27
Method__call__
(self, inputs, target)
core/utils/flow_transforms.py:52
Method__call__
(self, inputs, target)
core/utils/flow_transforms.py:126
Method__call__
(self, inputs, target)
core/utils/flow_transforms.py:244
Method__call__
(self, inputs, target)
core/utils/flow_transforms.py:321
Method__call__
(self, inputs, target)
core/utils/flow_transforms.py:418
Method__enter__
(self)
core/raft.py:18
Method__exit__
(self, *args)
core/raft.py:20
Method__getitem__
(self, index)
core/datasets.py:34
Method__getitem__
(self, index)
core/utils/datasets.py:39
Method__getitem__
(self, index)
core/utils/datasets.py:299
Method__init__
(self, dims, mode='sintel')
evaluate_FlowFormer_tile.py:34
Method__init__
(self)
train_FlowFormer.py:39
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, aug_params=None, sparse=False)
core/datasets.py:19
Method__init__
(self, aug_params=None, split='training', root='datasets/Sintel', dstype='clean')
core/datasets.py:103
Method__init__
(self, aug_params=None, split='train', root='datasets/FlyingChairs_release/data')
core/datasets.py:122
Method__init__
(self, aug_params=None, root='datasets/FlyingThings3D', dstype='frames_cleanpass', split='training')
core/datasets.py:138
Method__init__
(self, aug_params=None, split='training', root='datasets/KITTI')
core/datasets.py:163
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:63
Method__init__
(self, input_dim=128, hidden_dim=256)
core/update.py:7
Method__init__
(self, hidden_dim=128, input_dim=192+128)
core/update.py:34
Method__init__
(self, args)
core/update.py:63
Method__init__
(self, args)
core/update.py:80
Method__init__
(self, args, hidden_dim=96)
core/update.py:100
Method__init__
(self, args, hidden_dim=128, input_dim=128)
core/update.py:115
Method__init__
(self, enabled)
core/raft.py:16
Method__init__
(self, args)
core/raft.py:25
Method__init__
Args: max_shape (tuple): for 1/8 featmap, the max length of 256 corresponds to 2048 pixels
core/position_encoding.py:43
Method__init__
Args: max_shape (tuple): for 1/8 featmap, the max length of 256 corresponds to 2048 pixels
core/position_encoding.py:76
Method__init__
(self, crop_size, min_scale=-0.2, max_scale=0.5, do_flip=True, pwc_aug=False)
core/utils/augmentor.py:16
Method__init__
(self, crop_size, min_scale=-0.2, max_scale=0.5, do_flip=False)
core/utils/augmentor.py:148
Method__init__
(self, dims, mode='sintel')
core/utils/utils.py:9
Method__init__
(self, model, scheduler, cfg)
core/utils/logger.py:5
Method__init__
(self, aug_params=None, sparse=False)
core/utils/datasets.py:23
Method__init__
(self, aug_params=None, split='test', root='datasets/Sintel', dstype='clean')
core/utils/datasets.py:105
Method__init__
(self, aug_params=None, split='training', root='datasets/Sintel', dstype='clean')
core/utils/datasets.py:123
Method__init__
(self, aug_params=None, split='train', root='datasets/FlyingChairs_release/data')
core/utils/datasets.py:164
Method__init__
(self, aug_params=None, root='datasets/FlyingThings3D', dstype='frames_cleanpass')
core/utils/datasets.py:189
Method__init__
(self, aug_params=None, split='training', root='datasets/KITTI')
core/utils/datasets.py:226
Method__init__
(self, num_steps, crop_size, log_dir, root='datasets/')
core/utils/datasets.py:264
Method__init__
(self, co_transforms)
core/utils/flow_transforms.py:24
Method__init__
(self, size, order=1)
core/utils/flow_transforms.py:42
Method__init__
(self, crop, scale=None, rot=None, trans=None, squeeze=None, schedule_coeff=1, order=1, black=False)
core/utils/flow_transforms.py:71
Method__init__
(self, schedule_coeff=1)
core/utils/flow_transforms.py:241
Method__init__
(self, lmult_pow =[0.4, 0,-0.2], lmult_mult =[0.4, 0,0, ],
core/utils/flow_transforms.py:254
Method__init__
(self, noise = 0.06, gamma = 0.02, brightness = 0.02,
core/utils/flow_transforms.py:405
Method__init__
(self, pretrained=True)
core/FlowFormer/encoders.py:64
Method__init__
(self, dim, dropout = 0.)
core/FlowFormer/common.py:322
Method__init__
(self, dim, heads, num_kv_tokens, cfg, rpe_bias=None, use_rpe=False)
core/FlowFormer/common.py:354
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
core/FlowFormer/LatentCostFormer/cnn.py:9
Method__init__
(self, in_planes, planes, norm_fn='group', stride=1)
core/FlowFormer/LatentCostFormer/cnn.py:63
Method__init__
(self, input_dim=3, output_dim=128, norm_fn='batch', dropout=0.0)
core/FlowFormer/LatentCostFormer/cnn.py:121
Method__init__
(self, output_dim=128, norm_fn='batch', dropout=0.0)
core/FlowFormer/LatentCostFormer/cnn.py:206
Method__init__
(self, in_dim, out_dim, inter_dim, depth, stride=1)
core/FlowFormer/LatentCostFormer/cnn.py:280
Method__init__
(self, input_dim=128, hidden_dim=256)
core/FlowFormer/LatentCostFormer/cnn.py:297
Method__init__
(self, hidden_dim=128, input_dim=192+128)
core/FlowFormer/LatentCostFormer/cnn.py:324
Method__init__
(self, args)
core/FlowFormer/LatentCostFormer/cnn.py:353
Method__init__
(self, args)
core/FlowFormer/LatentCostFormer/cnn.py:373
Method__init__
(self, args, hidden_dim=128, input_dim=128)
core/FlowFormer/LatentCostFormer/cnn.py:413
Method__init__
(self, args)
core/FlowFormer/LatentCostFormer/cnn.py:437
Method__init__
(self, args, hidden_dim=128)
core/FlowFormer/LatentCostFormer/cnn.py:452
Method__init__
(self, args)
core/FlowFormer/LatentCostFormer/cnn.py:471
Method__init__
(self, args)
core/FlowFormer/LatentCostFormer/cnn.py:489
Method__init__
(self, args)
core/FlowFormer/LatentCostFormer/cnn.py:535
Method__init__
(self, qk_dim, v_dim, query_token_dim, tgt_token_dim, add_flow_token=True, num_heads=8, attn_drop=0., proj_dro
core/FlowFormer/LatentCostFormer/decoder.py:31
Method__init__
(self, dim, cfg)
core/FlowFormer/LatentCostFormer/decoder.py:94
Method__init__
(self, cfg)
core/FlowFormer/LatentCostFormer/decoder.py:120
Method__init__
( self, *, args, dim, max_pos_size = 100, heads = 4, d
core/FlowFormer/LatentCostFormer/gma.py:35
Method__init__
( self, args, dim, heads = 4, dim_head = 128, )
core/FlowFormer/LatentCostFormer/gma.py:80
Method__init__
(self, input_dim=128, hidden_dim=256)
core/FlowFormer/LatentCostFormer/gru.py:6
Method__init__
(self, hidden_dim=128, input_dim=192+128)
core/FlowFormer/LatentCostFormer/gru.py:33
Method__init__
(self, args)
core/FlowFormer/LatentCostFormer/gru.py:62
Method__init__
(self, args, hidden_dim=128, input_dim=128)
core/FlowFormer/LatentCostFormer/gru.py:86
Method__init__
(self, args, hidden_dim=128)
core/FlowFormer/LatentCostFormer/gru.py:111
Method__init__
(self, dim, depth=4)
core/FlowFormer/LatentCostFormer/convnext.py:8
Method__init__
(self, dim, layer_scale_init_value=1e-6)
core/FlowFormer/LatentCostFormer/convnext.py:35
Method__init__
(self, dim, fn)
core/FlowFormer/LatentCostFormer/mlpmixer.py:7
Method__init__
(self, dim, heads)
core/FlowFormer/LatentCostFormer/attention.py:10
Method__init__
(self, dim, heads)
core/FlowFormer/LatentCostFormer/attention.py:108
Method__init__
(self, cfg)
core/FlowFormer/LatentCostFormer/transformer.py:20
Method__init__
(self, dim, cfg, num_heads=8, attn_drop=0., proj_drop=0., drop_path=0., dropout=0.)
core/FlowFormer/LatentCostFormer/encoder.py:84
Method__init__
(self, dim, cfg, num_heads=8, attn_drop=0., proj_drop=0., drop_path=0., dropout=0.)
core/FlowFormer/LatentCostFormer/encoder.py:109
Method__init__
(self, dim, cfg, num_heads=8, attn_drop=0., proj_drop=0., drop_path=0., dropout=0.)
core/FlowFormer/LatentCostFormer/encoder.py:144
Method__init__
(self, qk_dim, v_dim, query_token_dim, tgt_token_dim, num_heads=8, attn_drop=0., proj_drop=0., drop_path=0., d
core/FlowFormer/LatentCostFormer/encoder.py:197
Method__init__
(self, cfg)
core/FlowFormer/LatentCostFormer/encoder.py:245
Method__init__
(self, cfg)
core/FlowFormer/LatentCostFormer/encoder.py:311
Method__init__
(self, dim, num_heads=8, attn_drop=0., proj_drop=0., ws=1, cfg=None, vert_c_dim=0)
core/FlowFormer/LatentCostFormer/twins.py:67
Method__init__
(self, dim, num_heads=8, attn_drop=0., proj_drop=0., ws=1, cfg=None)
core/FlowFormer/LatentCostFormer/twins.py:156
Method__init__
(self, dim, num_heads=8, attn_drop=0., proj_drop=0., ws=1, vert_c_dim=0)
core/FlowFormer/LatentCostFormer/twins.py:232
Method__init__
(self, dim, num_heads=8, attn_drop=0., proj_drop=0., sr_ratio=1, vert_c_dim=0)
core/FlowFormer/LatentCostFormer/twins.py:309
Method__init__
(self, dim, num_heads=8, attn_drop=0., proj_drop=0., ws=1)
core/FlowFormer/LatentCostFormer/twins.py:397
Method__init__
(self, dim, num_heads=8, attn_drop=0., proj_drop=0., sr_ratio=1)
core/FlowFormer/LatentCostFormer/twins.py:458
Method__init__
(self, dim, num_heads=8, attn_drop=0., proj_drop=0., sr_ratio=1)
core/FlowFormer/LatentCostFormer/twins.py:531
Method__init__
(self, dim, num_heads=8, attn_drop=0., proj_drop=0., ws=1)
core/FlowFormer/LatentCostFormer/twins.py:588
Method__init__
(self, dim, num_heads=8, attn_drop=0., proj_drop=0., sr_ratio=1)
core/FlowFormer/LatentCostFormer/twins.py:636
← previousnext →101–200 of 336, ranked by callers