MCPcopy Create free account

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

Functions336 in github.com/drinkingcoder/FlowFormer-Official

⨍ Functions 336◇ Types & classes 109
Method__init__
(self, dim, num_heads=8, attn_drop=0., proj_drop=0., sr_ratio=1)
core/FlowFormer/LatentCostFormer/twins.py:683
Method__init__
(self, dim, num_heads, mlp_ratio=4., drop=0., attn_drop=0., drop_path=0., act_layer=nn.GELU,
core/FlowFormer/LatentCostFormer/twins.py:729
Method__init__
(self, in_chans, embed_dim=768, stride=1)
core/FlowFormer/LatentCostFormer/twins.py:793
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)
core/FlowFormer/LatentCostFormer/twins.py:815
Method__init__
( self, img_size=224, patch_size=4, in_chans=3, num_classes=1000, embed_dims=(64, 128, 256, 512),
core/FlowFormer/LatentCostFormer/twins.py:843
Method__len__
(self)
core/datasets.py:98
Method__len__
(self)
core/utils/datasets.py:101
Method__len__
(self)
core/utils/datasets.py:296
Method__rmul__
(self, v)
core/datasets.py:93
Method__rmul__
(self, v)
core/utils/datasets.py:96
Method__rmul__
(self, v)
core/utils/datasets.py:291
Method_init_weights
(self, m)
core/FlowFormer/LatentCostFormer/cnn.py:508
Method_init_weights
(self, m)
core/FlowFormer/LatentCostFormer/cnn.py:554
Method_init_weights
(self, m)
core/FlowFormer/LatentCostFormer/twins.py:898
Methodcompute_params
(self, layer=2)
core/FlowFormer/encoders.py:38
Methodcompute_params
(self)
core/FlowFormer/LatentCostFormer/cnn.py:170
Methodcompute_params
(self)
core/FlowFormer/LatentCostFormer/convnext.py:17
Methodcompute_params
(self)
core/FlowFormer/LatentCostFormer/mlpmixer.py:38
Methodcompute_params
(self)
core/FlowFormer/LatentCostFormer/encoder.py:136
Methodcompute_params
(self)
core/FlowFormer/LatentCostFormer/encoder.py:188
Functioncompute_weight
(hws, image_shape, patch_size=TRAIN_SIZE, sigma=1.0, wtype='gaussian')
visualize_flow.py:49
Functioncorr_backward
alt_cuda_corr/correlation.cpp:36
Functioncorr_forward
alt_cuda_corr/correlation.cpp:23
Functioncreate_kitti_submission
Create submission for the Sintel leaderboard
evaluate_FlowFormer_tile.py:133
Functioncreate_sintel_submission
Create submission for the Sintel leaderboard
evaluate_FlowFormer_tile.py:89
Functionfetch_dataloader
Create the data loader for the corresponding trainign set
core/datasets.py:200
Functionfetch_dataloader
Create the data loader for the corresponding trainign set
core/utils/datasets.py:424
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)
core/update.py:13
Methodforward
(self, h, x)
core/update.py:23
Methodforward
(self, h, x)
core/update.py:45
Methodforward
(self, flow, corr)
core/update.py:71
Methodforward
(self, flow, corr)
core/update.py:89
Methodforward
(self, net, inp, corr, flow)
core/update.py:106
Methodforward
(self, net, inp, corr, flow, upsample=True)
core/update.py:127
Methodforward
Args: x: [N, C, H, W]
core/position_encoding.py:31
Methodforward
Args: x: [N, C, H, W]
core/position_encoding.py:62
Methodforward
Args: x: [N, C, H, W]
core/position_encoding.py:85
Methodforward
(self, x, data=None, layer=2)
core/FlowFormer/encoders.py:19
Methodforward
(self, x, data=None, layer=2)
core/FlowFormer/encoders.py:68
Methodforward
(self, x)
core/FlowFormer/common.py:331
Methodforward
(self, x)
core/FlowFormer/common.py:344
Methodforward
(self, Q, K, V, rpe_bias = None)
core/FlowFormer/common.py:402
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/cnn.py:50
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/cnn.py:109
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/cnn.py:178
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/cnn.py:254
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/cnn.py:289
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/cnn.py:303
Methodforward
(self, h, x)
core/FlowFormer/LatentCostFormer/cnn.py:313
Methodforward
(self, h, x)
core/FlowFormer/LatentCostFormer/cnn.py:335
Methodforward
(self, flow, corr)
core/FlowFormer/LatentCostFormer/cnn.py:362
Methodforward
(self, flow, feat, context1=None)
core/FlowFormer/LatentCostFormer/cnn.py:396
Methodforward
(self, net, inp, corr, flow, upsample=True)
core/FlowFormer/LatentCostFormer/cnn.py:425
Methodforward
(self, motion_features)
core/FlowFormer/LatentCostFormer/cnn.py:445
Methodforward
(self, latent, flow)
core/FlowFormer/LatentCostFormer/cnn.py:463
Methodforward
(self, rpe_tokens)
core/FlowFormer/LatentCostFormer/cnn.py:483
Methodforward
(self, x, tgt, size)
core/FlowFormer/LatentCostFormer/cnn.py:526
Methodforward
(self, x, tgt, size)
core/FlowFormer/LatentCostFormer/cnn.py:572
Methodforward
query_coord [B, 2, H1, W1]
core/FlowFormer/LatentCostFormer/decoder.py:57
Methodforward
x: [B*H1*W1, 1, C] memory: [B*H1*W1, H2'*W2', C] coords1 [B, 2, H2, W2] size: B, C, H1, W1
core/FlowFormer/LatentCostFormer/decoder.py:103
Methodforward
cost_maps - B*H1*W1, cost_heads_num, H2, W2 coords - B, 2, H1, W1
core/FlowFormer/LatentCostFormer/decoder.py:124
Methodforward
memory: [B*H1*W1, H2'*W2', C] context: [B, D, H1, W1]
core/FlowFormer/LatentCostFormer/decoder.py:205
Methodforward
(self, q)
core/FlowFormer/LatentCostFormer/gma.py:20
Methodforward
(self, fmap)
core/FlowFormer/LatentCostFormer/gma.py:54
Methodforward
(self, attn, fmap)
core/FlowFormer/LatentCostFormer/gma.py:102
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/gru.py:12
Methodforward
(self, h, x)
core/FlowFormer/LatentCostFormer/gru.py:22
Methodforward
(self, h, x)
core/FlowFormer/LatentCostFormer/gru.py:44
Methodforward
(self, flow, corr)
core/FlowFormer/LatentCostFormer/gru.py:75
Methodforward
(self, net, inp, corr, flow, upsample=True)
core/FlowFormer/LatentCostFormer/gru.py:98
Methodforward
(self, net, inp, corr, flow, attention)
core/FlowFormer/LatentCostFormer/gru.py:125
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/convnext.py:14
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/convnext.py:47
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/convnext.py:79
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/mlpmixer.py:12
Methodforward
x: [BH1W1, K, D]
core/FlowFormer/LatentCostFormer/mlpmixer.py:45
Methodforward
(self, Q, K, V)
core/FlowFormer/LatentCostFormer/attention.py:25
Methodforward
(self, Q, K, V)
core/FlowFormer/LatentCostFormer/attention.py:53
Methodforward
(self, Q, K, V, Q_r, K_r)
core/FlowFormer/LatentCostFormer/attention.py:139
Methodforward
(self, image1, image2, output=None, flow_init=None)
core/FlowFormer/LatentCostFormer/transformer.py:32
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/encoder.py:56
Methodforward
(self, x, size, context=None)
core/FlowFormer/LatentCostFormer/encoder.py:103
Methodforward
(self, x, size, context=None)
core/FlowFormer/LatentCostFormer/encoder.py:130
Methodforward
x: [BH1W1, H3W3, D]
core/FlowFormer/LatentCostFormer/encoder.py:170
Methodforward
x: [BH1W1, H3W3, D]
core/FlowFormer/LatentCostFormer/encoder.py:226
Methodforward
(self, cost_volume, data, context=None)
core/FlowFormer/LatentCostFormer/encoder.py:275
Methodforward
(self, img1, img2, data, context=None)
core/FlowFormer/LatentCostFormer/encoder.py:337
Methodforward
(self, x, size: Size_, context=None)
core/FlowFormer/LatentCostFormer/twins.py:92
Methodforward
(self, x, size: Size_, context=None)
core/FlowFormer/LatentCostFormer/twins.py:178
Methodforward
(self, x, size: Size_, context=None)
core/FlowFormer/LatentCostFormer/twins.py:253
Methodforward
(self, x, size: Size_, context=None)
core/FlowFormer/LatentCostFormer/twins.py:336
Methodforward
(self, x, size: Size_, context=None)
core/FlowFormer/LatentCostFormer/twins.py:415
Methodforward
(self, x, size: Size_, context=None)
core/FlowFormer/LatentCostFormer/twins.py:482
Methodforward
(self, x, tgt, size: Size_)
core/FlowFormer/LatentCostFormer/twins.py:555
Methodforward
(self, x, size: Size_)
core/FlowFormer/LatentCostFormer/twins.py:604
← previousnext →201–300 of 336, ranked by callers