SYMBOL INDEX (438 symbols across 57 files)

FILE: other_baselines/lifted_2d/eval_lifted_2d.py
  function cm_to_ious (line 37) | def cm_to_ious(cm):
  function fast_hist (line 48) | def fast_hist(pred, label, max_label=18):

FILE: other_baselines/static_world/eval_static_world.py
  function cm_to_ious (line 23) | def cm_to_ious(cm):
  function fast_hist (line 34) | def fast_hist(pred, label, max_label=18):

FILE: other_baselines/voxel_pcp/eval_voxel_pcp.py
  function voxel2world (line 36) | def voxel2world(voxel):
  function world2voxel (line 42) | def world2voxel(world):
  function cm_to_ious (line 48) | def cm_to_ious(cm):
  function fast_hist (line 59) | def fast_hist(pred, label, max_label=18):
  function nb_process_label (line 66) | def nb_process_label(processed_label, sorted_label_voxel_pair):
  function get_ego2lidar_pose (line 82) | def get_ego2lidar_pose(rec):
  function get_lidar_pose (line 90) | def get_lidar_pose(rec):

FILE: projects/occ_plugin/core/evaluation/efficiency_hooks.py
  class OccEfficiencyHook (line 12) | class OccEfficiencyHook(Hook):
    method __init__ (line 13) | def __init__(self, dataloader,  **kwargs):
    method construct_input (line 17) | def construct_input(self, DUMMY_SHAPE=None, m_info=None):
    method before_run (line 40) | def before_run(self, runner):
    method after_run (line 95) | def after_run(self, runner):
    method before_epoch (line 98) | def before_epoch(self, runner):
    method after_epoch (line 101) | def after_epoch(self, runner):
    method before_iter (line 104) | def before_iter(self, runner):
    method after_iter (line 107) | def after_iter(self, runner):

FILE: projects/occ_plugin/core/evaluation/eval_hooks.py
  class OccEvalHook (line 13) | class OccEvalHook(BaseEvalHook):
    method __init__ (line 14) | def __init__(self, *args,  **kwargs):
    method _do_evaluate (line 17) | def _do_evaluate(self, runner):
  class OccDistEvalHook (line 31) | class OccDistEvalHook(BaseDistEvalHook):
    method __init__ (line 32) | def __init__(self, *args,  **kwargs):
    method _do_evaluate (line 35) | def _do_evaluate(self, runner):

FILE: projects/occ_plugin/core/visualizer/show_occ.py
  function save_occ (line 8) | def save_occ(pred_c, pred_f, img_metas, path, visible_mask=None, gt_occ=...

FILE: projects/occ_plugin/datasets/builder.py
  function build_dataloader (line 18) | def build_dataloader(dataset,
  function worker_init_fn (line 94) | def worker_init_fn(worker_id, num_workers, rank, seed):
  function custom_build_dataset (line 121) | def custom_build_dataset(cfg, default_args=None):

FILE: projects/occ_plugin/datasets/cam4docc_dataset.py
  class Cam4DOccDataset (line 20) | class Cam4DOccDataset(NuScenesDataset):
    method __init__ (line 21) | def __init__(self, occ_size, pc_range, occ_root, idx_root, ori_data_ro...
    method _set_group_flag (line 79) | def _set_group_flag(self):
    method __len__ (line 85) | def __len__(self):
    method __getitem__ (line 91) | def __getitem__(self, idx):
    method get_indices (line 112) | def get_indices(self):
    method get_lidar_pose (line 141) | def get_lidar_pose(self, rec):
    method get_ego2lidar_pose (line 152) | def get_ego2lidar_pose(self, rec):
    method record_instance (line 162) | def record_instance(self, idx, instance_map):
    method get_future_egomotion (line 240) | def get_future_egomotion(self, idx):
    method _check_consistency (line 273) | def _check_consistency(translation, prev_translation, threshold=1.0):
    method refine_instance_poly (line 284) | def refine_instance_poly(self, instance):
    method prepare_train_data (line 308) | def prepare_train_data(self, index):
    method prepare_test_data (line 319) | def prepare_test_data(self, index):
    method prepare_sequential_data (line 332) | def prepare_sequential_data(self, index):
    method get_data_info (line 386) | def get_data_info(self, index):
    method evaluate (line 455) | def evaluate(self, results, logger=None, **kawrgs):

FILE: projects/occ_plugin/datasets/cam4docc_lyft_dataset.py
  class Cam4DOccLyftDataset (line 23) | class Cam4DOccLyftDataset(Dataset):
    method __init__ (line 24) | def __init__(self, occ_size, pc_range, occ_root, idx_root, ori_data_ro...
    method _set_group_flag (line 108) | def _set_group_flag(self):
    method __len__ (line 114) | def __len__(self):
    method __getitem__ (line 120) | def __getitem__(self, idx):
    method get_scenes (line 141) | def get_scenes(self):
    method get_samples (line 151) | def get_samples(self):
    method get_indices (line 163) | def get_indices(self):
    method get_lidar_pose (line 192) | def get_lidar_pose(self, rec):
    method get_ego2lidar_pose (line 205) | def get_ego2lidar_pose(self, rec):
    method record_instance (line 218) | def record_instance(self, idx, instance_map):
    method get_future_egomotion (line 287) | def get_future_egomotion(self, idx):
    method _check_consistency (line 326) | def _check_consistency(translation, prev_translation, threshold=1.0):
    method refine_instance_poly (line 337) | def refine_instance_poly(self, instance):
    method prepare_train_data (line 361) | def prepare_train_data(self, index):
    method prepare_test_data (line 369) | def prepare_test_data(self, index):
    method prepare_sequential_data (line 379) | def prepare_sequential_data(self, index):
    method evaluate (line 521) | def evaluate(self, results, logger=None, **kawrgs):

FILE: projects/occ_plugin/datasets/nuscenes_dataset.py
  class CustomNuScenesDataset (line 16) | class CustomNuScenesDataset(NuScenesDataset):
    method __init__ (line 22) | def __init__(self, queue_length=4, bev_size=(200, 200), overlap_test=F...
    method prepare_train_data (line 28) | def prepare_train_data(self, index):
    method union2one (line 55) | def union2one(self, queue):
    method get_data_info (line 83) | def get_data_info(self, index):
    method __getitem__ (line 166) | def __getitem__(self, idx):

FILE: projects/occ_plugin/datasets/pipelines/formating.py
  class OccDefaultFormatBundle3D (line 11) | class OccDefaultFormatBundle3D(DefaultFormatBundle3D):
    method __init__ (line 23) | def __init__(self, *args, **kwargs):
    method __call__ (line 27) | def __call__(self, results):

FILE: projects/occ_plugin/datasets/pipelines/loading_bevdet.py
  function mmlabNormalize (line 15) | def mmlabNormalize(img, img_norm_cfg=None):
  function depth_transform (line 32) | def depth_transform(cam_depth, resize, resize_dims, crop, flip, rotate):
  class LoadMultiViewImageFromFiles_BEVDet (line 80) | class LoadMultiViewImageFromFiles_BEVDet(object):
    method __init__ (line 91) | def __init__(self, data_config, is_train=False, using_ego=True, colorj...
    method get_rot (line 117) | def get_rot(self,h):
    method img_transform (line 123) | def img_transform(self, img, post_rot, post_tran,
    method img_transform_core (line 145) | def img_transform_core(self, img, resize_dims, crop, flip, rotate):
    method choose_cams (line 154) | def choose_cams(self):
    method sample_augmentation (line 162) | def sample_augmentation(self, H , W, flip=None, scale=None):
    method get_sensor2ego_transformation (line 189) | def get_sensor2ego_transformation(self, cam_info, key_info, cam_name):
    method get_sensor2lidar_transformation (line 242) | def get_sensor2lidar_transformation(self, cam_info, cam_name, sample_i...
    method get_seq_inputs (line 293) | def get_seq_inputs(self, results, flip=None, scale=None):
    method __call__ (line 427) | def __call__(self, results):
  function bev_transform (line 433) | def bev_transform(rotate_angle, scale_ratio, flip_dx, flip_dy):
  class LoadAnnotationsBEVDepth (line 450) | class LoadAnnotationsBEVDepth():
    method __init__ (line 451) | def __init__(self, bda_aug_conf, classes, is_train=True,
    method sample_bda_augmentation (line 465) | def sample_bda_augmentation(self):
    method __call__ (line 479) | def __call__(self, results):
  class PhotoMetricDistortionMultiViewImage (line 498) | class PhotoMetricDistortionMultiViewImage(object):
    method __init__ (line 517) | def __init__(self,
    method __call__ (line 527) | def __call__(self, img):

FILE: projects/occ_plugin/datasets/pipelines/loading_instance.py
  class LoadInstanceWithFlow (line 14) | class LoadInstanceWithFlow(object):
    method __init__ (line 15) | def __init__(self, cam4docc_dataset_path, grid_size=[512, 512, 40], pc...
    method get_poly_region (line 45) | def get_poly_region(self, instance_annotation, present_egopose, presen...
    method fill_occupancy (line 78) | def fill_occupancy(self, occ_instance, occ_segmentation, occ_attribute...
    method get_label (line 120) | def get_label(self, input_seq_data):
    method generate_flow (line 173) | def generate_flow(flow, occ_instance_seq, instance, instance_id):
    method get_flow_label (line 217) | def get_flow_label(self, input_seq_data, ignore_index=255):
    method convert_instance_mask_to_center_and_offset_label (line 239) | def convert_instance_mask_to_center_and_offset_label(input_seq_data, i...
    method __call__ (line 270) | def __call__(self, results):

FILE: projects/occ_plugin/datasets/pipelines/loading_occupancy.py
  class LoadOccupancy (line 15) | class LoadOccupancy(object):
    method __init__ (line 17) | def __init__(self, to_float32=True, occ_path=None, grid_size=[512, 512...
    method get_seq_pseudo_occ (line 34) | def get_seq_pseudo_occ(self, results):
    method get_seq_occ (line 46) | def get_seq_occ(self, results):
    method __call__ (line 114) | def __call__(self, results):
    method voxel2world (line 122) | def voxel2world(self, voxel):
    method world2voxel (line 129) | def world2voxel(self, world):
    method __repr__ (line 136) | def __repr__(self):
    method project_points (line 142) | def project_points(self, points, rots, trans, intrins, post_rots, post...
  function nb_process_img_points (line 164) | def nb_process_img_points(basic_valid_occ, depth_canva, nb_valid_mask):
  function nb_process_label_withvel (line 182) | def nb_process_label_withvel(processed_label, sorted_label_voxel_pair):
  function nb_process_label (line 201) | def nb_process_label(processed_label, sorted_label_voxel_pair):

FILE: projects/occ_plugin/datasets/pipelines/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 27) | def _pad_img(self, results):
    method __call__ (line 43) | def __call__(self, results):
    method __repr__ (line 53) | def __repr__(self):
  class NormalizeMultiviewImage (line 62) | class NormalizeMultiviewImage(object):
    method __init__ (line 72) | def __init__(self, mean, std, to_rgb=True):
    method __call__ (line 78) | def __call__(self, results):
    method __repr__ (line 92) | def __repr__(self):
  class PhotoMetricDistortionMultiViewImage (line 99) | class PhotoMetricDistortionMultiViewImage:
    method __init__ (line 118) | def __init__(self,
    method __call__ (line 128) | def __call__(self, results):
    method __repr__ (line 187) | def __repr__(self):
  class CustomCollect3D (line 200) | class CustomCollect3D(object):
    method __init__ (line 244) | def __init__(self,
    method __call__ (line 258) | def __call__(self, results):
    method __repr__ (line 281) | def __repr__(self):
  class CustomOccCollect3D (line 287) | class CustomOccCollect3D(object):
    method __init__ (line 331) | def __init__(self,
    method __call__ (line 345) | def __call__(self, results):
    method __repr__ (line 373) | def __repr__(self):
  class RandomScaleImageMultiViewImage (line 379) | class RandomScaleImageMultiViewImage(object):
    method __init__ (line 385) | def __init__(self, scales=[]):
    method __call__ (line 389) | def __call__(self, results):
    method __repr__ (line 414) | def __repr__(self):

FILE: projects/occ_plugin/datasets/samplers/distributed_sampler.py
  class DistributedSampler (line 9) | class DistributedSampler(_DistributedSampler):
    method __init__ (line 11) | def __init__(self,
    method __iter__ (line 22) | def __iter__(self):

FILE: projects/occ_plugin/datasets/samplers/group_sampler.py
  class DistributedGroupSampler (line 14) | class DistributedGroupSampler(Sampler):
    method __init__ (line 32) | def __init__(self,
    method __iter__ (line 61) | def __iter__(self):
    method __len__ (line 104) | def __len__(self):
    method set_epoch (line 107) | def set_epoch(self, epoch):

FILE: projects/occ_plugin/datasets/samplers/sampler.py
  function build_sampler (line 6) | def build_sampler(cfg, default_args):

FILE: projects/occ_plugin/occupancy/apis/mmdet_train.py
  function custom_train_detector (line 27) | def custom_train_detector(model,

FILE: projects/occ_plugin/occupancy/apis/test.py
  function custom_encode_mask_results (line 25) | def custom_encode_mask_results(mask_results):
  function custom_single_gpu_test (line 45) | def custom_single_gpu_test(model, data_loader, show=False, out_dir=None,...
  function custom_multi_gpu_test (line 74) | def custom_multi_gpu_test(model, data_loader, tmpdir=None, gpu_collect=F...
  function collect_results_cpu (line 141) | def collect_results_cpu(result_part, size, tmpdir=None, type='list'):

FILE: projects/occ_plugin/occupancy/apis/train.py
  function custom_train_model (line 5) | def custom_train_model(model,
  function train_model (line 30) | def train_model(model,

FILE: projects/occ_plugin/occupancy/backbones/pred_block.py
  class Residual (line 15) | class Residual(nn.Module):
    method __init__ (line 16) | def __init__(
    method forward (line 51) | def forward(self, x):
  class Predictor (line 60) | class Predictor(nn.Module):
    method __init__ (line 61) | def __init__(
    method forward (line 80) | def forward(self, x):

FILE: projects/occ_plugin/occupancy/backbones/resnet3d.py
  function get_inplanes (line 13) | def get_inplanes():
  function conv3x3x3 (line 17) | def conv3x3x3(in_planes, out_planes, stride=1):
  function conv1x1x1 (line 26) | def conv1x1x1(in_planes, out_planes, stride=1):
  class BasicBlock (line 34) | class BasicBlock(nn.Module):
    method __init__ (line 37) | def __init__(self, in_planes, planes, stride=1, downsample=None, norm_...
    method forward (line 48) | def forward(self, x):
  class Bottleneck (line 67) | class Bottleneck(nn.Module):
    method __init__ (line 70) | def __init__(self, in_planes, planes, stride=1, downsample=None, norm_...
    method forward (line 83) | def forward(self, x):
  class CustomResNet3D (line 106) | class CustomResNet3D(BaseModule):
    method __init__ (line 107) | def __init__(self,
    method _downsample_basic_block (line 160) | def _downsample_basic_block(self, x, planes, stride):
    method _make_layer (line 171) | def _make_layer(self, block, planes, blocks, shortcut_type, stride=1, ...
    method forward (line 196) | def forward(self, x):
  function generate_model (line 207) | def generate_model(model_depth, **kwargs):

FILE: projects/occ_plugin/occupancy/dense_heads/flow_head.py
  class FlowHead (line 18) | class FlowHead(nn.Module):
    method __init__ (line 19) | def __init__(
    method forward_coarse_voxel (line 89) | def forward_coarse_voxel(self, voxel_feats):
    method forward (line 113) | def forward(self, voxel_feats, img_feats=None, transform=None, **kwargs):
    method loss_voxel (line 125) | def loss_voxel(self, output_voxels, target_voxels, tag):
    method loss_point (line 141) | def loss_point(self, fine_coord, fine_output, target_voxels, tag):
    method loss (line 158) | def loss(self, output_voxels=None,

FILE: projects/occ_plugin/occupancy/dense_heads/lovasz_softmax.py
  function lovasz_grad (line 21) | def lovasz_grad(gt_sorted):
  function iou_binary (line 36) | def iou_binary(preds, labels, EMPTY=1., ignore=None, per_image=True):
  function iou (line 56) | def iou(preds, labels, C, EMPTY=1., ignore=None, per_image=False):
  function lovasz_hinge (line 81) | def lovasz_hinge(logits, labels, per_image=True, ignore=None):
  function lovasz_hinge_flat (line 97) | def lovasz_hinge_flat(logits, labels):
  function flatten_binary_scores (line 117) | def flatten_binary_scores(scores, labels, ignore=None):
  class StableBCELoss (line 132) | class StableBCELoss(torch.nn.modules.Module):
    method __init__ (line 133) | def __init__(self):
    method forward (line 135) | def forward(self, input, target):
  function binary_xloss (line 141) | def binary_xloss(logits, labels, ignore=None):
  function lovasz_softmax (line 156) | def lovasz_softmax(probas, labels, classes='present', per_image=False, i...
  function lovasz_softmax_flat (line 174) | def lovasz_softmax_flat(probas, labels, classes='present'):
  function flatten_probas (line 205) | def flatten_probas(probas, labels, ignore=None):
  function xloss (line 234) | def xloss(logits, labels, ignore=None):
  function jaccard_loss (line 240) | def jaccard_loss(probas, labels,ignore=None, smooth = 100, bk_class = No...
  function hinge_jaccard_loss (line 268) | def hinge_jaccard_loss(probas, labels,ignore=None, classes = 'present', ...
  function isnan (line 305) | def isnan(x):
  function mean (line 309) | def mean(l, ignore_nan=False, empty=0):

FILE: projects/occ_plugin/occupancy/dense_heads/occ_head.py
  class OccHead (line 17) | class OccHead(nn.Module):
    method __init__ (line 18) | def __init__(
    method forward_coarse_voxel (line 112) | def forward_coarse_voxel(self, voxel_feats):
    method forward (line 137) | def forward(self, voxel_feats, img_feats=None, transform=None, **kwargs):
    method loss_voxel (line 149) | def loss_voxel(self, output_voxels, target_voxels, tag):
    method loss_point (line 174) | def loss_point(self, fine_coord, fine_output, target_voxels, tag):
    method loss (line 191) | def loss(self, output_voxels=None,

FILE: projects/occ_plugin/occupancy/dense_heads/utils.py
  function initial_voxelize (line 17) | def initial_voxelize(z: PointTensor, after_res) -> SparseTensor:
  function point_to_voxel (line 46) | def point_to_voxel(x: SparseTensor, z: PointTensor) -> SparseTensor:
  function voxel_to_point (line 71) | def voxel_to_point(x: SparseTensor, z: PointTensor, nearest=False) -> to...
  function range_to_point (line 107) | def range_to_point(x,px,py):
  function point_to_range (line 118) | def point_to_range(range_shape,pF,px,py):

FILE: projects/occ_plugin/occupancy/detectors/bevdepth.py
  class BEVDet (line 15) | class BEVDet(CenterPoint):
    method __init__ (line 16) | def __init__(self, img_view_transformer=None,
    method image_encoder (line 36) | def image_encoder(self, img):
    method bev_encoder (line 50) | def bev_encoder(self, x):
    method extract_img_feat (line 57) | def extract_img_feat(self, img, img_metas):
    method extract_feat (line 64) | def extract_feat(self, points, img, img_metas):
    method forward_train (line 70) | def forward_train(self,
    method forward_test (line 115) | def forward_test(self, points=None, img_metas=None, img_inputs=None, *...
    method aug_test (line 147) | def aug_test(self, points, img_metas, img=None, rescale=False):
    method simple_test (line 157) | def simple_test(self, points, img_metas, img=None, rescale=False):
    method forward_dummy (line 167) | def forward_dummy(self, points=None, img_metas=None, img_inputs=None, ...
  class BEVDet4D (line 181) | class BEVDet4D(BEVDet):
    method __init__ (line 182) | def __init__(self, pre_process=None,
    method shift_feature (line 195) | def shift_feature(self, input, trans, rots):
    method prepare_bev_feat (line 252) | def prepare_bev_feat(self, img, rot, tran, intrin, post_rot, post_tran...
    method extract_img_feat (line 260) | def extract_img_feat(self, img, img_metas):
  class BEVDepth_Base (line 299) | class BEVDepth_Base(object):
    method extract_feat (line 300) | def extract_feat(self, points, img, img_metas):
    method simple_test (line 306) | def simple_test(self, points, img_metas, img=None, rescale=False):
    method forward_train (line 315) | def forward_train(self,
  class BEVDepth (line 370) | class BEVDepth(BEVDepth_Base, BEVDet):
    method extract_img_feat (line 371) | def extract_img_feat(self, img, img_metas):
  class BEVDepth4D (line 388) | class BEVDepth4D(BEVDepth_Base, BEVDet4D):
    method prepare_bev_feat (line 389) | def prepare_bev_feat(self, img, rot, tran, intrin,
    method extract_img_feat (line 398) | def extract_img_feat(self, img, img_metas):
  class BEVStereo (line 445) | class BEVStereo(BEVDepth4D):
    method __init__ (line 446) | def __init__(self, bevdet_model=False, **kwargs):
    method image_encoder (line 450) | def image_encoder(self, img):
    method extract_img_feat (line 474) | def extract_img_feat(self, img, img_metas):

FILE: projects/occ_plugin/occupancy/detectors/ocfnet.py
  class OCFNet (line 23) | class OCFNet(BEVDepth):
    method __init__ (line 24) | def __init__(self,
    method image_encoder (line 103) | def image_encoder(self, img):
    method occ_encoder (line 123) | def occ_encoder(self, x):
    method flow_encoder (line 132) | def flow_encoder(self, x):
    method mat2pose_vec (line 140) | def mat2pose_vec(self, matrix: torch.Tensor):
    method pack_dbatch_and_dtime (line 166) | def pack_dbatch_and_dtime(self, x):
    method unpack_dbatch_and_dtime (line 172) | def unpack_dbatch_and_dtime(self, x, b, s):
    method extract_img_feat (line 177) | def extract_img_feat(self, img_inputs_seq, img_metas):
    method warp_features (line 232) | def warp_features(self, x, flow, tseq):
    method cumulative_warp_occ (line 327) | def cumulative_warp_occ(self, lifted_feature_seq, future_egomotion, mo...
    method extract_feat (line 342) | def extract_feat(self, img_inputs_seq, img_metas, future_egomotion):
    method forward_pts_train (line 395) | def forward_pts_train(
    method forward_flow_train (line 437) | def forward_flow_train(
    method forward_train (line 478) | def forward_train(self,
    method forward_test (line 566) | def forward_test(self,
    method simple_test (line 587) | def simple_test(self, img_metas, img_inputs_seq=None, rescale=False, p...
    method evaluate_occupancy_forecasting (line 662) | def evaluate_occupancy_forecasting(self, pred, gt, img_metas=None, sav...
    method find_instance_centers (line 714) | def find_instance_centers(self, center_prediction: torch.Tensor, conf_...
    method group_pixels (line 728) | def group_pixels(self, centers: torch.Tensor, offset_predictions: torc...
    method update_instance_ids (line 755) | def update_instance_ids(self, instance_seg, old_ids, new_ids):
    method make_instance_seg_consecutive (line 762) | def make_instance_seg_consecutive(self, instance_seg):
    method get_instance_segmentation_and_centers (line 769) | def get_instance_segmentation_and_centers(self,
    method flow_warp (line 797) | def flow_warp(self, occupancy, flow, mode='nearest', padding_mode='zer...
    method make_instance_id_temporally_consecutive (line 838) | def make_instance_id_temporally_consecutive(self, pred_inst, preds, ba...
    method predict_instance_segmentation (line 860) | def predict_instance_segmentation(self, pred_seg, pred_flow):
    method combine_mask (line 882) | def combine_mask(self, segmentation: torch.Tensor, instance: torch.Ten...
    method panoptic_metrics (line 915) | def panoptic_metrics(self, pred_segmentation, pred_instance, gt_segmen...
    method evaluate_instance_prediction (line 993) | def evaluate_instance_prediction(self, pred_seg, pred_flow, gt_seg, gt...
    method forward_dummy (line 1039) | def forward_dummy(self,
  function fast_hist (line 1061) | def fast_hist(pred, label, max_label=18):

FILE: projects/occ_plugin/occupancy/fuser/addfuse.py
  class AddFuser (line 11) | class AddFuser(nn.Module):
    method __init__ (line 12) | def __init__(self, in_channels, out_channels, dropout, input_modality=...
    method forward (line 40) | def forward(self, img_voxel_feats, pts_voxel_feats):

FILE: projects/occ_plugin/occupancy/fuser/convfuse.py
  class ConvFuser (line 11) | class ConvFuser(nn.Module):
    method __init__ (line 12) | def __init__(self, in_channels, out_channels) -> None:
    method forward (line 22) | def forward(self, img_voxel_feats, pts_voxel_feats):

FILE: projects/occ_plugin/occupancy/fuser/visfuse.py
  class VisFuser (line 13) | class VisFuser(nn.Module):
    method __init__ (line 14) | def __init__(self, in_channels, out_channels, norm_cfg=None) -> None:
    method forward (line 42) | def forward(self, img_voxel_feats, pts_voxel_feats):

FILE: projects/occ_plugin/occupancy/image2bev/ViewTransformerLSSBEVDepth.py
  function gen_dx_bx (line 20) | def gen_dx_bx(xbound, ybound, zbound):
  function cumsum_trick (line 26) | def cumsum_trick(x, geom_feats, ranks):
  class QuickCumsum (line 35) | class QuickCumsum(torch.autograd.Function):
    method forward (line 37) | def forward(ctx, x, geom_feats, ranks):
    method backward (line 54) | def backward(ctx, gradx, gradgeom):
  class ViewTransformerLiftSplatShoot (line 63) | class ViewTransformerLiftSplatShoot(BaseModule):
    method __init__ (line 64) | def __init__(self, grid_config=None, data_config=None,
    method get_depth_dist (line 100) | def get_depth_dist(self, x):
    method create_frustum (line 103) | def create_frustum(self):
    method get_geometry (line 116) | def get_geometry(self, rots_seq, trans_seq, intrins_seq, post_rots_seq...
    method voxel_pooling (line 145) | def voxel_pooling(self, geom_feats, x):
    method voxel_pooling_accelerated (line 190) | def voxel_pooling_accelerated(self, rots, trans, intrins, post_rots, p...
    method voxel_pooling_bevdepth (line 259) | def voxel_pooling_bevdepth(self, geom_feats, x):
    method forward (line 269) | def forward(self, input):
  class _ASPPModule (line 297) | class _ASPPModule(nn.Module):
    method __init__ (line 298) | def __init__(self, inplanes, planes, kernel_size, padding, dilation,
    method forward (line 313) | def forward(self, x):
    method _init_weight (line 319) | def _init_weight(self):
  class ASPP (line 328) | class ASPP(nn.Module):
    method __init__ (line 329) | def __init__(self, inplanes, mid_channels=256, norm_cfg=dict(type='BN2...
    method forward (line 374) | def forward(self, x):
    method _init_weight (line 392) | def _init_weight(self):
  class Mlp (line 401) | class Mlp(nn.Module):
    method __init__ (line 402) | def __init__(self,
    method forward (line 417) | def forward(self, x):
  class SELayer (line 426) | class SELayer(nn.Module):
    method __init__ (line 427) | def __init__(self, channels, act_layer=nn.ReLU, gate_layer=nn.Sigmoid):
    method forward (line 434) | def forward(self, x, x_se):
  class DepthNet (line 441) | class DepthNet(nn.Module):
    method __init__ (line 442) | def __init__(self, in_channels, mid_channels, context_channels,
    method forward (line 487) | def forward(self, x, mlp_input):
  class DepthAggregation (line 498) | class DepthAggregation(nn.Module):
    method __init__ (line 502) | def __init__(self, in_channels, mid_channels, out_channels, norm_cfg):
    method forward (line 547) | def forward(self, x):
  class ViewTransformerLSSBEVDepth (line 557) | class ViewTransformerLSSBEVDepth(ViewTransformerLiftSplatShoot):
    method __init__ (line 558) | def __init__(self, loss_depth_weight, cam_channels=27, loss_depth_reg_...
    method _forward_voxel_net (line 573) | def _forward_voxel_net(self, img_feat_with_depth):
    method get_mlp_input (line 586) | def get_mlp_input(self, rot, tran, intrin, post_rot, post_tran, bda=No...
    method get_downsampled_gt_depth (line 638) | def get_downsampled_gt_depth(self, gt_depths):
    method _prepare_depth_gt (line 662) | def _prepare_depth_gt(self, gt_depths):
    method get_depth_reg_loss (line 680) | def get_depth_reg_loss(self, depth_labels, depth_preds):
    method get_depth_loss (line 700) | def get_depth_loss(self, depth_labels, depth_preds):
    method forward (line 717) | def forward(self, input):
  class ConvBnReLU3D (line 747) | class ConvBnReLU3D(nn.Module):
    method __init__ (line 749) | def __init__(
    method forward (line 777) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class DepthNetStereo (line 782) | class DepthNetStereo(nn.Module):
    method __init__ (line 783) | def __init__(self,
    method forward (line 861) | def forward(self, x, mlp_input):
  class ViewTransformerLSSBEVStereo (line 884) | class ViewTransformerLSSBEVStereo(ViewTransformerLSSBEVDepth):
    method __init__ (line 885) | def __init__(self, num_ranges=4, use_mask=True, em_iteration=3,
    method depth_sampling (line 959) | def depth_sampling(self):
    method create_depth_sample_frustum (line 973) | def create_depth_sample_frustum(self, depth_sample, downsample_factor=...
    method homo_warping (line 1001) | def homo_warping(
    method _forward_mask (line 1081) | def _forward_mask(
    method _generate_cost_volume (line 1140) | def _generate_cost_volume(
    method _forward_stereo (line 1206) | def _forward_stereo(
    method forward (line 1368) | def forward(self, input):

FILE: projects/occ_plugin/occupancy/image2bev/ViewTransformerLSSVoxel.py
  function get_gpu_memory_usage (line 22) | def get_gpu_memory_usage():
  class ViewTransformerLiftSplatShootVoxel (line 28) | class ViewTransformerLiftSplatShootVoxel(ViewTransformerLSSBEVDepth):
    method __init__ (line 29) | def __init__(self, loss_depth_weight, loss_depth_type='bce', **kwargs):
    method get_downsampled_gt_depth (line 36) | def get_downsampled_gt_depth(self, gt_depths):
    method get_bce_depth_loss (line 63) | def get_bce_depth_loss(self, depth_labels, depth_preds):
    method get_klv_depth_loss (line 76) | def get_klv_depth_loss(self, depth_labels, depth_preds):
    method get_depth_loss (line 91) | def get_depth_loss(self, depth_labels, depth_preds):
    method voxel_pooling (line 103) | def voxel_pooling(self, geom_feats, x):
    method forward (line 129) | def forward(self, input):

FILE: projects/occ_plugin/occupancy/necks/fpn3d.py
  class FPN3D (line 14) | class FPN3D(BaseModule):
    method __init__ (line 27) | def __init__(self,
    method forward (line 69) | def forward(self, inputs):

FILE: projects/occ_plugin/occupancy/necks/second_fpn_3d.py
  class SECONDFPN3D (line 13) | class SECONDFPN3D(BaseModule):
    method __init__ (line 26) | def __init__(self,
    method forward (line 78) | def forward(self, x):

FILE: projects/occ_plugin/occupancy/voxel_encoder/sparse_lidar_enc.py
  function post_act_block (line 17) | def post_act_block(in_channels, out_channels, kernel_size, indice_key=No...
  class SparseBasicBlock (line 40) | class SparseBasicBlock(spconv.SparseModule):
    method __init__ (line 42) | def __init__(self, inplanes, planes, stride=1, norm_cfg=None, indice_k...
    method forward (line 55) | def forward(self, x):
  class SparseLiDAREnc4x (line 66) | class SparseLiDAREnc4x(nn.Module):
    method __init__ (line 67) | def __init__(self, input_channel, norm_cfg, base_channel, out_channel,
    method forward (line 103) | def forward(self, voxel_features, coors, batch_size):
  class SparseLiDAREnc8x (line 124) | class SparseLiDAREnc8x(nn.Module):
    method __init__ (line 125) | def __init__(self, input_channel, norm_cfg, base_channel, out_channel,
    method forward (line 162) | def forward(self, voxel_features, coors, batch_size):

FILE: projects/occ_plugin/ops/occ_pooling/OCC_Pool.py
  class QuickCumsum (line 8) | class QuickCumsum(torch.autograd.Function):
    method forward (line 10) | def forward(ctx, x, geom_feats, ranks):
    method backward (line 27) | def backward(ctx, gradx, gradgeom):
  class QuickCumsumCuda (line 37) | class QuickCumsumCuda(torch.autograd.Function):
    method forward (line 39) | def forward(ctx, x, geom_feats, ranks, B, D, H, W):
    method backward (line 64) | def backward(ctx, out_grad):
  function occ_pool (line 83) | def occ_pool(feats, coords, B, D, H, W):

FILE: projects/occ_plugin/ops/occ_pooling/src/occ_pool.cpp
  function occ_pool_forward (line 22) | at::Tensor occ_pool_forward(
  function occ_pool_backward (line 60) | at::Tensor occ_pool_backward(
  function PYBIND11_MODULE (line 89) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: projects/occ_plugin/utils/coordinate_transform.py
  function coarse_to_fine_coordinates (line 4) | def coarse_to_fine_coordinates(coarse_cor, ratio, topk=30000):
  function project_points_on_img (line 26) | def project_points_on_img(points, rots, trans, intrins, post_rots, post_...

FILE: projects/occ_plugin/utils/formating.py
  function cm_to_ious (line 4) | def cm_to_ious(cm):
  function format_results (line 17) | def format_results(mean_ious, return_dic=False):
  function format_iou_results (line 55) | def format_iou_results(mean_ious, return_dic=False):
  function format_vel_results (line 92) | def format_vel_results(mean_epe, return_dic=False):

FILE: projects/occ_plugin/utils/gaussian.py
  function gaussian_2d (line 7) | def gaussian_2d(shape, sigma=1):
  function draw_heatmap_gaussian (line 26) | def draw_heatmap_gaussian(heatmap, center, radius, k=1):
  function gaussian_radius (line 58) | def gaussian_radius(det_size, min_overlap=0.5):
  function generate_guassian_depth_target (line 90) | def generate_guassian_depth_target(depth, stride, cam_depth_range, const...

FILE: projects/occ_plugin/utils/geometry.py
  function convert_egopose_to_matrix_numpy (line 8) | def convert_egopose_to_matrix_numpy(trans, rot):
  function invert_matrix_egopose_numpy (line 18) | def invert_matrix_egopose_numpy(egopose):

FILE: projects/occ_plugin/utils/metric_util.py
  function fast_hist (line 7) | def fast_hist(pred, label, n):
  function per_class_iu (line 13) | def per_class_iu(hist):
  function fast_hist_crop (line 16) | def fast_hist_crop(output, target, unique_label):
  class SSCMetrics (line 24) | class SSCMetrics:
    method __init__ (line 25) | def __init__(self, class_names, ignore_idx=255, empty_idx=None):
    method hist_info (line 32) | def hist_info(self, n_cl, pred, gt):
    method compute_score (line 47) | def compute_score(hist, correct, labeled):
    method add_batch (line 57) | def add_batch(self, y_pred, y_true, nonsurface=None):
    method get_stats (line 80) | def get_stats(self):
    method reset (line 98) | def reset(self):
    method get_score_completion (line 118) | def get_score_completion(self, predict, target, nonempty=None):
    method get_score_semantic_and_completion (line 153) | def get_score_semantic_and_completion(self, predict, target, nonempty=...

FILE: projects/occ_plugin/utils/nusc_param.py
  function KL_sep (line 53) | def KL_sep(p, target):
  function geo_scal_loss (line 63) | def geo_scal_loss(pred, ssc_target):
  function sem_scal_loss (line 90) | def sem_scal_loss(pred, ssc_target):
  function CE_ssc_loss (line 137) | def CE_ssc_loss(pred, target, class_weights):

FILE: projects/occ_plugin/utils/semkitti.py
  function KL_sep (line 55) | def KL_sep(p, target):
  function geo_scal_loss (line 65) | def geo_scal_loss(pred, ssc_target, ignore_index=255, non_empty_idx=0):
  function sem_scal_loss (line 93) | def sem_scal_loss(pred, ssc_target, ignore_index=255):
  function CE_ssc_loss (line 140) | def CE_ssc_loss(pred, target, class_weights=None, ignore_index=255):
  function Smooth_L1_loss (line 151) | def Smooth_L1_loss(pred, target, ignore_index=255):
  function vel_loss (line 165) | def vel_loss(pred, gt):

FILE: projects/occ_plugin/utils/voxel_to_points.py
  function query_points_from_voxels (line 4) | def query_points_from_voxels(pred, gt, img_metas):

FILE: setup.py
  function make_cuda_ext (line 10) | def make_cuda_ext(name,

FILE: tools/gen_data/gen_depth_gt.py
  function map_pointcloud_to_image (line 12) | def map_pointcloud_to_image(
  function worker (line 90) | def worker(info):

FILE: tools/misc/browse_dataset.py
  function parse_args (line 15) | def parse_args():
  function build_data_cfg (line 53) | def build_data_cfg(config_path, skip_type, cfg_options):
  function to_depth_mode (line 79) | def to_depth_mode(points, bboxes):
  function show_det_data (line 90) | def show_det_data(idx, dataset, out_dir, filename, show=False):
  function show_seg_data (line 107) | def show_seg_data(idx, dataset, out_dir, filename, show=False):
  function show_proj_bbox_img (line 124) | def show_proj_bbox_img(idx,
  function main (line 184) | def main():

FILE: tools/misc/fuse_conv_bn.py
  function fuse_conv_bn (line 10) | def fuse_conv_bn(conv, bn):
  function fuse_module (line 26) | def fuse_module(m):
  function parse_args (line 47) | def parse_args():
  function main (line 57) | def main():

FILE: tools/misc/print_config.py
  function parse_args (line 6) | def parse_args():
  function main (line 16) | def main():

FILE: tools/misc/visualize_results.py
  function parse_args (line 9) | def parse_args():
  function main (line 21) | def main():

FILE: tools/test.py
  function parse_args (line 29) | def parse_args():
  function main (line 111) | def main():

FILE: tools/train.py
  function parse_args (line 34) | def parse_args():
  function main (line 95) | def main():

FILE: viz/viz_gt.py
  function viz_occ (line 16) | def viz_occ(occ, occ_mo, file_name, voxel_size, show_occ, show_time_chan...
  function main (line 91) | def main():

FILE: viz/viz_pred.py
  function viz_occ (line 11) | def viz_occ(occ, occ_mo, file_name, voxel_size, show_occ, show_time_chan...
  function main (line 86) | def main():