SYMBOL INDEX (265 symbols across 27 files)

FILE: load_llff.py
  function _minify (line 10) | def _minify(basedir, factors=[], resolutions=[]):
  function _load_data (line 62) | def _load_data(basedir, factor=None, width=None, height=None, load_imgs=...
  function normalize (line 199) | def normalize(x):
  function viewmatrix (line 202) | def viewmatrix(z, up, pos):
  function poses_avg (line 211) | def poses_avg(poses):
  function render_path_spiral (line 224) | def render_path_spiral(c2w, up, rads, focal, zdelta, zrate, rots, N):
  function recenter_poses (line 241) | def recenter_poses(poses):
  function spherify_poses (line 256) | def spherify_poses(poses, bds):
  function load_llff_data (line 315) | def load_llff_data(args, basedir,
  function generate_path (line 367) | def generate_path(c2w, args):

FILE: render_utils.py
  function batchify_rays (line 14) | def batchify_rays(t, chain_5frames,
  function render (line 30) | def render(t, chain_5frames,
  function render_path_batch (line 106) | def render_path_batch(render_poses, time2render,
  function render_path (line 190) | def render_path(render_poses,
  function raw2outputs (line 322) | def raw2outputs(raw_s,
  function raw2outputs_d (line 416) | def raw2outputs_d(raw_d,
  function render_rays (line 463) | def render_rays(t,

FILE: run_nerf.py
  function config_parser (line 16) | def config_parser():
  function train (line 151) | def train():

FILE: run_nerf_helpers.py
  function img2mse (line 12) | def img2mse(x, y, M=None):
  function img2mae (line 19) | def img2mae(x, y, M=None):
  function L1 (line 26) | def L1(x, M=None):
  function L2 (line 33) | def L2(x, M=None):
  function entropy (line 40) | def entropy(x):
  function mse2psnr (line 44) | def mse2psnr(x): return -10. * torch.log(x) / torch.log(torch.Tensor([10...
  function to8b (line 47) | def to8b(x): return (255 * np.clip(x, 0, 1)).astype(np.uint8)
  class Embedder (line 50) | class Embedder:
    method __init__ (line 52) | def __init__(self, **kwargs):
    method create_embedding_fn (line 57) | def create_embedding_fn(self):
    method embed (line 83) | def embed(self, inputs):
  function get_embedder (line 87) | def get_embedder(multires, i=0, input_dims=3):
  class NeRF_d (line 107) | class NeRF_d(nn.Module):
    method __init__ (line 108) | def __init__(self, D=8, W=256, input_ch=3, input_ch_views=3, output_ch...
    method forward (line 134) | def forward(self, x):
  class NeRF_s (line 166) | class NeRF_s(nn.Module):
    method __init__ (line 167) | def __init__(self, D=8, W=256, input_ch=3, input_ch_views=3, output_ch...
    method forward (line 192) | def forward(self, x):
  function batchify (line 219) | def batchify(fn, chunk):
  function run_network (line 230) | def run_network(inputs, viewdirs, fn, embed_fn, embeddirs_fn, netchunk=1...
  function create_nerf (line 249) | def create_nerf(args):
  function get_rays (line 361) | def get_rays(H, W, focal, c2w):
  function ndc_rays (line 374) | def ndc_rays(H, W, focal, near, rays_o, rays_d):
  function get_grid (line 409) | def get_grid(H, W, num_img, flows_f, flow_masks_f, flows_b, flow_masks_b):
  function NDC2world (line 433) | def NDC2world(pts, H, W, f):
  function render_3d_point (line 444) | def render_3d_point(H, W, f, pose, weights, pts):
  function induce_flow (line 471) | def induce_flow(H, W, focal, pose_neighbor, weights, pts_3d_neighbor, pt...
  function compute_depth_loss (line 483) | def compute_depth_loss(dyn_depth, gt_depth):
  function normalize_depth (line 496) | def normalize_depth(depth):
  function percentile (line 500) | def percentile(t, q):
  function save_res (line 519) | def save_res(moviebase, ret, fps=None):
  function norm_sf_channel (line 577) | def norm_sf_channel(sf_ch):
  function norm_sf (line 586) | def norm_sf(sf):
  function compute_sf_smooth_s_loss (line 596) | def compute_sf_smooth_s_loss(pts1, pts2, H, W, f):
  function compute_sf_smooth_loss (line 611) | def compute_sf_smooth_loss(pts, pts_f, pts_b, H, W, f):

FILE: utils/RAFT/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 CorrLayer (line 63) | class CorrLayer(torch.autograd.Function):
    method forward (line 65) | def forward(ctx, fmap1, fmap2, coords, r):
    method backward (line 75) | def backward(ctx, grad_corr):
  class AlternateCorrBlock (line 83) | class AlternateCorrBlock:
    method __init__ (line 84) | def __init__(self, fmap1, fmap2, num_levels=4, radius=4):
    method __call__ (line 94) | def __call__(self, coords):

FILE: utils/RAFT/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: utils/RAFT/demo.py
  function load_image (line 18) | def load_image(imfile):
  function load_image_list (line 24) | def load_image_list(image_files):
  function viz (line 36) | def viz(img, flo):
  function demo (line 50) | def demo(args):
  function RAFT_infer (line 71) | def RAFT_infer(args):

FILE: utils/RAFT/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: utils/RAFT/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 59) | def freeze_bn(self):
    method initialize_flow (line 64) | def initialize_flow(self, img):
    method upsample_flow (line 73) | def upsample_flow(self, flow, mask):
    method forward (line 87) | def forward(self, image1, image2, iters=12, flow_init=None, upsample=T...

FILE: utils/RAFT/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: utils/RAFT/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: utils/RAFT/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: utils/RAFT/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: utils/RAFT/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):
  function upflow8 (line 80) | def upflow8(flow, mode='bilinear'):

FILE: utils/colmap_utils.py
  class Image (line 48) | class Image(BaseImage):
    method qvec2rotmat (line 49) | def qvec2rotmat(self):
  function read_next_bytes (line 70) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact...
  function read_cameras_text (line 82) | def read_cameras_text(path):
  function read_cameras_binary (line 108) | def read_cameras_binary(path_to_model_file):
  function read_images_text (line 137) | def read_images_text(path):
  function read_images_binary (line 168) | def read_images_binary(path_to_model_file):
  function read_points3D_text (line 203) | def read_points3D_text(path):
  function read_points3d_binary (line 230) | def read_points3d_binary(path_to_model_file):
  function read_model (line 260) | def read_model(path, ext):
  function qvec2rotmat (line 272) | def qvec2rotmat(qvec):
  function rotmat2qvec (line 285) | def rotmat2qvec(R):
  function main (line 299) | def main():

FILE: utils/evaluation.py
  function im2tensor (line 9) | def im2tensor(img):
  function create_dir (line 13) | def create_dir(dir):
  function readimage (line 18) | def readimage(data_dir, sequence, time, method):
  function calculate_metrics (line 23) | def calculate_metrics(data_dir, sequence, methods, lpips_loss):

FILE: utils/flow_utils.py
  function flow_to_image (line 8) | def flow_to_image(flow, global_max=None):
  function compute_color (line 50) | def compute_color(u, v):
  function make_color_wheel (line 94) | def make_color_wheel():
  function resize_flow (line 144) | def resize_flow(flow, H_new, W_new):
  function warp_flow (line 153) | def warp_flow(img, flow):
  function consistCheck (line 165) | def consistCheck(flowB, flowF):
  function read_optical_flow (line 188) | def read_optical_flow(basedir, img_i_name, read_fwd):
  function compute_epipolar_distance (line 204) | def compute_epipolar_distance(T_21, K, p_1, p_2):
  function skew (line 223) | def skew(x):

FILE: utils/generate_data.py
  function create_dir (line 13) | def create_dir(dir):
  function multi_view_multi_time (line 18) | def multi_view_multi_time(args):

FILE: utils/generate_depth.py
  function create_dir (line 15) | def create_dir(dir):
  function read_image (line 20) | def read_image(path):
  function run (line 39) | def run(input_path, output_path, output_img_path, model_path):

FILE: utils/generate_flow.py
  function create_dir (line 18) | def create_dir(dir):
  function load_image (line 23) | def load_image(imfile):
  function warp_flow (line 29) | def warp_flow(img, flow):
  function compute_fwdbwd_mask (line 39) | def compute_fwdbwd_mask(fwd_flow, bwd_flow):
  function run (line 56) | def run(args, input_path, output_path, output_img_path):

FILE: utils/generate_motion_mask.py
  function create_dir (line 17) | def create_dir(dir):
  function extract_poses (line 22) | def extract_poses(im):
  function load_colmap_data (line 32) | def load_colmap_data(realdir):
  function run_maskrcnn (line 60) | def run_maskrcnn(model, img_path, intWidth=1024, intHeight=576):
  function motion_segmentation (line 111) | def motion_segmentation(basedir, threshold,

FILE: utils/generate_pose.py
  function load_colmap_data (line 8) | def load_colmap_data(realdir):

FILE: utils/midas/base_model.py
  class BaseModel (line 4) | class BaseModel(torch.nn.Module):
    method load (line 5) | def load(self, path):

FILE: utils/midas/blocks.py
  function _make_encoder (line 11) | def _make_encoder(backbone, features, use_pretrained, groups=1, expand=F...
  function _make_scratch (line 49) | def _make_scratch(in_shape, out_shape, groups=1, expand=False):
  function _make_pretrained_efficientnet_lite3 (line 78) | def _make_pretrained_efficientnet_lite3(use_pretrained, exportable=False):
  function _make_efficientnet_backbone (line 88) | def _make_efficientnet_backbone(effnet):
  function _make_resnet_backbone (line 101) | def _make_resnet_backbone(resnet):
  function _make_pretrained_resnext101_wsl (line 114) | def _make_pretrained_resnext101_wsl(use_pretrained):
  class Interpolate (line 120) | class Interpolate(nn.Module):
    method __init__ (line 124) | def __init__(self, scale_factor, mode, align_corners=False):
    method forward (line 138) | def forward(self, x):
  class ResidualConvUnit (line 155) | class ResidualConvUnit(nn.Module):
    method __init__ (line 159) | def __init__(self, features):
    method forward (line 177) | def forward(self, x):
  class FeatureFusionBlock (line 194) | class FeatureFusionBlock(nn.Module):
    method __init__ (line 198) | def __init__(self, features):
    method forward (line 209) | def forward(self, *xs):
  class ResidualConvUnit_custom (line 231) | class ResidualConvUnit_custom(nn.Module):
    method __init__ (line 235) | def __init__(self, features, activation, bn):
    method forward (line 263) | def forward(self, x):
  class FeatureFusionBlock_custom (line 291) | class FeatureFusionBlock_custom(nn.Module):
    method __init__ (line 295) | def __init__(self, features, activation, deconv=False, bn=False, expan...
    method forward (line 320) | def forward(self, *xs):

FILE: utils/midas/midas_net.py
  class MidasNet (line 12) | class MidasNet(BaseModel):
    method __init__ (line 16) | def __init__(self, path=None, features=256, non_negative=True):
    method forward (line 49) | def forward(self, x):

FILE: utils/midas/transforms.py
  function apply_min_size (line 6) | def apply_min_size(sample, size, image_interpolation_method=cv2.INTER_AR...
  class Resize (line 48) | class Resize(object):
    method __init__ (line 52) | def __init__(
    method constrain_to_multiple_of (line 94) | def constrain_to_multiple_of(self, x, min_val=0, max_val=None):
    method get_size (line 105) | def get_size(self, width, height):
    method __call__ (line 162) | def __call__(self, sample):
  class NormalizeImage (line 197) | class NormalizeImage(object):
    method __init__ (line 201) | def __init__(self, mean, std):
    method __call__ (line 205) | def __call__(self, sample):
  class PrepareForNet (line 211) | class PrepareForNet(object):
    method __init__ (line 215) | def __init__(self):
    method __call__ (line 218) | def __call__(self, sample):

FILE: utils/midas/vit.py
  class Slice (line 9) | class Slice(nn.Module):
    method __init__ (line 10) | def __init__(self, start_index=1):
    method forward (line 14) | def forward(self, x):
  class AddReadout (line 18) | class AddReadout(nn.Module):
    method __init__ (line 19) | def __init__(self, start_index=1):
    method forward (line 23) | def forward(self, x):
  class ProjectReadout (line 31) | class ProjectReadout(nn.Module):
    method __init__ (line 32) | def __init__(self, in_features, start_index=1):
    method forward (line 38) | def forward(self, x):
  class Transpose (line 45) | class Transpose(nn.Module):
    method __init__ (line 46) | def __init__(self, dim0, dim1):
    method forward (line 51) | def forward(self, x):
  function forward_vit (line 56) | def forward_vit(pretrained, x):
  function _resize_pos_embed (line 100) | def _resize_pos_embed(self, posemb, gs_h, gs_w):
  function forward_flex (line 117) | def forward_flex(self, x):
  function get_activation (line 159) | def get_activation(name):
  function get_readout_oper (line 166) | def get_readout_oper(vit_features, features, use_readout, start_index=1):
  function _make_vit_b16_backbone (line 183) | def _make_vit_b16_backbone(
  function _make_pretrained_vitl16_384 (line 297) | def _make_pretrained_vitl16_384(pretrained, use_readout="ignore", hooks=...
  function _make_pretrained_vitb16_384 (line 310) | def _make_pretrained_vitb16_384(pretrained, use_readout="ignore", hooks=...
  function _make_pretrained_deitb16_384 (line 319) | def _make_pretrained_deitb16_384(pretrained, use_readout="ignore", hooks...
  function _make_pretrained_deitb16_distil_384 (line 328) | def _make_pretrained_deitb16_distil_384(pretrained, use_readout="ignore"...
  function _make_vit_b_rn50_backbone (line 343) | def _make_vit_b_rn50_backbone(
  function _make_pretrained_vitb_rn50_384 (line 478) | def _make_pretrained_vitb_rn50_384(