SYMBOL INDEX (534 symbols across 54 files) FILE: datasets/augmentor.py class AdjustGamma (line 13) | class AdjustGamma(object): method __init__ (line 14) | def __init__(self, gamma_min, gamma_max, gain_min=1.0, gain_max=1.0): method __call__ (line 22) | def __call__(self, sample): method __repr__ (line 27) | def __repr__(self): class SequenceDispFlowAugmentor (line 31) | class SequenceDispFlowAugmentor: method __init__ (line 32) | def __init__( method color_transform (line 71) | def color_transform(self, seq): method eraser_transform (line 95) | def eraser_transform(self, seq, bounds=[50, 100]): method spatial_transform (line 111) | def spatial_transform(self, img, disp): method __call__ (line 183) | def __call__(self, img, disp): class SequenceDispSparseFlowAugmentor (line 197) | class SequenceDispSparseFlowAugmentor: method __init__ (line 198) | def __init__( method color_transform (line 237) | def color_transform(self, seq): method eraser_transform (line 252) | def eraser_transform(self, seq, bounds=[50, 100]): method resize_sparse_flow_map (line 268) | def resize_sparse_flow_map(self, flow, valid, fx=1.0, fy=1.0): method spatial_transform (line 302) | def spatial_transform(self, img, disp, valid): method __call__ (line 352) | def __call__(self, img, disp, valid): FILE: datasets/frame_utils.py function readFlow (line 14) | def readFlow(fn): function readPFM (line 36) | def readPFM(file): function readDispSintelStereo (line 74) | def readDispSintelStereo(file_name): function readDispMiddlebury (line 87) | def readDispMiddlebury(file_name): function read_gen (line 98) | def read_gen(file_name, pil=False): FILE: datasets/video_datasets.py class DynamicReplicaFrameAnnotation (line 31) | class DynamicReplicaFrameAnnotation(ImplicitronFrameAnnotation): class StereoSequenceDataset (line 37) | class StereoSequenceDataset(data.Dataset): method __init__ (line 38) | def __init__(self, aug_params=None, sparse=False, reader=None): method _load_depth (line 60) | def _load_depth(self, depth_path): method _load_npy_depth (line 73) | def _load_npy_depth(self, depth_npy): method _load_vkitti2 (line 77) | def _load_vkitti2(self, depth_png): method _load_kitti_depth (line 84) | def _load_kitti_depth(self, depth_png): method _load_16big_png_depth (line 96) | def _load_16big_png_depth(self, depth_png): method load_tartanair_pose (line 107) | def load_tartanair_pose(self, filepath, index=0): method parse_txt_file (line 122) | def parse_txt_file(self, file_path): method _get_pytorch3d_camera (line 158) | def _get_pytorch3d_camera( method _get_pytorch3d_camera_from_blender (line 204) | def _get_pytorch3d_camera_from_blender(self, R, T, K, image_size, scal... method _get_output_tensor (line 247) | def _get_output_tensor(self, sample): method __getitem__ (line 666) | def __getitem__(self, index): method __mul__ (line 766) | def __mul__(self, v): method __len__ (line 772) | def __len__(self): class DynamicReplicaDataset (line 776) | class DynamicReplicaDataset(StereoSequenceDataset): method __init__ (line 777) | def __init__( class InfinigenStereoVideoDataset (line 876) | class InfinigenStereoVideoDataset(StereoSequenceDataset): method __init__ (line 877) | def __init__( class SouthKensingtonStereoVideoDataset (line 961) | class SouthKensingtonStereoVideoDataset(StereoSequenceDataset): method __init__ (line 962) | def __init__( class KITTIDepthDataset (line 1025) | class KITTIDepthDataset(StereoSequenceDataset): method __init__ (line 1026) | def __init__( function split_train_valid (line 1139) | def split_train_valid(path_list, valid_keywords): class TartanAirDataset (line 1148) | class TartanAirDataset(StereoSequenceDataset): method __init__ (line 1149) | def __init__( class VKITTI2Dataset (line 1275) | class VKITTI2Dataset(StereoSequenceDataset): method __init__ (line 1276) | def __init__( class SequenceSpringDataset (line 1367) | class SequenceSpringDataset(StereoSequenceDataset): method __init__ (line 1368) | def __init__( class SequenceSceneFlowDataset (line 1424) | class SequenceSceneFlowDataset(StereoSequenceDataset): method __init__ (line 1425) | def __init__( method _add_things (line 1450) | def _add_things(self, split="TRAIN"): method _add_monkaa (line 1496) | def _add_monkaa(self): method _add_driving (line 1530) | def _add_driving(self): method _append_sample (line 1565) | def _append_sample(self, images, disparities): class SequenceSintelStereo (line 1583) | class SequenceSintelStereo(StereoSequenceDataset): method __init__ (line 1584) | def __init__( function fetch_dataloader (line 1649) | def fetch_dataloader(args): FILE: demo.py function load_image (line 19) | def load_image(imfile): function viz (line 25) | def viz(img, flo): function demo (line 37) | def demo(args): FILE: evaluation/core/evaluator.py function depth_to_colormap (line 19) | def depth_to_colormap(depth, colormap='jet', eps=1e-3, scale_vmin=1.0): class Evaluator (line 33) | class Evaluator(Configurable): method setup_visualization (line 43) | def setup_visualization(self, cfg: DictConfig) -> None: method evaluate_sequence (line 52) | def evaluate_sequence( FILE: evaluation/evaluate.py class DefaultConfig (line 25) | class DefaultConfig: function run_eval (line 57) | def run_eval(cfg: DefaultConfig): function evaluate (line 141) | def evaluate(cfg: DefaultConfig) -> None: FILE: evaluation/utils/eval_utils.py class PerceptionMetric (line 14) | class PerceptionMetric: method __str__ (line 20) | def __str__(self): function compute_flow (line 33) | def compute_flow(seq, is_seq=True): function flow_warp (line 50) | def flow_warp(x, flow): function eval_endpoint_error_sequence (line 75) | def eval_endpoint_error_sequence( function eval_TCC_sequence (line 146) | def eval_TCC_sequence( function eval_TCM_sequence (line 185) | def eval_TCM_sequence( function eval_OPW_sequence (line 242) | def eval_OPW_sequence( function eval_RTC_sequence (line 313) | def eval_RTC_sequence( function depth2disparity_scale (line 379) | def depth2disparity_scale(left_camera, right_camera, image_size_tensor): function depth_to_pcd (line 394) | def depth_to_pcd( function filter_outliers (line 438) | def filter_outliers(pcd, sigma=3): function eval_batch (line 446) | def eval_batch(batch_dict, predictions, scale) -> Dict[str, Union[float,... FILE: evaluation/utils/ssim.py function _fspecial_gauss_1d (line 9) | def _fspecial_gauss_1d(size, sigma): function gaussian_filter (line 26) | def gaussian_filter(input, win): function _ssim (line 54) | def _ssim(X, Y, data_range, win, size_average=True, K=(0.01, 0.03)): function ssim (line 94) | def ssim( function ms_ssim (line 152) | def ms_ssim( class SSIM (line 227) | class SSIM(torch.nn.Module): method __init__ (line 228) | def __init__( method forward (line 258) | def forward(self, X, Y): class MS_SSIM (line 270) | class MS_SSIM(torch.nn.Module): method __init__ (line 271) | def __init__( method forward (line 301) | def forward(self, X, Y): FILE: evaluation/utils/utils.py class PerceptionPrediction (line 28) | class PerceptionPrediction: function aggregate_eval_results (line 39) | def aggregate_eval_results(per_batch_eval_results, reduction="mean"): function aggregate_and_print_results (line 62) | def aggregate_and_print_results( function pretty_print_perception_metrics (line 76) | def pretty_print_perception_metrics(results): function read_calibration (line 92) | def read_calibration(calibration_file, resolution_string): function filter_depth_discontinuities (line 216) | def filter_depth_discontinuities(pcd, depth_map, threshold=5): function visualize_batch (line 246) | def visualize_batch( FILE: models/Video-Depth-Anything/app.py function infer_video_depth (line 40) | def infer_video_depth( function construct_demo (line 64) | def construct_demo(): FILE: models/Video-Depth-Anything/utils/dc_utils.py function read_video_frames (line 21) | def read_video_frames(video_path, process_length, target_fps=-1, max_res... function save_video (line 74) | def save_video( class ColorMapper (line 92) | class ColorMapper: method __init__ (line 94) | def __init__(self, colormap: str = "inferno"): method apply (line 97) | def apply(self, image: torch.Tensor, v_min=None, v_max=None): function vis_sequence_depth (line 109) | def vis_sequence_depth(depths: np.ndarray, v_min=None, v_max=None): FILE: models/Video-Depth-Anything/utils/util.py function compute_scale_and_shift (line 16) | def compute_scale_and_shift(prediction, target, mask, scale_only=False): function compute_scale (line 23) | def compute_scale(prediction, target, mask): function compute_scale_and_shift_full (line 40) | def compute_scale_and_shift_full(prediction, target, mask): function get_interpolate_frames (line 65) | def get_interpolate_frames(frame_list_pre, frame_list_post): FILE: models/Video-Depth-Anything/video_depth_anything/dinov2.py function named_apply (line 26) | def named_apply(fn: Callable, module: nn.Module, name="", depth_first=Tr... class BlockChunk (line 37) | class BlockChunk(nn.ModuleList): method forward (line 38) | def forward(self, x): class DinoVisionTransformer (line 44) | class DinoVisionTransformer(nn.Module): method __init__ (line 45) | def __init__( method init_weights (line 172) | def init_weights(self): method interpolate_pos_encoding (line 179) | def interpolate_pos_encoding(self, x, w, h): method prepare_tokens_with_masks (line 212) | def prepare_tokens_with_masks(self, x, masks=None): method forward_features_list (line 233) | def forward_features_list(self, x_list, masks_list): method forward_features (line 253) | def forward_features(self, x, masks=None): method _get_intermediate_layers_not_chunked (line 271) | def _get_intermediate_layers_not_chunked(self, x, n=1): method _get_intermediate_layers_chunked (line 283) | def _get_intermediate_layers_chunked(self, x, n=1): method get_intermediate_layers (line 297) | def get_intermediate_layers( method forward (line 323) | def forward(self, *args, is_training=False, **kwargs): function init_weights_vit_timm (line 331) | def init_weights_vit_timm(module: nn.Module, name: str = ""): function vit_small (line 339) | def vit_small(patch_size=16, num_register_tokens=0, **kwargs): function vit_base (line 353) | def vit_base(patch_size=16, num_register_tokens=0, **kwargs): function vit_large (line 367) | def vit_large(patch_size=16, num_register_tokens=0, **kwargs): function vit_giant2 (line 381) | def vit_giant2(patch_size=16, num_register_tokens=0, **kwargs): function DINOv2 (line 398) | def DINOv2(model_name): FILE: models/Video-Depth-Anything/video_depth_anything/dinov2_layers/attention.py class Attention (line 29) | class Attention(nn.Module): method __init__ (line 30) | def __init__( method forward (line 49) | def forward(self, x: Tensor) -> Tensor: class MemEffAttention (line 65) | class MemEffAttention(Attention): method forward (line 66) | def forward(self, x: Tensor, attn_bias=None) -> Tensor: FILE: models/Video-Depth-Anything/video_depth_anything/dinov2_layers/block.py class Block (line 36) | class Block(nn.Module): method __init__ (line 37) | def __init__( method forward (line 82) | def forward(self, x: Tensor) -> Tensor: function drop_add_residual_stochastic_depth (line 110) | def drop_add_residual_stochastic_depth( function get_branges_scales (line 134) | def get_branges_scales(x, sample_drop_ratio=0.0): function add_residual (line 142) | def add_residual(x, brange, residual, residual_scale_factor, scaling_vec... function get_attn_bias_and_cat (line 157) | def get_attn_bias_and_cat(x_list, branges=None): function drop_add_residual_stochastic_depth_list (line 181) | def drop_add_residual_stochastic_depth_list( class NestedTensorBlock (line 204) | class NestedTensorBlock(Block): method forward_nested (line 205) | def forward_nested(self, x_list: List[Tensor]) -> List[Tensor]: method forward (line 245) | def forward(self, x_or_x_list): FILE: models/Video-Depth-Anything/video_depth_anything/dinov2_layers/drop_path.py function drop_path (line 15) | def drop_path(x, drop_prob: float = 0.0, training: bool = False): class DropPath (line 27) | class DropPath(nn.Module): method __init__ (line 30) | def __init__(self, drop_prob=None): method forward (line 34) | def forward(self, x): FILE: models/Video-Depth-Anything/video_depth_anything/dinov2_layers/layer_scale.py class LayerScale (line 16) | class LayerScale(nn.Module): method __init__ (line 17) | def __init__( method forward (line 27) | def forward(self, x: Tensor) -> Tensor: FILE: models/Video-Depth-Anything/video_depth_anything/dinov2_layers/mlp.py class Mlp (line 17) | class Mlp(nn.Module): method __init__ (line 18) | def __init__( method forward (line 35) | def forward(self, x: Tensor) -> Tensor: FILE: models/Video-Depth-Anything/video_depth_anything/dinov2_layers/patch_embed.py function make_2tuple (line 17) | def make_2tuple(x): class PatchEmbed (line 26) | class PatchEmbed(nn.Module): method __init__ (line 38) | def __init__( method forward (line 69) | def forward(self, x: Tensor) -> Tensor: method flops (line 84) | def flops(self) -> float: FILE: models/Video-Depth-Anything/video_depth_anything/dinov2_layers/swiglu_ffn.py class SwiGLUFFN (line 13) | class SwiGLUFFN(nn.Module): method __init__ (line 14) | def __init__( method forward (line 29) | def forward(self, x: Tensor) -> Tensor: class SwiGLUFFNFused (line 45) | class SwiGLUFFNFused(SwiGLU): method __init__ (line 46) | def __init__( FILE: models/Video-Depth-Anything/video_depth_anything/dpt.py function _make_fusion_block (line 21) | def _make_fusion_block(features, use_bn, size=None): class ConvBlock (line 33) | class ConvBlock(nn.Module): method __init__ (line 34) | def __init__(self, in_feature, out_feature): method forward (line 43) | def forward(self, x): class DPTHead (line 47) | class DPTHead(nn.Module): method __init__ (line 48) | def __init__( method forward (line 126) | def forward(self, out_features, patch_h, patch_w): FILE: models/Video-Depth-Anything/video_depth_anything/dpt_temporal.py class DPTHeadTemporal (line 22) | class DPTHeadTemporal(DPTHead): method __init__ (line 23) | def __init__(self, method forward (line 53) | def forward(self, out_features, patch_h, patch_w, frame_length): FILE: models/Video-Depth-Anything/video_depth_anything/motion_module/attention.py class CrossAttention (line 30) | class CrossAttention(nn.Module): method __init__ (line 45) | def __init__( method reshape_heads_to_batch_dim (line 93) | def reshape_heads_to_batch_dim(self, tensor): method reshape_heads_to_4d (line 100) | def reshape_heads_to_4d(self, tensor): method reshape_batch_dim_to_heads (line 106) | def reshape_batch_dim_to_heads(self, tensor): method reshape_4d_to_heads (line 113) | def reshape_4d_to_heads(self, tensor): method set_attention_slice (line 119) | def set_attention_slice(self, slice_size): method forward (line 125) | def forward(self, hidden_states, encoder_hidden_states=None, attention... method _attention (line 182) | def _attention(self, query, key, value, attention_mask=None): method _sliced_attention (line 213) | def _sliced_attention(self, query, key, value, sequence_length, dim, a... method _memory_efficient_attention_xformers (line 256) | def _memory_efficient_attention_xformers(self, query, key, value, atte... method _memory_efficient_attention_split (line 275) | def _memory_efficient_attention_split(self, query, key, value, attenti... class FeedForward (line 296) | class FeedForward(nn.Module): method __init__ (line 308) | def __init__( method forward (line 335) | def forward(self, hidden_states): class GELU (line 341) | class GELU(nn.Module): method __init__ (line 346) | def __init__(self, dim_in: int, dim_out: int): method gelu (line 350) | def gelu(self, gate): method forward (line 356) | def forward(self, hidden_states): class GEGLU (line 363) | class GEGLU(nn.Module): method __init__ (line 372) | def __init__(self, dim_in: int, dim_out: int): method gelu (line 376) | def gelu(self, gate): method forward (line 382) | def forward(self, hidden_states): class ApproximateGELU (line 387) | class ApproximateGELU(nn.Module): method __init__ (line 394) | def __init__(self, dim_in: int, dim_out: int): method forward (line 398) | def forward(self, x): function precompute_freqs_cis (line 403) | def precompute_freqs_cis(dim: int, end: int, theta: float = 10000.0): function reshape_for_broadcast (line 411) | def reshape_for_broadcast(freqs_cis: torch.Tensor, x: torch.Tensor): function apply_rotary_emb (line 419) | def apply_rotary_emb( FILE: models/Video-Depth-Anything/video_depth_anything/motion_module/motion_module.py function zero_module (line 25) | def zero_module(module): class TemporalModule (line 32) | class TemporalModule(nn.Module): method __init__ (line 33) | def __init__( method forward (line 60) | def forward(self, input_tensor, encoder_hidden_states, attention_mask=... class TemporalTransformer3DModel (line 68) | class TemporalTransformer3DModel(nn.Module): method __init__ (line 69) | def __init__( method forward (line 102) | def forward(self, hidden_states, encoder_hidden_states=None, attention... class TemporalTransformerBlock (line 129) | class TemporalTransformerBlock(nn.Module): method __init__ (line 130) | def __init__( method forward (line 164) | def forward(self, hidden_states, encoder_hidden_states=None, attention... class PositionalEncoding (line 180) | class PositionalEncoding(nn.Module): method __init__ (line 181) | def __init__( method forward (line 196) | def forward(self, x): class TemporalAttention (line 200) | class TemporalAttention(CrossAttention): method __init__ (line 201) | def __init__( method forward (line 230) | def forward(self, hidden_states, encoder_hidden_states=None, attention... FILE: models/Video-Depth-Anything/video_depth_anything/util/blocks.py function _make_scratch (line 4) | def _make_scratch(in_shape, out_shape, groups=1, expand=False): class ResidualConvUnit (line 37) | class ResidualConvUnit(nn.Module): method __init__ (line 40) | def __init__(self, features, activation, bn): method forward (line 68) | def forward(self, x): class FeatureFusionBlock (line 94) | class FeatureFusionBlock(nn.Module): method __init__ (line 97) | def __init__( method forward (line 135) | def forward(self, *xs, size=None): FILE: models/Video-Depth-Anything/video_depth_anything/util/transform.py class Resize (line 5) | class Resize(object): method __init__ (line 9) | def __init__( method constrain_to_multiple_of (line 51) | def constrain_to_multiple_of(self, x, min_val=0, max_val=None): method get_size (line 62) | def get_size(self, width, height): method __call__ (line 109) | def __call__(self, sample): class NormalizeImage (line 125) | class NormalizeImage(object): method __init__ (line 129) | def __init__(self, mean, std): method __call__ (line 133) | def __call__(self, sample): class PrepareForNet (line 139) | class PrepareForNet(object): method __init__ (line 143) | def __init__(self): method __call__ (line 146) | def __call__(self, sample): FILE: models/Video-Depth-Anything/video_depth_anything/video_depth.py class VideoDepthAnything (line 35) | class VideoDepthAnything(nn.Module): method __init__ (line 36) | def __init__( method forward (line 57) | def forward(self, x): method infer_video_depth (line 66) | def infer_video_depth(self, frames, target_fps, input_size=518, device... FILE: models/core/attention.py function elu_feature_map (line 13) | def elu_feature_map(x): class PositionEncodingSine (line 17) | class PositionEncodingSine(nn.Module): method __init__ (line 22) | def __init__(self, d_model, max_shape=(256, 256), temp_bug_fix=True): method forward (line 53) | def forward(self, x): class LinearAttention (line 61) | class LinearAttention(Module): method __init__ (line 62) | def __init__(self, eps=1e-6): method forward (line 67) | def forward(self, queries, keys, values, q_mask=None, kv_mask=None): class FullAttention (line 97) | class FullAttention(Module): method __init__ (line 98) | def __init__(self, use_dropout=False, attention_dropout=0.1): method forward (line 103) | def forward(self, queries, keys, values, q_mask=None, kv_mask=None): class LoFTREncoderLayer (line 134) | class LoFTREncoderLayer(nn.Module): method __init__ (line 135) | def __init__(self, d_model, nhead, attention="linear"): method forward (line 159) | def forward(self, x, source, x_mask=None, source_mask=None): class LocalFeatureTransformer (line 187) | class LocalFeatureTransformer(nn.Module): method __init__ (line 190) | def __init__(self, d_model, nhead, layer_names, attention): method _reset_parameters (line 202) | def _reset_parameters(self): method forward (line 207) | def forward(self, feat0, feat1, mask0=None, mask1=None): FILE: models/core/corr.py function bilinear_sampler (line 6) | def bilinear_sampler(img, coords, mode="bilinear", mask=False): function coords_grid (line 24) | def coords_grid(batch, ht, wd, device): class AAPC (line 32) | class AAPC: method __init__ (line 36) | def __init__(self, fmap1, fmap2, att=None): method __call__ (line 43) | def __call__(self, flow, extra_offset, small_patch=False): method correlation (line 49) | def correlation(self, left_feature, right_feature, flow, small_patch): method get_correlation (line 73) | def get_correlation(self, left_feature, right_feature, psize=(3, 3), d... FILE: models/core/extractor.py class ResidualBlock (line 12) | class ResidualBlock(nn.Module): method __init__ (line 13) | def __init__(self, in_planes, planes, norm_fn="group", stride=1): method forward (line 48) | def forward(self, x): class BasicEncoder (line 58) | class BasicEncoder(nn.Module): method __init__ (line 59) | def __init__(self, input_dim=3, output_dim=128, norm_fn="batch", dropo... method _make_layer (line 99) | def _make_layer(self, dim, stride=1): method forward (line 107) | def forward(self, x): class MultiBasicEncoder (line 134) | class MultiBasicEncoder(nn.Module): method __init__ (line 135) | def __init__(self, output_dim=[128], norm_fn='batch', dropout=0.0, dow... method _make_layer (line 201) | def _make_layer(self, dim, stride=1): method forward (line 209) | def forward(self, x, dual_inp=False, num_layers=3): class DepthExtractor (line 238) | class DepthExtractor(nn.Module): method __init__ (line 239) | def __init__(self): method forward (line 258) | def forward(self, x): FILE: models/core/model_zoo.py function model_zoo (line 18) | def model_zoo(model_name: str, **kwargs): function get_all_model_default_configs (line 37) | def get_all_model_default_configs(): FILE: models/core/stereoanyvideo.py function _ntuple (line 19) | def _ntuple(n): function exists (line 28) | def exists(val): function default (line 32) | def default(val, d): class Mlp (line 38) | class Mlp(nn.Module): method __init__ (line 39) | def __init__( method forward (line 66) | def forward(self, x): class StereoAnyVideo (line 75) | class StereoAnyVideo(nn.Module): method __init__ (line 76) | def __init__(self, mixed_precision=False): method no_weight_decay (line 93) | def no_weight_decay(self): method freeze_bn (line 96) | def freeze_bn(self): method convex_upsample (line 101) | def convex_upsample(self, flow, mask, rate=4): method convex_upsample_3D (line 114) | def convex_upsample_3D(self, flow, mask, b, T, rate=4): method zero_init (line 144) | def zero_init(self, fmap): method forward_batch_test (line 151) | def forward_batch_test( method forward (line 205) | def forward(self, image1, image2, flow_init=None, iters=12, test_mode=... FILE: models/core/update.py function pool2x (line 9) | def pool2x(x): function pool4x (line 12) | def pool4x(x): function interp (line 15) | def interp(x, dest): class FlowHead (line 20) | class FlowHead(nn.Module): method __init__ (line 21) | def __init__(self, input_dim=128, hidden_dim=256, output_dim=2): method forward (line 27) | def forward(self, x): class FlowHead3D (line 31) | class FlowHead3D(nn.Module): method __init__ (line 32) | def __init__(self, input_dim=128, hidden_dim=256): method forward (line 38) | def forward(self, x): class ConvGRU (line 42) | class ConvGRU(nn.Module): method __init__ (line 43) | def __init__(self, hidden_dim, input_dim, kernel_size=3): method forward (line 49) | def forward(self, h, cz, cr, cq, *x_list): class SepConvGRU (line 61) | class SepConvGRU(nn.Module): method __init__ (line 62) | def __init__(self, hidden_dim=128, input_dim=192+128): method forward (line 73) | def forward(self, h, *x): class BasicMotionEncoder (line 92) | class BasicMotionEncoder(nn.Module): method __init__ (line 93) | def __init__(self, cor_planes): method forward (line 102) | def forward(self, flow, corr): class BasicMotionEncoder3D (line 113) | class BasicMotionEncoder3D(nn.Module): method __init__ (line 114) | def __init__(self, cor_planes): method forward (line 123) | def forward(self, flow, corr): class SepConvGRU3D (line 134) | class SepConvGRU3D(nn.Module): method __init__ (line 135) | def __init__(self, hidden_dim=128, input_dim=192 + 128): method forward (line 167) | def forward(self, h, x): class SKSepConvGRU3D (line 191) | class SKSepConvGRU3D(nn.Module): method __init__ (line 192) | def __init__(self, hidden_dim=128, input_dim=192 + 128): method forward (line 228) | def forward(self, h, x): class BasicUpdateBlock (line 252) | class BasicUpdateBlock(nn.Module): method __init__ (line 253) | def __init__(self, hidden_dim, cor_planes, mask_size=8, attention_type... method forward (line 273) | def forward(self, net, inp, corr, flow, upsample=True, t=1): class Attention (line 289) | class Attention(nn.Module): method __init__ (line 290) | def __init__(self, dim, num_heads=8, qkv_bias=False, qk_scale=None): method forward (line 298) | def forward(self, x): class TimeAttnBlock (line 312) | class TimeAttnBlock(nn.Module): method __init__ (line 313) | def __init__(self, dim=256, num_heads=8): method forward (line 322) | def forward(self, x, T=1): class SpaceAttnBlock (line 340) | class SpaceAttnBlock(nn.Module): method __init__ (line 341) | def __init__(self, dim=256, num_heads=8): method forward (line 345) | def forward(self, x, T=1): class SequenceUpdateBlock3D (line 353) | class SequenceUpdateBlock3D(nn.Module): method __init__ (line 354) | def __init__(self, hidden_dim, cor_planes, mask_size=8): method forward (line 368) | def forward(self, net, inp, corrs, flows, t): FILE: models/core/utils/config.py class ReplaceableBase (line 174) | class ReplaceableBase: method __new__ (line 180) | def __new__(cls, *args, **kwargs): class Configurable (line 192) | class Configurable: method __new__ (line 200) | def __new__(cls, *args, **kwargs): class _Registry (line 215) | class _Registry: method __init__ (line 222) | def __init__(self) -> None: method register (line 227) | def register(self, some_class: Type[_X]) -> Type[_X]: method _register (line 235) | def _register( method get (line 260) | def get( method get_all (line 292) | def get_all( method _is_base_class (line 320) | def _is_base_class(some_class: Type[ReplaceableBase]) -> bool: method _base_class_from_class (line 328) | def _base_class_from_class( class _ProcessType (line 344) | class _ProcessType(Enum): function _default_create (line 355) | def _default_create( function run_auto_creation (line 413) | def run_auto_creation(self: Any) -> None: function _is_configurable_class (line 421) | def _is_configurable_class(C) -> bool: function get_default_args (line 425) | def get_default_args(C, *, _do_not_process: Tuple[type, ...] = ()) -> Di... function _dataclass_name_for_function (line 487) | def _dataclass_name_for_function(C: Any) -> str: function enable_get_default_args (line 496) | def enable_get_default_args(C: Any, *, overwrite: bool = True) -> None: function _params_iter (line 553) | def _params_iter(C): function _is_immutable_type (line 563) | def _is_immutable_type(type_: Type, val: Any) -> bool: function _resolve_optional (line 575) | def _resolve_optional(type_: Any) -> Tuple[bool, Any]: function _is_actually_dataclass (line 587) | def _is_actually_dataclass(some_class) -> bool: function expand_args_fields (line 597) | def expand_args_fields( function get_default_args_field (line 776) | def get_default_args_field(C, *, _do_not_process: Tuple[type, ...] = ()): function _get_type_to_process (line 793) | def _get_type_to_process(type_) -> Optional[Tuple[Type, _ProcessType]]: function _process_member (line 825) | def _process_member( function remove_unused_components (line 917) | def remove_unused_components(dict_: DictConfig) -> None: FILE: models/core/utils/utils.py function interp (line 7) | def interp(tensor, size): class InputPadder (line 16) | class InputPadder: method __init__ (line 19) | def __init__(self, dims, mode="sintel", divis_by=8): method pad (line 33) | def pad(self, *inputs): method unpad (line 37) | def unpad(self, x): function coords_grid (line 44) | def coords_grid(batch, ht, wd): function upflow8 (line 50) | def upflow8(flow, mode='bilinear'): FILE: models/raft_model.py class RAFTModel (line 13) | class RAFTModel(Configurable, torch.nn.Module): method __post_init__ (line 16) | def __post_init__(self): method forward (line 45) | def forward(self, image1, image2, iters=10): method forward_fullres (line 59) | def forward_fullres(self, image1, image2, iters=20): FILE: models/stereoanyvideo_model.py class StereoAnyVideoModel (line 9) | class StereoAnyVideoModel(Configurable, torch.nn.Module): method __post_init__ (line 14) | def __post_init__(self): method forward (line 32) | def forward(self, batch_dict, iters=20): FILE: third_party/RAFT/alt_cuda_corr/correlation.cpp function corr_forward (line 23) | std::vector corr_forward( function corr_backward (line 36) | std::vector corr_backward( function PYBIND11_MODULE (line 51) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: third_party/RAFT/core/corr.py class CorrBlock (line 12) | class CorrBlock: method __init__ (line 13) | def __init__(self, fmap1, fmap2, num_levels=4, radius=4): method __call__ (line 29) | def __call__(self, coords): method corr (line 53) | def corr(fmap1, fmap2): class AlternateCorrBlock (line 63) | class AlternateCorrBlock: method __init__ (line 64) | def __init__(self, fmap1, fmap2, num_levels=4, radius=4): method __call__ (line 74) | def __call__(self, coords): FILE: third_party/RAFT/core/datasets.py class FlowDataset (line 18) | class FlowDataset(data.Dataset): method __init__ (line 19) | def __init__(self, aug_params=None, sparse=False): method __getitem__ (line 34) | def __getitem__(self, index): method __rmul__ (line 93) | def __rmul__(self, v): method __len__ (line 98) | def __len__(self): class MpiSintel (line 102) | class MpiSintel(FlowDataset): method __init__ (line 103) | def __init__(self, aug_params=None, split='training', root='datasets/S... class FlyingChairs (line 121) | class FlyingChairs(FlowDataset): method __init__ (line 122) | def __init__(self, aug_params=None, split='train', root='datasets/Flyi... class FlyingThings3D (line 137) | class FlyingThings3D(FlowDataset): method __init__ (line 138) | def __init__(self, aug_params=None, root='datasets/FlyingThings3D', ds... class KITTI (line 161) | class KITTI(FlowDataset): method __init__ (line 162) | def __init__(self, aug_params=None, split='training', root='datasets/K... class HD1K (line 180) | class HD1K(FlowDataset): method __init__ (line 181) | def __init__(self, aug_params=None, root='datasets/HD1k'): function fetch_dataloader (line 199) | def fetch_dataloader(args, TRAIN_DS='C+T+K+S+H'): FILE: third_party/RAFT/core/extractor.py class ResidualBlock (line 6) | class ResidualBlock(nn.Module): method __init__ (line 7) | def __init__(self, in_planes, planes, norm_fn='group', stride=1): method forward (line 48) | def forward(self, x): class BottleneckBlock (line 60) | class BottleneckBlock(nn.Module): method __init__ (line 61) | def __init__(self, in_planes, planes, norm_fn='group', stride=1): method forward (line 107) | def forward(self, x): class BasicEncoder (line 118) | class BasicEncoder(nn.Module): method __init__ (line 119) | def __init__(self, output_dim=128, norm_fn='batch', dropout=0.0): method _make_layer (line 159) | def _make_layer(self, dim, stride=1): method forward (line 168) | def forward(self, x): class SmallEncoder (line 195) | class SmallEncoder(nn.Module): method __init__ (line 196) | def __init__(self, output_dim=128, norm_fn='batch', dropout=0.0): method _make_layer (line 235) | def _make_layer(self, dim, stride=1): method forward (line 244) | def forward(self, x): FILE: third_party/RAFT/core/raft.py class autocast (line 15) | class autocast: method __init__ (line 16) | def __init__(self, enabled): method __enter__ (line 18) | def __enter__(self): method __exit__ (line 20) | def __exit__(self, *args): class RAFT (line 24) | class RAFT(nn.Module): method __init__ (line 25) | def __init__(self, args): method freeze_bn (line 58) | def freeze_bn(self): method initialize_flow (line 63) | def initialize_flow(self, img): method upsample_flow (line 72) | def upsample_flow(self, flow, mask): method forward (line 86) | def forward(self, image1, image2, iters=12, flow_init=None, upsample=T... FILE: third_party/RAFT/core/update.py class FlowHead (line 6) | class FlowHead(nn.Module): method __init__ (line 7) | def __init__(self, input_dim=128, hidden_dim=256): method forward (line 13) | def forward(self, x): class ConvGRU (line 16) | class ConvGRU(nn.Module): method __init__ (line 17) | def __init__(self, hidden_dim=128, input_dim=192+128): method forward (line 23) | def forward(self, h, x): class SepConvGRU (line 33) | class SepConvGRU(nn.Module): method __init__ (line 34) | def __init__(self, hidden_dim=128, input_dim=192+128): method forward (line 45) | def forward(self, h, x): class SmallMotionEncoder (line 62) | class SmallMotionEncoder(nn.Module): method __init__ (line 63) | def __init__(self, args): method forward (line 71) | def forward(self, flow, corr): class BasicMotionEncoder (line 79) | class BasicMotionEncoder(nn.Module): method __init__ (line 80) | def __init__(self, args): method forward (line 89) | def forward(self, flow, corr): class SmallUpdateBlock (line 99) | class SmallUpdateBlock(nn.Module): method __init__ (line 100) | def __init__(self, args, hidden_dim=96): method forward (line 106) | def forward(self, net, inp, corr, flow): class BasicUpdateBlock (line 114) | class BasicUpdateBlock(nn.Module): method __init__ (line 115) | def __init__(self, args, hidden_dim=128, input_dim=128): method forward (line 127) | def forward(self, net, inp, corr, flow, upsample=True): FILE: third_party/RAFT/core/utils/augmentor.py class FlowAugmentor (line 15) | class FlowAugmentor: method __init__ (line 16) | def __init__(self, crop_size, min_scale=-0.2, max_scale=0.5, do_flip=T... method color_transform (line 36) | def color_transform(self, img1, img2): method eraser_transform (line 52) | def eraser_transform(self, img1, img2, bounds=[50, 100]): method spatial_transform (line 67) | def spatial_transform(self, img1, img2, flow): method __call__ (line 111) | def __call__(self, img1, img2, flow): class SparseFlowAugmentor (line 122) | class SparseFlowAugmentor: method __init__ (line 123) | def __init__(self, crop_size, min_scale=-0.2, max_scale=0.5, do_flip=F... method color_transform (line 142) | def color_transform(self, img1, img2): method eraser_transform (line 148) | def eraser_transform(self, img1, img2): method resize_sparse_flow_map (line 161) | def resize_sparse_flow_map(self, flow, valid, fx=1.0, fy=1.0): method spatial_transform (line 195) | def spatial_transform(self, img1, img2, flow, valid): method __call__ (line 236) | def __call__(self, img1, img2, flow, valid): FILE: third_party/RAFT/core/utils/flow_viz.py function make_colorwheel (line 20) | def make_colorwheel(): function flow_uv_to_colors (line 70) | def flow_uv_to_colors(u, v, convert_to_bgr=False): function flow_to_image (line 109) | def flow_to_image(flow_uv, clip_flow=None, convert_to_bgr=False): FILE: third_party/RAFT/core/utils/frame_utils.py function readFlow (line 12) | def readFlow(fn): function readPFM (line 33) | def readPFM(file): function writeFlow (line 70) | def writeFlow(filename,uv,v=None): function readFlowKITTI (line 102) | def readFlowKITTI(filename): function readDispKITTI (line 109) | def readDispKITTI(filename): function writeFlowKITTI (line 116) | def writeFlowKITTI(filename, uv): function read_gen (line 123) | def read_gen(file_name, pil=False): FILE: third_party/RAFT/core/utils/utils.py class InputPadder (line 7) | class InputPadder: method __init__ (line 9) | def __init__(self, dims, mode='sintel'): method pad (line 18) | def pad(self, *inputs): method unpad (line 21) | def unpad(self,x): function forward_interpolate (line 26) | def forward_interpolate(flow): function bilinear_sampler (line 57) | def bilinear_sampler(img, coords, mode='bilinear', mask=False): function coords_grid (line 74) | def coords_grid(batch, ht, wd, device): function upflow8 (line 80) | def upflow8(flow, mode='bilinear'): FILE: third_party/RAFT/demo.py function load_image (line 20) | def load_image(imfile): function viz (line 26) | def viz(img, flo): function demo (line 42) | def demo(args): FILE: third_party/RAFT/evaluate.py function create_sintel_submission (line 22) | def create_sintel_submission(model, iters=32, warm_start=False, output_p... function create_kitti_submission (line 54) | def create_kitti_submission(model, iters=24, output_path='kitti_submissi... function validate_chairs (line 75) | def validate_chairs(model, iters=24): function validate_sintel (line 96) | def validate_sintel(model, iters=32): function validate_kitti (line 131) | def validate_kitti(model, iters=24): FILE: third_party/RAFT/train.py class GradScaler (line 28) | class GradScaler: method __init__ (line 29) | def __init__(self): method scale (line 31) | def scale(self, loss): method unscale_ (line 33) | def unscale_(self, optimizer): method step (line 35) | def step(self, optimizer): method update (line 37) | def update(self): function sequence_loss (line 47) | def sequence_loss(flow_preds, flow_gt, valid, gamma=0.8, max_flow=MAX_FL... function count_parameters (line 75) | def count_parameters(model): function fetch_optimizer (line 79) | def fetch_optimizer(args, model): class Logger (line 89) | class Logger: method __init__ (line 90) | def __init__(self, model, scheduler): method _print_training_status (line 97) | def _print_training_status(self): method push (line 112) | def push(self, metrics): method write_dict (line 125) | def write_dict(self, results): method close (line 132) | def close(self): function train (line 136) | def train(args): FILE: train_stereoanyvideo.py function fetch_optimizer (line 29) | def fetch_optimizer(args, model): function forward_batch (line 49) | def forward_batch(batch, model, args): class Lite (line 69) | class Lite(LightningLite): method run (line 70) | def run(self, args): FILE: train_utils/logger.py class Logger (line 7) | class Logger: method __init__ (line 11) | def __init__(self, model, scheduler, ckpt_path): method _print_training_status (line 22) | def _print_training_status(self): method push (line 43) | def push(self, metrics, task): method update (line 50) | def update(self): method write_dict (line 57) | def write_dict(self, results): method close (line 64) | def close(self): FILE: train_utils/losses.py function sequence_loss (line 6) | def sequence_loss(flow_preds, flow_gt, valid, loss_gamma=0.9, max_flow=7... function temporal_loss (line 65) | def temporal_loss(flow_preds, flow_preds2, flow_gt, flow_gt2, valid, los... function compute_flow (line 126) | def compute_flow(Flow_Model, seq): function flow_warp (line 143) | def flow_warp(x, flow): function bidirectional_alignment (line 169) | def bidirectional_alignment(seq, flows_backward, flows_forward): function consistency_loss (line 202) | def consistency_loss(seq, disparities, Flow_Model, alpha=50): FILE: train_utils/utils.py function count_parameters (line 13) | def count_parameters(model): function run_test_eval (line 17) | def run_test_eval(ckpt_path, eval_type, evaluator, model, dataloaders, w... function fig2data (line 66) | def fig2data(fig): function save_ims_to_tb (line 89) | def save_ims_to_tb(writer, batch, output, total_steps):