Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/drinkingcoder/FlowFormer-Official
/ types & classes
Types & classes
109 in github.com/drinkingcoder/FlowFormer-Official
⨍
Functions
336
◇
Types & classes
109
↓ 6 callers
Class
Block
core/FlowFormer/LatentCostFormer/twins.py:750
↓ 6 callers
Class
InputPadder
Pads images such that dimensions are divisible by 8
core/utils/utils.py:7
↓ 3 callers
Class
Attention
core/FlowFormer/LatentCostFormer/gma.py:34
↓ 3 callers
Class
FlowHead
core/FlowFormer/LatentCostFormer/cnn.py:296
↓ 3 callers
Class
FlyingThings3D
core/datasets.py:137
↓ 3 callers
Class
FlyingThings3D
core/utils/datasets.py:188
↓ 3 callers
Class
MpiSintel
core/utils/datasets.py:122
↓ 3 callers
Class
ResidualBlock
core/FlowFormer/LatentCostFormer/cnn.py:8
↓ 3 callers
Class
autocast
core/raft.py:15
↓ 3 callers
Class
twins_svt_large
core/FlowFormer/encoders.py:6
↓ 2 callers
Class
BasicEncoder
core/extractor.py:118
↓ 2 callers
Class
BasicEncoder
core/FlowFormer/LatentCostFormer/cnn.py:120
↓ 2 callers
Class
BasicMotionEncoder
core/FlowFormer/LatentCostFormer/cnn.py:352
↓ 2 callers
Class
BasicMotionEncoder
core/FlowFormer/LatentCostFormer/gru.py:61
↓ 2 callers
Class
BottleneckBlock
core/extractor.py:60
↓ 2 callers
Class
BottleneckBlock
core/FlowFormer/LatentCostFormer/cnn.py:62
↓ 2 callers
Class
FlowAugmentor
core/utils/augmentor.py:15
↓ 2 callers
Class
FlowHead
core/update.py:6
↓ 2 callers
Class
FlowHead
core/FlowFormer/LatentCostFormer/gru.py:5
↓ 2 callers
Class
KITTI
core/datasets.py:162
↓ 2 callers
Class
KITTI
core/utils/datasets.py:225
↓ 2 callers
Class
MpiSintel
core/datasets.py:102
↓ 2 callers
Class
MultiHeadAttention
core/FlowFormer/LatentCostFormer/attention.py:37
↓ 2 callers
Class
PreNormResidual
core/FlowFormer/LatentCostFormer/mlpmixer.py:6
↓ 2 callers
Class
ResidualBlock
core/extractor.py:6
↓ 2 callers
Class
SepConvGRU
core/FlowFormer/LatentCostFormer/gru.py:32
↓ 2 callers
Class
SmallEncoder
core/extractor.py:195
↓ 2 callers
Class
SparseFlowAugmentor
core/utils/augmentor.py:147
↓ 1 callers
Class
Aggregate
core/FlowFormer/LatentCostFormer/gma.py:79
↓ 1 callers
Class
AlternateCorrBlock
core/corr.py:62
↓ 1 callers
Class
AutoFlow
core/utils/datasets.py:263
↓ 1 callers
Class
BasicMotionEncoder
core/update.py:79
↓ 1 callers
Class
BasicUpdateBlock
core/update.py:114
↓ 1 callers
Class
BasicUpdateBlock
core/FlowFormer/LatentCostFormer/cnn.py:412
↓ 1 callers
Class
BroadMultiHeadAttention
core/FlowFormer/LatentCostFormer/attention.py:9
↓ 1 callers
Class
ConvGRU
core/update.py:16
↓ 1 callers
Class
ConvNextBlock
r""" ConvNeXt Block. There are two equivalent implementations: (1) DwConv -> LayerNorm (channels_first) -> 1x1 Conv -> GELU -> 1x1 Conv; all in (N
core/FlowFormer/LatentCostFormer/convnext.py:24
↓ 1 callers
Class
ConvNextLayer
core/FlowFormer/LatentCostFormer/convnext.py:7
↓ 1 callers
Class
CorrBlock
core/corr.py:12
↓ 1 callers
Class
CostPerceiverEncoder
core/FlowFormer/LatentCostFormer/encoder.py:244
↓ 1 callers
Class
CrossAttentionLayer
core/FlowFormer/LatentCostFormer/decoder.py:29
↓ 1 callers
Class
CrossAttentionLayer
core/FlowFormer/LatentCostFormer/encoder.py:196
↓ 1 callers
Class
CrossBlock
core/FlowFormer/LatentCostFormer/twins.py:727
↓ 1 callers
Class
CrossGlobalSubSampleAttnRPE
GSA: using a key to summarize the information for a group to be efficient.
core/FlowFormer/LatentCostFormer/twins.py:528
↓ 1 callers
Class
FlowFormer
core/FlowFormer/LatentCostFormer/transformer.py:19
↓ 1 callers
Class
FlyingChairs
core/datasets.py:121
↓ 1 callers
Class
FlyingChairs
core/utils/datasets.py:163
↓ 1 callers
Class
GMAUpdateBlock
core/FlowFormer/LatentCostFormer/gru.py:110
↓ 1 callers
Class
GlobalSubSampleAttn
GSA: using a key to summarize the information for a group to be efficient.
core/FlowFormer/LatentCostFormer/twins.py:633
↓ 1 callers
Class
GlobalSubSampleAttnRPE
GSA: using a key to summarize the information for a group to be efficient.
core/FlowFormer/LatentCostFormer/twins.py:455
↓ 1 callers
Class
GlobalSubSampleAttnRPEContext
GSA: using a key to summarize the information for a group to be efficient.
core/FlowFormer/LatentCostFormer/twins.py:306
↓ 1 callers
Class
GradScaler
train_FlowFormer.py:38
↓ 1 callers
Class
GroupAttnRPE
Latent cost tokens attend to different group
core/FlowFormer/LatentCostFormer/twins.py:153
↓ 1 callers
Class
GroupAttnRPEContext
Latent cost tokens attend to different group
core/FlowFormer/LatentCostFormer/twins.py:64
↓ 1 callers
Class
HD1K
core/datasets.py:181
↓ 1 callers
Class
HD1K
core/utils/datasets.py:392
↓ 1 callers
Class
LayerNorm
r""" LayerNorm that supports two data formats: channels_last (default) or channels_first. The ordering of the dimensions in the inputs. channels_
core/FlowFormer/LatentCostFormer/convnext.py:63
↓ 1 callers
Class
LocallyGroupedAttn
LSA: self attention within a group
core/FlowFormer/LatentCostFormer/twins.py:585
↓ 1 callers
Class
LocallyGroupedAttnRPE
LSA: self attention within a group
core/FlowFormer/LatentCostFormer/twins.py:394
↓ 1 callers
Class
LocallyGroupedAttnRPEContext
LSA: self attention within a group
core/FlowFormer/LatentCostFormer/twins.py:229
↓ 1 callers
Class
Logger
core/utils/logger.py:4
↓ 1 callers
Class
MLPMixerLayer
core/FlowFormer/LatentCostFormer/mlpmixer.py:24
↓ 1 callers
Class
MemoryDecoder
core/FlowFormer/LatentCostFormer/decoder.py:151
↓ 1 callers
Class
MemoryDecoderLayer
core/FlowFormer/LatentCostFormer/decoder.py:93
↓ 1 callers
Class
MemoryEncoder
core/FlowFormer/LatentCostFormer/encoder.py:310
↓ 1 callers
Class
PatchEmbed
core/FlowFormer/LatentCostFormer/encoder.py:24
↓ 1 callers
Class
PatchEmbed
Image to Patch Embedding
core/FlowFormer/LatentCostFormer/twins.py:811
↓ 1 callers
Class
PosConv
core/FlowFormer/LatentCostFormer/twins.py:791
↓ 1 callers
Class
RelPosEmb
core/FlowFormer/LatentCostFormer/gma.py:6
↓ 1 callers
Class
SelfAttentionLayer
core/FlowFormer/LatentCostFormer/encoder.py:143
↓ 1 callers
Class
SepConvGRU
core/update.py:33
↓ 1 callers
Class
SepConvGRU
core/FlowFormer/LatentCostFormer/cnn.py:323
↓ 1 callers
Class
SmallMotionEncoder
core/update.py:62
↓ 1 callers
Class
SmallUpdateBlock
core/update.py:99
↓ 1 callers
Class
VerticalSelfAttentionLayer
core/FlowFormer/LatentCostFormer/encoder.py:108
Class
BaiscMeanPredictor
core/FlowFormer/LatentCostFormer/cnn.py:451
Class
BasicFuseMotion
core/FlowFormer/LatentCostFormer/cnn.py:372
Class
BasicRPEEncoder
core/FlowFormer/LatentCostFormer/cnn.py:470
Class
BasicUpdateBlock
core/FlowFormer/LatentCostFormer/gru.py:85
Class
ChromaticAug
Chromatic augmentation: https://github.com/lmb-freiburg/flownet2/blob/master/src/caffe/layers/data_augmentation_layer.cu
core/utils/flow_transforms.py:401
Class
Compose
Composes several co_transforms together. For example: >>> co_transforms.Compose([ >>> co_transforms.CenterCrop(10), >>> co_tr
core/utils/flow_transforms.py:15
Class
ConvGRU
core/FlowFormer/LatentCostFormer/cnn.py:306
Class
ConvGRU
core/FlowFormer/LatentCostFormer/gru.py:15
Class
ConvNets
core/FlowFormer/LatentCostFormer/cnn.py:279
Class
CrossGlobalSubSampleAttn
GSA: using a key to summarize the information for a group to be efficient.
core/FlowFormer/LatentCostFormer/twins.py:680
Class
DirectMeanMaskPredictor
core/FlowFormer/LatentCostFormer/cnn.py:436
Class
FeedForward
core/FlowFormer/common.py:321
Class
FlowDataset
core/datasets.py:18
Class
FlowDataset
core/utils/datasets.py:22
Class
GroupVerticalSelfAttentionLayer
core/FlowFormer/LatentCostFormer/encoder.py:83
Class
InputPadder
Pads images such that dimensions are divisible by 8
evaluate_FlowFormer_tile.py:32
Class
LearnedPositionEncoding
This is a sinusoidal position encoding that generalized to 2-dimensional images
core/position_encoding.py:71
Class
LinearPositionEncoding
This is a sinusoidal position encoding that generalized to 2-dimensional images
core/position_encoding.py:38
Class
MLP
core/FlowFormer/common.py:335
Class
MpiSintel_submission
core/utils/datasets.py:104
Class
MultiHeadAttention
core/FlowFormer/common.py:353
Class
MultiHeadAttentionRelative
core/FlowFormer/LatentCostFormer/attention.py:107
Class
PCAAug
Chromatic Eigen Augmentation: https://github.com/lmb-freiburg/flownet2/blob/master/src/caffe/layers/data_augmentation_layer.cu
core/utils/flow_transforms.py:250
Class
PositionEncodingSine
This is a sinusoidal position encoding that generalized to 2-dimensional images
core/position_encoding.py:7
Class
RAFT
core/raft.py:24
next →
1–100 of 109, ranked by callers