SYMBOL INDEX (448 symbols across 78 files)

FILE: dataset/__init__.py
  function get_dataloader (line 20) | def get_dataloader(

FILE: dataset/dataset_one_frame_eval.py
  class nuScenes_One_Frame_Eval (line 16) | class nuScenes_One_Frame_Eval:
    method __init__ (line 17) | def __init__(
    method __len__ (line 67) | def __len__(self):
    method get_depth_from_lidar (line 71) | def get_depth_from_lidar(self, lidar_path, lidar2img, image_size):
    method __getitem__ (line 95) | def __getitem__(self, index):
    method read_surround_imgs (line 152) | def read_surround_imgs(self, img_paths):
    method get_data_info_temporal (line 163) | def get_data_info_temporal(self, info, info_tem):
    method get_data_info (line 209) | def get_data_info(self, info):

FILE: dataset/dataset_one_frame_sweeps_dist.py
  function get_xyz (line 15) | def get_xyz(pose_dict):
  function get_img2global (line 18) | def get_img2global(calib_dict, pose_dict):
  function get_lidar2global (line 35) | def get_lidar2global(calib_dict, pose_dict):
  class nuScenes_One_Frame_Sweeps_Dist (line 50) | class nuScenes_One_Frame_Sweeps_Dist:
    method __init__ (line 51) | def __init__(
    method __len__ (line 153) | def __len__(self):
    method get_depth_from_lidar (line 158) | def get_depth_from_lidar(self, lidar_path, lidar2img, image_size):
    method composite_dict (line 183) | def composite_dict(self, anchor_info):
    method __getitem__ (line 199) | def __getitem__(self, index):
    method read_surround_imgs (line 305) | def read_surround_imgs(self, img_paths, crop_size):
    method get_data_info_temporal (line 316) | def get_data_info_temporal(self, info, info_tem):
    method get_data_info_anchor (line 337) | def get_data_info_anchor(self, info, info_tem):
    method get_data_info (line 358) | def get_data_info(self, info):
  function list2tensor (line 458) | def list2tensor(imgs):
  function get_diagonal (line 471) | def get_diagonal(trans):
  function filter_invalid_pts (line 479) | def filter_invalid_pts(pts):
  function get_pixel (line 507) | def get_pixel(uv, eps=1e-5):
  function filter_invalid_points (line 528) | def filter_invalid_points(uv):
  function saveimg (line 537) | def saveimg(i, imgs, uv, name):

FILE: dataset/dataset_one_frame_sweeps_dist_vis.py
  function get_xyz (line 8) | def get_xyz(pose_dict):
  function get_img2global (line 11) | def get_img2global(calib_dict, pose_dict):
  function get_lidar2global (line 28) | def get_lidar2global(calib_dict, pose_dict):
  class nuScenes_One_Frame_Sweeps_Dist_Vis (line 43) | class nuScenes_One_Frame_Sweeps_Dist_Vis:
    method __init__ (line 44) | def __init__(
    method __len__ (line 73) | def __len__(self):
    method __getitem__ (line 77) | def __getitem__(self, index):
    method read_surround_imgs (line 109) | def read_surround_imgs(self, img_paths):
    method get_data_info (line 117) | def get_data_info(self, info):

FILE: dataset/dataset_wrapper_temporal.py
  class tpvformer_dataset_nuscenes_temporal (line 19) | class tpvformer_dataset_nuscenes_temporal(data.Dataset):
    method __init__ (line 20) | def __init__(
    method __len__ (line 77) | def __len__(self):
    method to_tensor (line 80) | def to_tensor(self, imgs):
    method __getitem__ (line 86) | def __getitem__(self, index):
    method deal_with_length3_dataset (line 93) | def deal_with_length3_dataset(self, data):
    method deal_with_length2_dataset (line 119) | def deal_with_length2_dataset(self, data):
  function custom_collate_fn_temporal (line 178) | def custom_collate_fn_temporal(data):
  function forward_aug (line 191) | def forward_aug(imgs, metas, transforms):

FILE: dataset/dataset_wrapper_vis.py
  class tpvformer_dataset_nuscenes_vis (line 14) | class tpvformer_dataset_nuscenes_vis(data.Dataset):
    method __init__ (line 15) | def __init__(
    method __len__ (line 35) | def __len__(self):
    method to_tensor (line 38) | def to_tensor(self, imgs):
    method __getitem__ (line 44) | def __getitem__(self, index):
    method deal_with_length2_dataset (line 48) | def deal_with_length2_dataset(self, data):
  function custom_collate_fn_temporal (line 68) | def custom_collate_fn_temporal(data):
  function forward_aug (line 81) | def forward_aug(imgs, metas, transforms):

FILE: dataset/kitti/helpers.py
  function apply_transform (line 17) | def apply_transform(pts, T):
  function dump_xyz (line 29) | def dump_xyz(P):
  function read_rgb (line 33) | def read_rgb(path):
  function read_poses (line 43) | def read_poses(path):
  function read_calib (line 56) | def read_calib(calib_path):

FILE: dataset/kitti/io_data.py
  function unpack (line 10) | def unpack(compressed):
  function img_normalize (line 25) | def img_normalize(img, mean, std):
  function pack (line 33) | def pack(array):
  function get_grid_coords (line 45) | def get_grid_coords(dims, resolution):
  function _get_remap_lut (line 77) | def _get_remap_lut(config_path):
  function get_inv_map (line 99) | def get_inv_map():
  function _read_SemKITTI (line 113) | def _read_SemKITTI(path, dtype, do_unpack):
  function _read_label_SemKITTI (line 120) | def _read_label_SemKITTI(path):
  function _read_invalid_SemKITTI (line 125) | def _read_invalid_SemKITTI(path):
  function _read_occluded_SemKITTI (line 130) | def _read_occluded_SemKITTI(path):
  function _read_occupancy_SemKITTI (line 135) | def _read_occupancy_SemKITTI(path):
  function _read_rgb_SemKITTI (line 140) | def _read_rgb_SemKITTI(path):
  function _read_pointcloud_SemKITTI (line 145) | def _read_pointcloud_SemKITTI(path):
  function _read_calib_SemKITTI (line 152) | def _read_calib_SemKITTI(calib_path):
  function get_remap_lut (line 173) | def get_remap_lut(path):
  function data_augmentation_3Dflips (line 196) | def data_augmentation_3Dflips(flip, data):
  function get_cmap_semanticKITTI20 (line 216) | def get_cmap_semanticKITTI20():

FILE: dataset/kitti/kitti_dataset_eval.py
  class Kitti_Novel_View_Eval (line 23) | class Kitti_Novel_View_Eval:
    method __init__ (line 24) | def __init__(
    method get_depth_from_lidar (line 287) | def get_depth_from_lidar(self, lidar_path, lidar2img, image_size):
    method __len__ (line 311) | def __len__(self):
    method prepare_temImg2lidar (line 314) | def prepare_temImg2lidar(self, scan, anchor):
    method prepare_img_metas (line 327) | def prepare_img_metas(self, scan):
    method read_surround_imgs (line 346) | def read_surround_imgs(self, img_paths):
    method __getitem__ (line 357) | def __getitem__(self, index):

FILE: dataset/kitti/kitti_dataset_one_frame.py
  class Kitti_One_Frame (line 21) | class Kitti_One_Frame:
    method __init__ (line 22) | def __init__(
    method get_depth_from_lidar (line 294) | def get_depth_from_lidar(self, lidar_path, lidar2img, image_size):
    method load_2d_sem_label (line 318) | def load_2d_sem_label(self, scan):
    method __len__ (line 325) | def __len__(self):
    method prepare_img_metas (line 328) | def prepare_img_metas(self, scan, anchor_scan, anchor_prev, anchor_next):
    method read_surround_imgs (line 375) | def read_surround_imgs(self, img_paths):
    method __getitem__ (line 386) | def __getitem__(self, index):

FILE: dataset/kitti_raw/kitti_raw_dataset.py
  class Kitti_Raw (line 20) | class Kitti_Raw:
    method __init__ (line 21) | def __init__(
    method _get_sequences (line 213) | def _get_sequences(data_path):
    method _load_split (line 228) | def _load_split(split_path):
    method _load_calibs (line 241) | def _load_calibs(data_path):
    method _load_poses (line 299) | def _load_poses(pose_path, sequences):
    method get_depth_from_lidar (line 325) | def get_depth_from_lidar(self, lidar_path, lidar2img, image_size):
    method load_depth (line 355) | def load_depth(self, lidar_path, lidar2img, image_size):
    method __len__ (line 404) | def __len__(self):
    method prepare_img_metas (line 407) | def prepare_img_metas(self, scan, anchor_scan, anchor_prev, anchor_next):
    method read_surround_imgs (line 471) | def read_surround_imgs(self, img_paths):
    method to_scan_idx (line 479) | def to_scan_idx(self, index):
    method __getitem__ (line 485) | def __getitem__(self, index):

FILE: dataset/kitti_raw/kitti_raw_dataset_stereo.py
  class Kitti_Raw_Stereo (line 20) | class Kitti_Raw_Stereo:
    method __init__ (line 24) | def __init__(
    method _get_sequences (line 221) | def _get_sequences(data_path):
    method _load_split (line 236) | def _load_split(split_path):
    method _load_calibs (line 249) | def _load_calibs(data_path):
    method _load_poses (line 307) | def _load_poses(pose_path, sequences):
    method get_depth_from_lidar (line 333) | def get_depth_from_lidar(self, lidar_path, lidar2img, image_size):
    method load_depth (line 363) | def load_depth(self, lidar_path, lidar2img, image_size):
    method __len__ (line 412) | def __len__(self):
    method prepare_img_metas (line 415) | def prepare_img_metas(self, scan, anchor_scan, anchor_prev, anchor_next):
    method read_surround_imgs (line 480) | def read_surround_imgs(self, img_paths):
    method to_scan_idx (line 488) | def to_scan_idx(self, index):
    method append_self (line 494) | def append_self(self, flag, obj):
    method __getitem__ (line 502) | def __getitem__(self, index):

FILE: dataset/loading.py
  class LoadMultiViewImageFromFilesHF (line 11) | class LoadMultiViewImageFromFilesHF:
    method __init__ (line 25) | def __init__(self,
    method load (line 34) | def load(self, results):
    method __call__ (line 69) | def __call__(self, results):
  class OriLoadMultiViewImageFromFilesHF (line 108) | class OriLoadMultiViewImageFromFilesHF(object):
    method __init__ (line 122) | def __init__(self,
    method __call__ (line 131) | def __call__(self, results):
    method __repr__ (line 177) | def __repr__(self):
  class LoadPtsFromFilesHF (line 185) | class LoadPtsFromFilesHF:
    method __init__ (line 187) | def __init__(self,
    method load (line 194) | def load(self, filename):
  class FFrecordClient (line 218) | class FFrecordClient(BaseStorageBackend):
    method __init__ (line 222) | def __init__(self, fname, check_data=False, filename2idx='filename2idx...
    method get (line 233) | def get(self, filepath: Union[str, Path]) -> bytes:
    method get_text (line 258) | def get_text(self,

FILE: dataset/sampler.py
  class CustomDistributedSampler (line 12) | class CustomDistributedSampler(Sampler[T_co]):
    method __init__ (line 59) | def __init__(self, dataset: Dataset, num_replicas: Optional[int] = None,
    method __iter__ (line 94) | def __iter__(self) -> Iterator[T_co]:
    method __len__ (line 122) | def __len__(self) -> int:
    method set_epoch (line 125) | def set_epoch(self, epoch: int) -> None:
    method set_last_iter (line 136) | def set_last_iter(self, last_iter: int):

FILE: dataset/transform_3d.py
  class PadMultiViewImage (line 8) | class PadMultiViewImage(object):
    method __init__ (line 19) | def __init__(self, size=None, size_divisor=None, pad_val=0):
    method _pad_img (line 29) | def _pad_img(self, results):
    method __call__ (line 61) | def __call__(self, results):
    method __repr__ (line 71) | def __repr__(self):
  class NormalizeMultiviewImage (line 79) | class NormalizeMultiviewImage(object):
    method __init__ (line 89) | def __init__(self, mean, std, to_rgb=True):
    method __call__ (line 95) | def __call__(self, results):
    method __repr__ (line 109) | def __repr__(self):
  class RandomFlip (line 114) | class RandomFlip(object):
    method __init__ (line 116) | def __init__(self, prob=0.5) -> None:
    method __call__ (line 119) | def __call__(self, results):
  class PhotoMetricDistortionMultiViewImage (line 126) | class PhotoMetricDistortionMultiViewImage:
    method __init__ (line 151) | def __init__(self,
    method __call__ (line 163) | def __call__(self, results):
    method __repr__ (line 222) | def __repr__(self):
  class RandomScaleImageMultiViewImage (line 323) | class RandomScaleImageMultiViewImage(object):
    method __init__ (line 329) | def __init__(self, scales=[], ref_focal_len=None, random_scale=None, p...
    method __call__ (line 341) | def __call__(self, results):
    method __repr__ (line 380) | def __repr__(self):
  function read_surround_imgs (line 399) | def read_surround_imgs(img_paths, crop_size=[768, 1600]):

FILE: dataset/utils.py
  function get_rm (line 4) | def get_rm(angle, axis, deg=False):

FILE: eval_depth.py
  function pass_print (line 19) | def pass_print(*args, **kwargs):
  function main (line 22) | def main(local_rank, args):

FILE: eval_iou.py
  function pass_print (line 23) | def pass_print(*args, **kwargs):
  function read_occ3d_label (line 26) | def read_occ3d_label(metas, nusc):
  function read_openoccupancy_label (line 34) | def read_openoccupancy_label(metas, nusc):
  function main (line 43) | def main(local_rank, args):

FILE: eval_iou_kitti.py
  function pass_print (line 21) | def pass_print(*args, **kwargs):
  function read_semantic_kitti (line 24) | def read_semantic_kitti(metas):
  function main (line 44) | def main(local_rank, args):

FILE: eval_novel_depth.py
  function pass_print (line 20) | def pass_print(*args, **kwargs):
  function main (line 23) | def main(local_rank, args):
  function evaluate_depth (line 285) | def evaluate_depth(gt_depth, pred_depth):
  function print_metrics (line 306) | def print_metrics(agg_depth_errors, n_frames, logger, num_cams=6):

FILE: eval_novel_depth_kitti.py
  function pass_print (line 20) | def pass_print(*args, **kwargs):
  function main (line 23) | def main(local_rank, args):
  function evaluate_depth (line 264) | def evaluate_depth(gt_depth, pred_depth):
  function print_metrics (line 281) | def print_metrics(agg_depth_errors, n_frames, logger):

FILE: examine_sweeps.py
  function gather_sensor (line 15) | def gather_sensor(sample_data_token):

FILE: generate_videos.py
  function cat_images (line 6) | def cat_images(dir, cam_img_size, pred_img_size, pred_img_size2, spacing...
  function get_video (line 70) | def get_video(img_path, video_path, fps, size):

FILE: loss/base_loss.py
  class BaseLoss (line 8) | class BaseLoss(nn.Module):
    method __init__ (line 19) | def __init__(
    method forward (line 34) | def forward(self, inputs):

FILE: loss/edge_loss_3d_ms.py
  function get_smooth_loss (line 7) | def get_smooth_loss(disp, img):
  class EdgeLoss3DMS (line 24) | class EdgeLoss3DMS(BaseLoss):
    method __init__ (line 26) | def __init__(self, weight=1.0, input_dict=None, **kwargs):
    method edge_loss (line 44) | def edge_loss(self, curr_imgs, ms_depths, ms_rays, ms_accs=None, max_d...

FILE: loss/eikonal_loss.py
  class EikonalLoss (line 6) | class EikonalLoss(BaseLoss):
    method __init__ (line 8) | def __init__(self, weight=1.0, input_dict=None, **kwargs):
    method eikonal_loss (line 19) | def eikonal_loss(self, eik_grad):

FILE: loss/multi_loss.py
  class MultiLoss (line 10) | class MultiLoss(nn.Module):
    method __init__ (line 12) | def __init__(self, loss_cfgs):
    method forward (line 24) | def forward(self, inputs):

FILE: loss/reproj_loss_mono_multi_new.py
  class SSIM (line 7) | class SSIM(nn.Module):
    method __init__ (line 10) | def __init__(self):
    method forward (line 23) | def forward(self, x, y):
  class ReprojLossMonoMultiNew (line 41) | class ReprojLossMonoMultiNew(BaseLoss):
    method __init__ (line 43) | def __init__(self, weight=1.0, input_dict=None, **kwargs):
    method reproj_loss (line 72) | def reproj_loss(

FILE: loss/reproj_loss_mono_multi_new_combine.py
  class SSIM (line 7) | class SSIM(nn.Module):
    method __init__ (line 10) | def __init__(self):
    method forward (line 23) | def forward(self, x, y):
  class ReprojLossMonoMultiNewCombine (line 41) | class ReprojLossMonoMultiNewCombine(BaseLoss):
    method __init__ (line 43) | def __init__(self, weight=1.0, input_dict=None, **kwargs):
    method reproj_loss (line 70) | def reproj_loss(

FILE: loss/rgb_loss_ms.py
  class SSIM (line 7) | class SSIM(nn.Module):
    method __init__ (line 10) | def __init__(self):
    method forward (line 23) | def forward(self, x, y):
  class RGBLossMS (line 41) | class RGBLossMS(BaseLoss):
    method __init__ (line 43) | def __init__(
    method rgb_loss (line 68) | def rgb_loss(self, ms_colors, ms_rays, gt_imgs):
  class SemLossMS (line 103) | class SemLossMS(BaseLoss):
    method __init__ (line 105) | def __init__(
    method sem_loss (line 126) | def sem_loss(self, sem, metas, ms_rays):
  class SemCELossMS (line 160) | class SemCELossMS(BaseLoss):
    method __init__ (line 162) | def __init__(
    method sem_loss (line 183) | def sem_loss(self, sem, metas, ms_rays):

FILE: loss/second_grad_loss.py
  class SecondGradLoss (line 6) | class SecondGradLoss(BaseLoss):
    method __init__ (line 8) | def __init__(self, weight=1.0, input_dict=None, **kwargs):
    method second_grad_loss (line 19) | def second_grad_loss(self, second_grad):

FILE: loss/sparsity_loss.py
  class SparsityLoss (line 7) | class SparsityLoss(BaseLoss):
    method __init__ (line 12) | def __init__(self, weight=1.0, scale=1.0, input_dict=None, **kwargs):
    method sparsity_loss (line 24) | def sparsity_loss(self, density):
  class HardSparsityLoss (line 30) | class HardSparsityLoss(BaseLoss):
    method __init__ (line 32) | def __init__(self, weight=1.0, scale=1.0, thresh=0.2, crop=[[0, 0], [0...
    method hard_sigmoid (line 46) | def hard_sigmoid(self, density):
    method hard_sparsity_loss (line 49) | def hard_sparsity_loss(self, density):
  class SoftSparsityLoss (line 67) | class SoftSparsityLoss(BaseLoss):
    method __init__ (line 69) | def __init__(self, weight=1.0, input_dict=None, **kwargs):
    method soft_sparsity_loss (line 80) | def soft_sparsity_loss(self, density):
  class AdaptiveSparsityLoss (line 85) | class AdaptiveSparsityLoss(BaseLoss):
    method __init__ (line 87) | def __init__(self, weight=1, input_dict=None, slack=4.0, **kwargs):
    method adaptive_sparsity_loss (line 102) | def adaptive_sparsity_loss(self, sdfs, ts, ms_depths):

FILE: model/backbone/unet2d.py
  class UpSampleBN (line 11) | class UpSampleBN(nn.Module):
    method __init__ (line 12) | def __init__(self, skip_input, output_features):
    method forward (line 25) | def forward(self, x, concat_with):
  class DecoderBN (line 36) | class DecoderBN(nn.Module):
    method __init__ (line 37) | def __init__(
    method forward (line 96) | def forward(self, features):
  class Encoder (line 144) | class Encoder(nn.Module):
    method __init__ (line 145) | def __init__(self, backend):
    method forward (line 149) | def forward(self, x):
  class UNet2D (line 161) | class UNet2D(nn.Module):
    method __init__ (line 162) | def __init__(self, out_feature, use_decoder=True):
    method forward (line 194) | def forward(self, x, **kwargs):
    method get_encoder_params (line 199) | def get_encoder_params(self):  # lr/10 learning rate
    method get_decoder_params (line 202) | def get_decoder_params(self):  # lr learning rate

FILE: model/encoder/base_encoder.py
  class BaseEncoder (line 6) | class BaseEncoder(BaseModule):
    method __init__ (line 11) | def __init__(self, init_cfg=None, **kwargs):
    method forward (line 14) | def forward(

FILE: model/encoder/bevformer/attention/image_cross_attention.py
  class BEVCrossAttention (line 12) | class BEVCrossAttention(BaseModule):
    method __init__ (line 18) | def __init__(self,
    method init_weight (line 40) | def init_weight(self):
    method forward (line 46) | def forward(self,
  class BEVDeformableAttention (line 149) | class BEVDeformableAttention(BaseModule):
    method __init__ (line 178) | def __init__(self,
    method init_weights (line 226) | def init_weights(self) -> None:
    method forward (line 247) | def forward(self,

FILE: model/encoder/bevformer/bevformer_encoder.py
  class BEVFormerEncoder (line 18) | class BEVFormerEncoder(BaseEncoder):
    method __init__ (line 20) | def __init__(
    method init_weights (line 119) | def init_weights(self):
    method forward_layers (line 135) | def forward_layers(
    method forward (line 171) | def forward(

FILE: model/encoder/bevformer/bevformer_encoder_layer.py
  class BEVFormerLayer (line 11) | class BEVFormerLayer(BaseModule):
    method __init__ (line 45) | def __init__(self,
    method forward (line 121) | def forward(self,

FILE: model/encoder/bevformer/bevformer_pos_embed.py
  class BEVPositionalEncoding (line 7) | class BEVPositionalEncoding(BaseModule):
    method __init__ (line 9) | def __init__(self, num_freqs, embed_dims,
    method forward (line 33) | def forward(self):

FILE: model/encoder/bevformer/mappings.py
  class LinearMapping (line 4) | class LinearMapping:
    method __init__ (line 6) | def __init__(
    method grid2meter (line 39) | def grid2meter(self, grid):
    method meter2grid (line 97) | def meter2grid(self, meter, normalize=False):
  class GridMeterMapping (line 153) | class GridMeterMapping:
    method __init__ (line 155) | def __init__(
  class NonLinearMapping (line 199) | class NonLinearMapping:
    method __init__ (line 201) | def __init__(
    method grid2meter (line 227) | def grid2meter(self, grid):
    method meter2grid (line 254) | def meter2grid(self, meter, normalize=False):

FILE: model/encoder/bevformer/mappings_old.py
  class GridMeterMapping (line 5) | class GridMeterMapping:
    method __init__ (line 7) | def __init__(
    method grid2meter (line 37) | def grid2meter(self, grid):
    method meter2grid (line 64) | def meter2grid(self, meter):

FILE: model/encoder/bevformer/utils.py
  function get_cross_view_ref_points (line 5) | def get_cross_view_ref_points(tpv_h, tpv_w, tpv_z, num_points_in_pillar):
  function get_reference_points (line 74) | def get_reference_points(H, W, Z=8, num_points_in_pillar=4, dim='3d', bs...
  function point_sampling (line 116) | def point_sampling(reference_points, img_metas):

FILE: model/encoder/tpvformer/attention/cross_view_hybrid_attention.py
  class CrossViewHybridAttention (line 12) | class CrossViewHybridAttention(MultiScaleDeformableAttention):
    method forward (line 17) | def forward(self,

FILE: model/encoder/tpvformer/attention/image_cross_attention.py
  class TPVCrossAttention (line 8) | class TPVCrossAttention(BaseModule):
    method __init__ (line 10) | def __init__(
    method forward (line 71) | def forward(self,

FILE: model/encoder/tpvformer/modules/camera_se_net.py
  class Mlp (line 4) | class Mlp(nn.Module):
    method __init__ (line 6) | def __init__(self,
    method forward (line 21) | def forward(self, x):
  class SELayer (line 44) | class SELayer(nn.Module):
    method __init__ (line 46) | def __init__(self, channels, act_layer=nn.ReLU, gate_layer=nn.Sigmoid):
    method forward (line 49) | def forward(self, x, x_se):
  class CameraAwareSE (line 52) | class CameraAwareSE(nn.Module):
    method __init__ (line 54) | def __init__(
    method _init_layers (line 65) | def _init_layers(self):
    method init_weight (line 87) | def init_weight(self):
    method forward (line 93) | def forward(self, ms_img_feats, metas):

FILE: model/encoder/tpvformer/modules/split_fpn.py
  class MultiPlaneFFN (line 8) | class MultiPlaneFFN(BaseModule):
    method __init__ (line 10) | def __init__(
    method _init_layers (line 29) | def _init_layers(self):
    method forward (line 35) | def forward(self, tpv, identity=None):

FILE: model/encoder/tpvformer/modules/split_norm.py
  class MultiPlaneNorm (line 8) | class MultiPlaneNorm(BaseModule):
    method __init__ (line 10) | def __init__(
    method _init_layers (line 22) | def _init_layers(self):
    method forward (line 28) | def forward(self, tpv):

FILE: model/encoder/tpvformer/tpvformer_encoder.py
  class TPVFormerEncoder (line 20) | class TPVFormerEncoder(BaseEncoder):
    method __init__ (line 22) | def __init__(
    method init_weights (line 174) | def init_weights(self):
    method forward_layers (line 192) | def forward_layers(
    method forward (line 234) | def forward(

FILE: model/encoder/tpvformer/tpvformer_encoder_layer.py
  class TPVFormerLayer (line 11) | class TPVFormerLayer(BaseModule):
    method __init__ (line 45) | def __init__(self,
    method forward (line 123) | def forward(self,

FILE: model/encoder/tpvformer/tpvformer_pos_embed.py
  function get_feat_from_meter (line 6) | def get_feat_from_meter(num_freqs, meter):
  class TPVPositionalEncoding (line 17) | class TPVPositionalEncoding(BaseModule):
    method __init__ (line 19) | def __init__(
    method forward (line 54) | def forward(self):

FILE: model/encoder/tpvformer/utils.py
  function get_cross_view_ref_points (line 5) | def get_cross_view_ref_points(tpv_h, tpv_w, tpv_z, num_points_in_pillar,...

FILE: model/head/base_head.py
  class BaseTaskHead (line 5) | class BaseTaskHead(BaseModule):
    method __init__ (line 11) | def __init__(self, init_cfg=None, **kwargs):
    method forward (line 14) | def forward(

FILE: model/head/nerfacc_head/bev_nerf.py
  class BEVNeRF (line 8) | class BEVNeRF(BaseModule):
    method __init__ (line 10) | def __init__(
    method _init_layers (line 62) | def _init_layers(self):
    method pre_compute_density_color (line 74) | def pre_compute_density_color(self, bev):
    method query_density (line 99) | def query_density(self, x):
    method forward (line 120) | def forward(self, x, condition=None):
    method forward_geo (line 155) | def forward_geo(self, x):

FILE: model/head/nerfacc_head/estimator.py
  class CustomOccGridEstimator (line 8) | class CustomOccGridEstimator(OccGridEstimator):
    method sampling (line 11) | def sampling(

FILE: model/head/nerfacc_head/img2lidar.py
  class Img2LiDAR (line 6) | class Img2LiDAR(nn.Module):
    method __init__ (line 8) | def __init__(
    method forward (line 25) | def forward(self, metas, rays):

FILE: model/head/nerfacc_head/nerfacc_head.py
  function namedtuple_map (line 13) | def namedtuple_map(fn, tup):
  class NeRFAccHead (line 21) | class NeRFAccHead(BaseTaskHead):
    method __init__ (line 23) | def __init__(
    method prepare (line 144) | def prepare(
    method render (line 153) | def render(
    method render_image_with_occgrid (line 242) | def render_image_with_occgrid(self, rays):
    method forward_occ (line 346) | def forward_occ(
    method get_uniform_sdf (line 366) | def get_uniform_sdf(self, aabb, resolution, device, shift=False):
    method forward (line 396) | def forward(

FILE: model/head/nerfacc_head/ray_sampler.py
  class RaySampler (line 5) | class RaySampler(nn.Module):
    method __init__ (line 7) | def __init__(
    method forward (line 48) | def forward(self):

FILE: model/head/nerfacc_head/rendering.py
  function custom_rendering (line 15) | def custom_rendering(

FILE: model/head/neus_head/neus_head.py
  class NeuSHead (line 22) | class NeuSHead(BaseTaskHead):
    method __init__ (line 24) | def __init__(
    method forward_occ (line 237) | def forward_occ(
    method get_uniform_sdf (line 265) | def get_uniform_sdf(self, aabb, resolution, device, shift=False):
    method prepare (line 295) | def prepare(
    method render (line 308) | def render(
    method forward (line 473) | def forward(
  function chunk_cams (line 716) | def chunk_cams(tensor, num_cams):

FILE: model/head/utils/sh_render.py
  function eval_sh_bases (line 35) | def eval_sh_bases(deg, dirs):
  function SHRender (line 84) | def SHRender(xyz_sampled, viewdirs, features, deg=2, act='relu'):

FILE: model/lifter/base_lifter.py
  class BaseLifter (line 6) | class BaseLifter(BaseModule):
    method __init__ (line 13) | def __init__(self, init_cfg=None, **kwargs) -> None:
    method forward (line 16) | def forward(

FILE: model/lifter/bev_query_lifter.py
  class BEVQueryLifter (line 7) | class BEVQueryLifter(BaseLifter):
    method __init__ (line 9) | def __init__(
    method forward (line 23) | def forward(self, ms_img_feats, *args, **kwargs):

FILE: model/lifter/tpv_pos_lifter.py
  function get_feat_from_meter (line 6) | def get_feat_from_meter(num_freqs, meter):
  class TPVPositionLifter (line 18) | class TPVPositionLifter(BaseLifter):
    method __init__ (line 20) | def __init__(
    method forward (line 81) | def forward(self, ms_img_feats, *args, **kwargs):

FILE: model/lifter/tpv_query_lifter.py
  class TPVQueryLifter (line 7) | class TPVQueryLifter(BaseLifter):
    method __init__ (line 9) | def __init__(
    method forward (line 27) | def forward(self, ms_img_feats, *args, **kwargs):

FILE: model/neck/identity_neck.py
  class IdentityNeck (line 6) | class IdentityNeck(BaseModule):
    method __init__ (line 8) | def __init__(self, init_cfg = None):
    method forward (line 11) | def forward(self, inputs):

FILE: model/segmentor/base_segmentor.py
  class CustomBaseSegmentor (line 7) | class CustomBaseSegmentor(BaseModule):
    method __init__ (line 9) | def __init__(
    method extract_img_feat (line 34) | def extract_img_feat(self, imgs, **kwargs):
    method forward (line 51) | def forward(

FILE: model/segmentor/tpv_segmentor.py
  class TPVSegmentor (line 10) | class TPVSegmentor(CustomBaseSegmentor):
    method __init__ (line 12) | def __init__(
    method extract_img_feat (line 36) | def extract_img_feat(self, imgs, metas, **kwargs):
    method forward_extra_img_backbone (line 71) | def forward_extra_img_backbone(self, imgs, **kwargs):
    method forward (line 87) | def forward(self,

FILE: train.py
  function pass_print (line 18) | def pass_print(*args, **kwargs):
  function main (line 21) | def main(local_rank, args):

FILE: utils/config_tools.py
  function modify_for_eval (line 10) | def modify_for_eval(cfg, dataset='nuscenes', novel_depth=False, args=None):

FILE: utils/feat_tools.py
  function multi2single_scale (line 4) | def multi2single_scale(

FILE: utils/metric_util.py
  function cityscapes2semantickitti (line 10) | def cityscapes2semantickitti(sem):
  function openseed2nuscenes (line 37) | def openseed2nuscenes(sem):
  class MeanIoU (line 66) | class MeanIoU:
    method __init__ (line 68) | def __init__(self,
    method reset (line 85) | def reset(self) -> None:
    method _after_step (line 90) | def _after_step(self, outputs, targets, mask=None):
    method _after_epoch (line 122) | def _after_epoch(self):
  class IoU (line 168) | class IoU(nn.Module):
    method __init__ (line 170) | def __init__(self, use_mask=False):
    method reset (line 186) | def reset(self) -> None:
    method _after_step (line 191) | def _after_step(self, outputs, targets, occ3d=False):
    method _after_step_occ3d (line 203) | def _after_step_occ3d(self, outputs, targets):
    method _after_epoch (line 221) | def _after_epoch(self):
  function cal_depth_metric (line 247) | def cal_depth_metric(depth_pred, depth_gt):
  class DepthMetric (line 282) | class DepthMetric(nn.Module):
    method __init__ (line 284) | def __init__(self, camera_names=['front'], eval_types=['raw', 'median']):
    method _reset (line 300) | def _reset(self):
    method _after_step (line 311) | def _after_step(self, depth_loc, depth_gt, depth_mask, depth_pred):
    method _after_epoch (line 351) | def _after_epoch(self):
  function compute_depth_errors (line 400) | def compute_depth_errors(gt, pred, min_depth=1e-3, max_depth=80):
  function compute_depth_errors_torch (line 424) | def compute_depth_errors_torch(gt, pred, min_depth=1e-3, max_depth=80):

FILE: utils/misc.py
  function _load_model_from_metafile (line 12) | def _load_model_from_metafile(model: str, scope: str = 'mmseg'):
  function _init_model (line 49) | def _init_model(
  function generate_segmentation_map_2d (line 85) | def generate_segmentation_map_2d(model_2d, inputs):
  function read_segmentation_map_2d (line 122) | def read_segmentation_map_2d(map_path, img_metas):

FILE: utils/openseed_utils.py
  function build_openseed_model (line 12) | def build_openseed_model():
  function forward_openseed_model (line 54) | def forward_openseed_model(model, img, size):

FILE: utils/scenerf_metric.py
  function get_iou (line 7) | def get_iou(iou_sum, cnt_class):
  function get_accuracy (line 17) | def get_accuracy(predict, target, weight=None):  # 0.05s
  class SSCMetrics (line 39) | class SSCMetrics:
    method __init__ (line 40) | def __init__(self, n_classes):
    method hist_info (line 44) | def hist_info(self, n_cl, pred, gt):
    method compute_score (line 59) | def compute_score(hist, correct, labeled):
    method add_batch (line 69) | def add_batch(self, y_pred, y_true, nonempty=None, nonsurface=None):
    method get_stats (line 93) | def get_stats(self):
    method reset (line 120) | def reset(self):
    method get_score_completion (line 140) | def get_score_completion(self, predict, target, nonempty=None):
    method get_score_semantic_and_completion (line 177) | def get_score_semantic_and_completion(self, predict, target, nonempty=...

FILE: utils/tb_wrapper.py
  class WrappedTBWriter (line 4) | class WrappedTBWriter(SummaryWriter, ManagerMixin):
    method __init__ (line 6) | def __init__(self, name, **kwargs):

FILE: utils/temporal_pkl.py
  function arange_according_to_scene (line 4) | def arange_according_to_scene(infos, nusc):

FILE: vis_3d.py
  function pass_print (line 37) | def pass_print(*args, **kwargs):
  function read_semantic_kitti (line 42) | def read_semantic_kitti(metas, idx):
  function read_occ3d_label (line 65) | def read_occ3d_label(metas, nusc, idx):
  function get_grid_coords (line 75) | def get_grid_coords(dims, resolution):
  function draw (line 97) | def draw(
  function main (line 344) | def main(local_rank, args):

FILE: vis_3d_scene.py
  function pass_print (line 38) | def pass_print(*args, **kwargs):
  function read_semantic_kitti (line 43) | def read_semantic_kitti(metas, idx):
  function read_occ3d_label (line 66) | def read_occ3d_label(metas, nusc, idx):
  function draw (line 76) | def draw(
  function main (line 228) | def main(local_rank, args):

FILE: vis_pics.py
  function pass_print (line 26) | def pass_print(*args, **kwargs):
  function create_label_colormap (line 29) | def create_label_colormap():
  function save_depth_map (line 56) | def save_depth_map(disp_resized_np, save_path):
  function main (line 65) | def main(local_rank, args):