MCPcopy Create free account

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

Functions336 in github.com/drinkingcoder/FlowFormer-Official

Methodforward
(self, x, size: Size_)
core/FlowFormer/LatentCostFormer/twins.py:659
Methodforward
(self, x, tgt, size: Size_)
core/FlowFormer/LatentCostFormer/twins.py:706
Methodforward
(self, src, tgt, size: Size_)
core/FlowFormer/LatentCostFormer/twins.py:739
Methodforward
(self, x, size: Size_, context=None)
core/FlowFormer/LatentCostFormer/twins.py:785
Methodforward
(self, x, size: Size_)
core/FlowFormer/LatentCostFormer/twins.py:798
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/twins.py:829
Methodforward
(self, x)
core/FlowFormer/LatentCostFormer/twins.py:931
Functionforward_interpolate
(flow)
core/utils/utils.py:28
Methodfreeze_bn
(self)
core/raft.py:58
Functionget_cfg
()
configs/things_eval.py:64
Functionget_cfg
()
configs/sintel.py:64
Functionget_cfg
()
configs/things.py:63
Functionget_cfg
()
configs/default.py:65
Functionget_cfg
()
configs/things_flowformer_sharp.py:63
Functionget_cfg
()
configs/small_things_eval.py:64
Methodget_classifier
(self)
core/FlowFormer/LatentCostFormer/twins.py:891
Functionnerf_encoding
x is of shape [*, 2]. The last dimension are two coordinates (x and y).
core/FlowFormer/common.py:14
Methodno_weight_decay
(self)
core/FlowFormer/LatentCostFormer/twins.py:807
Methodno_weight_decay
(self)
core/FlowFormer/LatentCostFormer/twins.py:888
Functionpyramid_retrieve_tokens
(feat_pyramid, center, image_size, window_sizes, sampler='bilinear')
core/FlowFormer/common.py:306
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
Methodreset_classifier
(self, num_classes, global_pool='')
core/FlowFormer/LatentCostFormer/twins.py:894
Functionsampler
(feat, center, window_size)
core/FlowFormer/common.py:263
Functionsampler_gaussian
(latent, mean, std, image_size, point_num=25)
core/FlowFormer/common.py:21
Functionsampler_gaussian_fix
(latent, mean, image_size, point_num=49)
core/FlowFormer/common.py:101
Functionsampler_gaussian_fix_MH
different heads have different mean
core/FlowFormer/common.py:199
Functionsampler_gaussian_fix_pyramid
(latent, feat_pyramid, scale_weight, mean, image_size, point_num=25)
core/FlowFormer/common.py:126
Functionsampler_gaussian_fix_pyramid_MH
(latent, feat_pyramid, scale_head_weight, mean, image_size, point_num=25)
core/FlowFormer/common.py:225
Functionsampler_gaussian_pyramid
(latent, feat_pyramid, scale_weight, mean, std, image_size, point_num=25)
core/FlowFormer/common.py:162
Functionsampler_gaussian_zy
(latent, mean, std, image_size, point_num=25, return_deltaXY=False, beta=1)
core/FlowFormer/common.py:46
Functionvalidate_kitti
(model, sigma=0.05)
evaluate_FlowFormer_tile.py:192
Functionvalidate_sintel
Peform validation using the Sintel (train) split
evaluate_FlowFormer_tile.py:250
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
core/utils/frame_utils.py:70
FunctionwriteFlowKITTI
(filename, uv)
core/utils/frame_utils.py:116
← previous301–336 of 336, ranked by callers