SYMBOL INDEX (353 symbols across 38 files)

FILE: gen_instance_info.py
  function convert_to_nusc_box (line 39) | def convert_to_nusc_box(bboxes, lift_center=False, wlh_margin=0.0):
  function meshgrid3d (line 63) | def meshgrid3d(occ_size, pc_range):  # points in ego coord
  function process_add_instance_info (line 77) | def process_add_instance_info(sample):
  function add_instance_info (line 200) | def add_instance_info(sample_infos):

FILE: gen_sweep_info.py
  function get_cam_info (line 18) | def get_cam_info(nusc, sample_data):
  function add_sweep_info (line 40) | def add_sweep_info(nusc, sample_infos):

FILE: lib/dvr/dvr.cpp
  function render_forward (line 39) | std::vector<torch::Tensor>
  function render (line 51) | std::vector<torch::Tensor> render(torch::Tensor sigma, torch::Tensor ori...
  function init (line 61) | torch::Tensor init(torch::Tensor points, torch::Tensor tindex,
  function PYBIND11_MODULE (line 68) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: loaders/builder.py
  function build_dataloader (line 9) | def build_dataloader(dataset,

FILE: loaders/ego_pose_dataset.py
  function trans_matrix (line 8) | def trans_matrix(T, R):
  class EgoPoseDataset (line 16) | class EgoPoseDataset(Dataset):
    method __init__ (line 17) | def __init__(self, data_infos):
    method __len__ (line 29) | def __len__(self):
    method get_ego_from_lidar (line 32) | def get_ego_from_lidar(self, info):
    method get_global_pose (line 38) | def get_global_pose(self, info, inverse=False):
    method __getitem__ (line 50) | def __getitem__(self, idx):

FILE: loaders/nuscenes_dataset.py
  class CustomNuScenesDataset (line 9) | class CustomNuScenesDataset(NuScenesDataset):
    method collect_sweeps (line 11) | def collect_sweeps(self, index, into_past=60, into_future=0):
    method get_data_info (line 34) | def get_data_info(self, index):

FILE: loaders/nuscenes_occ_dataset.py
  class NuSceneOcc (line 19) | class NuSceneOcc(NuScenesDataset):
    method __init__ (line 20) | def __init__(self, occ_gt_root, *args, **kwargs):
    method collect_sweeps (line 35) | def collect_sweeps(self, index, into_past=150, into_future=0):
    method get_data_info (line 58) | def get_data_info(self, index):
    method evaluate (line 118) | def evaluate(self, occ_results, runner=None, show_dir=None, **eval_kwa...
    method format_results (line 177) | def format_results(self, occ_results, submission_prefix, **kwargs):

FILE: loaders/old_metrics.py
  function pcolor (line 12) | def pcolor(string, color, on_color=None, attrs=None):
  function getCellCoordinates (line 35) | def getCellCoordinates(points, voxelSize):
  function getNumUniqueCells (line 39) | def getNumUniqueCells(cells):
  class Metric_mIoU (line 44) | class Metric_mIoU():
    method __init__ (line 45) | def __init__(self,
    method hist_info (line 76) | def hist_info(self, n_cl, pred, gt):
    method per_class_iu (line 104) | def per_class_iu(self, hist):
    method compute_mIoU (line 110) | def compute_mIoU(self, pred, label, n_classes):
    method add_batch (line 120) | def add_batch(self,semantics_pred,semantics_gt,mask_lidar,mask_camera):
    method count_miou (line 143) | def count_miou(self):
  class Metric_FScore (line 156) | class Metric_FScore():
    method __init__ (line 157) | def __init__(self,
    method voxel2points (line 181) | def voxel2points(self, voxel):
    method add_batch (line 193) | def add_batch(self,semantics_pred,semantics_gt,mask_lidar,mask_camera ):
    method count_fscore (line 237) | def count_fscore(self,):
  class Metric_mRecall (line 242) | class Metric_mRecall():
    method __init__ (line 243) | def __init__(self,
    method hist_info (line 276) | def hist_info(self, n_cl, p_cl, pred, gt):
    method per_class_recall (line 304) | def per_class_recall(self, hist):
    method compute_mRecall (line 307) | def compute_mRecall(self, pred, label, n_classes, p_classes):
    method add_batch (line 317) | def add_batch(self,semantics_pred,semantics_gt,mask_lidar,mask_camera):
    method count_mrecall (line 338) | def count_mrecall(self):
  class Metric_Panoptic (line 351) | class Metric_Panoptic():
    method __init__ (line 352) | def __init__(self,
    method add_batch (line 393) | def add_batch(self,semantics_pred,semantics_gt,instances_pred,instance...
    method add_panoptic_sample (line 412) | def add_panoptic_sample(self, semantics_pred, semantics_gt, instances_...
    method count_pq (line 531) | def count_pq(self, ):

FILE: loaders/pipelines/loading.py
  function compose_lidar2img (line 13) | def compose_lidar2img(ego2global_translation_curr,
  class LoadMultiViewImageFromMultiSweeps (line 40) | class LoadMultiViewImageFromMultiSweeps(object):
    method __init__ (line 41) | def __init__(self,
    method load_offline (line 57) | def load_offline(self, results):
    method load_online (line 111) | def load_online(self, results):
    method __call__ (line 158) | def __call__(self, results):
  class LoadOccGTFromFile (line 170) | class LoadOccGTFromFile(object):
    method __init__ (line 171) | def __init__(self, num_classes=18, inst_class_ids=[]):
    method __call__ (line 175) | def __call__(self, results):
  class BEVAug (line 246) | class BEVAug(object):
    method __init__ (line 247) | def __init__(self, bda_aug_conf, classes, is_train=True):
    method sample_bda_augmentation (line 252) | def sample_bda_augmentation(self):
    method bev_transform (line 266) | def bev_transform(self, rotate_angle, scale_ratio, flip_dx, flip_dy):
    method __call__ (line 290) | def __call__(self, results):

FILE: loaders/pipelines/transforms.py
  class PadMultiViewImage (line 10) | class PadMultiViewImage(object):
    method __init__ (line 21) | def __init__(self, size=None, size_divisor=None, pad_val=0):
    method _pad_img (line 29) | def _pad_img(self, img):
    method _pad_imgs (line 40) | def _pad_imgs(self, results):
    method __call__ (line 50) | def __call__(self, results):
    method __repr__ (line 60) | def __repr__(self):
  class NormalizeMultiviewImage (line 69) | class NormalizeMultiviewImage(object):
    method __init__ (line 79) | def __init__(self, mean, std, to_rgb=True):
    method __call__ (line 84) | def __call__(self, results):
    method __repr__ (line 110) | def __repr__(self):
  class PhotoMetricDistortionMultiViewImage (line 117) | class PhotoMetricDistortionMultiViewImage:
    method __init__ (line 136) | def __init__(self,
    method __call__ (line 146) | def __call__(self, results):
    method __repr__ (line 207) | def __repr__(self):
  class RandomTransformImage (line 219) | class RandomTransformImage(object):
    method __init__ (line 220) | def __init__(self, ida_aug_conf=None, training=True):
    method __call__ (line 224) | def __call__(self, results):
    method img_transform (line 270) | def img_transform(self, img, resize, resize_dims, crop, flip, rotate):
    method sample_augmentation (line 313) | def sample_augmentation(self):
  class GlobalRotScaleTransImage (line 345) | class GlobalRotScaleTransImage(object):
    method __init__ (line 346) | def __init__(self,
    method __call__ (line 354) | def __call__(self, results):
    method rotate_z (line 369) | def rotate_z(self, results, rot_angle):
    method scale_xyz (line 384) | def scale_xyz(self, results, scale_ratio):

FILE: loaders/ray_metrics.py
  function get_rendered_pcds (line 19) | def get_rendered_pcds(origin, points, tindex, pred_dist):
  function meshgrid3d (line 37) | def meshgrid3d(occ_size, pc_range):
  function generate_lidar_rays (line 51) | def generate_lidar_rays():
  function process_one_sample (line 77) | def process_one_sample(sem_pred, lidar_rays, output_origin, instance_pre...
  function calc_rayiou (line 138) | def calc_rayiou(pcd_pred_list, pcd_gt_list, occ_class_names):
  function main_rayiou (line 175) | def main_rayiou(sem_pred_list, sem_gt_list, lidar_origin_list, occ_class...
  function main_raypq (line 231) | def main_raypq(sem_pred_list, sem_gt_list, inst_pred_list, inst_gt_list,...

FILE: loaders/ray_pq.py
  class Metric_RayPQ (line 5) | class Metric_RayPQ:
    method __init__ (line 6) | def __init__(self,
    method add_batch (line 36) | def add_batch(self,semantics_pred,semantics_gt,instances_pred,instance...
    method add_panoptic_sample (line 40) | def add_panoptic_sample(self, semantics_pred, semantics_gt, instances_...
    method count_pq (line 155) | def count_pq(self):

FILE: models/backbones/vovnet.py
  function dw_conv3x3 (line 93) | def dw_conv3x3(in_channels, out_channels, module_name, postfix, stride=1...
  function conv3x3 (line 117) | def conv3x3(in_channels, out_channels, module_name, postfix, stride=1, g...
  function conv1x1 (line 137) | def conv1x1(in_channels, out_channels, module_name, postfix, stride=1, g...
  class Hsigmoid (line 157) | class Hsigmoid(nn.Module):
    method __init__ (line 158) | def __init__(self, inplace=True):
    method forward (line 162) | def forward(self, x):
  class eSEModule (line 166) | class eSEModule(nn.Module):
    method __init__ (line 167) | def __init__(self, channel, reduction=4):
    method forward (line 173) | def forward(self, x):
  class _OSA_module (line 181) | class _OSA_module(nn.Module):
    method __init__ (line 182) | def __init__(self, in_ch, stage_ch, concat_ch, layer_per_block, module...
    method _forward (line 211) | def _forward(self, x):
    method forward (line 234) | def forward(self, x):
  class _OSA_stage (line 241) | class _OSA_stage(nn.Sequential):
    method __init__ (line 242) | def __init__(self, in_ch, stage_ch, concat_ch, block_per_stage, layer_...
  class VoVNet (line 276) | class VoVNet(BaseModule):
    method __init__ (line 277) | def __init__(self, spec_name, input_ch=3, out_features=None, frozen_st...
    method _initialize_weights (line 344) | def _initialize_weights(self):
    method forward (line 349) | def forward(self, x):
    method _freeze_stages (line 361) | def _freeze_stages(self):
    method train (line 374) | def train(self, mode=True):

FILE: models/bbox/assigners/hungarian_assigner_3d.py
  class HungarianAssigner3D (line 16) | class HungarianAssigner3D(BaseAssigner):
    method __init__ (line 17) | def __init__(self,
    method assign (line 27) | def assign(self,

FILE: models/bbox/coders/nms_free_coder.py
  class NMSFreeCoder (line 9) | class NMSFreeCoder(BaseBBoxCoder):
    method __init__ (line 20) | def __init__(self,
    method encode (line 34) | def encode(self):
    method decode_single (line 37) | def decode_single(self, cls_scores, bbox_preds):
    method decode (line 90) | def decode(self, preds_dicts):

FILE: models/bbox/match_costs/match_cost.py
  class BBox3DL1Cost (line 6) | class BBox3DL1Cost(object):
    method __init__ (line 12) | def __init__(self, weight=1.0):
    method __call__ (line 15) | def __call__(self, bbox_pred, gt_bboxes):
  class BBoxBEVL1Cost (line 31) | class BBoxBEVL1Cost(object):
    method __init__ (line 32) | def __init__(self, weight, pc_range):
    method __call__ (line 36) | def __call__(self, bboxes, gt_bboxes):
  class IoU3DCost (line 47) | class IoU3DCost(object):
    method __init__ (line 48) | def __init__(self, weight):
    method __call__ (line 51) | def __call__(self, iou):

FILE: models/bbox/utils.py
  function normalize_bbox (line 4) | def normalize_bbox(bboxes):
  function denormalize_bbox (line 23) | def denormalize_bbox(normalized_bboxes):
  function encode_bbox (line 46) | def encode_bbox(bboxes, pc_range=None):
  function decode_bbox (line 63) | def decode_bbox(bboxes, pc_range=None):
  function bbox2occrange (line 79) | def bbox2occrange(bboxes, occ_size, query_cube_size=None):
  function occrange2bbox (line 107) | def occrange2bbox(occ_range, occ_size, pc_range):

FILE: models/checkpoint.py
  function detach_variable (line 15) | def detach_variable(inputs: Tuple[Any, ...]) -> Tuple[torch.Tensor, ...]:
  function check_backward_validity (line 32) | def check_backward_validity(inputs: Iterable[Any]) -> None:
  function get_device_states (line 44) | def get_device_states(*args) -> Tuple[List[int], List[torch.Tensor]]:
  function set_device_states (line 58) | def set_device_states(devices, states) -> None:
  function _get_autocast_kwargs (line 63) | def _get_autocast_kwargs():
  class CheckpointFunction (line 74) | class CheckpointFunction(torch.autograd.Function):
    method forward (line 77) | def forward(ctx, run_function, preserve_rng_state, *args):
    method backward (line 114) | def backward(ctx, *args):
  function checkpoint (line 167) | def checkpoint(function, *args, use_reentrant: bool = True, **kwargs):
  function checkpoint_sequential (line 262) | def checkpoint_sequential(functions, segments, input, use_reentrant=True...
  function _checkpoint_without_reentrant (line 342) | def _checkpoint_without_reentrant(function, preserve_rng_state=True, *ar...

FILE: models/csrc/msmv_sampling/msmv_sampling.cpp
  function ms_deform_attn_cuda_c2345_forward (line 98) | at::Tensor ms_deform_attn_cuda_c2345_forward(
  function ms_deform_attn_cuda_c23456_forward (line 152) | at::Tensor ms_deform_attn_cuda_c23456_forward(
  function ms_deform_attn_cuda_c2345_backward (line 212) | std::vector<at::Tensor> ms_deform_attn_cuda_c2345_backward(
  function ms_deform_attn_cuda_c23456_backward (line 283) | std::vector<at::Tensor> ms_deform_attn_cuda_c23456_backward(
  function PYBIND11_MODULE (line 363) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: models/csrc/setup.py
  function get_ext_modules (line 5) | def get_ext_modules():

FILE: models/csrc/wrapper.py
  function msmv_sampling_pytorch (line 7) | def msmv_sampling_pytorch(mlvl_feats, sampling_locations, scale_weights):
  class MSMVSamplingC2345 (line 35) | class MSMVSamplingC2345(torch.autograd.Function):
    method forward (line 37) | def forward(ctx, feat_c2, feat_c3, feat_c4, feat_c5, sampling_location...
    method backward (line 46) | def backward(ctx, grad_output):
  class MSMVSamplingC23456 (line 58) | class MSMVSamplingC23456(torch.autograd.Function):
    method forward (line 60) | def forward(ctx, feat_c2, feat_c3, feat_c4, feat_c5, feat_c6, sampling...
    method backward (line 69) | def backward(ctx, grad_output):
  function msmv_sampling (line 81) | def msmv_sampling(mlvl_feats, sampling_locations, scale_weights):

FILE: models/loss_utils.py
  function get_voxel_decoder_loss_input (line 11) | def get_voxel_decoder_loss_input(voxel_semantics, occ_loc_i, seg_pred_i,...
  function compute_scal_loss (line 34) | def compute_scal_loss(pred, gt, class_id, reverse=False, ignore_index=255):
  class GeoScalLoss (line 85) | class GeoScalLoss(nn.Module):
    method __init__ (line 86) | def __init__(self,
    method forward (line 93) | def forward(self, pred, gt):
  class SemScalLoss (line 102) | class SemScalLoss(nn.Module):
    method __init__ (line 103) | def __init__(self,
    method forward (line 116) | def forward(self, pred, gt):
  function dice_loss (line 130) | def dice_loss(
  function sigmoid_ce_loss (line 164) | def sigmoid_ce_loss(
  function CE_ssc_loss (line 196) | def CE_ssc_loss(pred, target, class_weights=None, ignore_index=255):
  function lovasz_grad (line 211) | def lovasz_grad(gt_sorted):
  function lovasz_softmax (line 227) | def lovasz_softmax(probas, labels, classes='present', per_image=False, i...
  function lovasz_softmax_flat (line 247) | def lovasz_softmax_flat(probas, labels, classes='present'):
  function flatten_probas (line 279) | def flatten_probas(probas, labels, ignore=None):
  class Mask2FormerLoss (line 311) | class Mask2FormerLoss(nn.Module):
    method __init__ (line 312) | def __init__(self,
    method forward (line 334) | def forward(self, mask_pred, class_pred, mask_gt, class_gt, indices, m...
    method loss_masks (line 376) | def loss_masks(self, src_mask, tgt_mask, avg_factor=None, mask_camera=...
    method loss_labels (line 394) | def loss_labels(self, src_class, tgt_class, empty_weight=None, avg_fac...
  function mean (line 403) | def mean(l, empty=0):

FILE: models/matcher.py
  function batch_dice_loss (line 12) | def batch_dice_loss(inputs: torch.Tensor, targets: torch.Tensor, mask_ca...
  function batch_sigmoid_ce_loss (line 39) | def batch_sigmoid_ce_loss(inputs: torch.Tensor, targets: torch.Tensor, m...
  class HungarianMatcher (line 84) | class HungarianMatcher(BaseModule):
    method __init__ (line 92) | def __init__(self, cost_class: float = 1, cost_mask: float = 1, cost_d...
    method forward (line 110) | def forward(self, mask_pred, class_pred, mask_gt, class_gt, mask_camera):

FILE: models/sparse_voxel_decoder.py
  function index2point (line 11) | def index2point(coords, pc_range, voxel_size):
  function point2bbox (line 22) | def point2bbox(coords, box_size):
  function upsample (line 32) | def upsample(pre_feat, pre_coords, interval):
  class SparseVoxelDecoder (line 55) | class SparseVoxelDecoder(BaseModule):
    method __init__ (line 56) | def __init__(self,
    method init_weights (line 106) | def init_weights(self):
    method forward (line 110) | def forward(self, mlvl_feats, img_metas):
  class SparseVoxelDecoderLayer (line 165) | class SparseVoxelDecoderLayer(BaseModule):
    method __init__ (line 166) | def __init__(self,
    method init_weights (line 211) | def init_weights(self):
    method forward (line 218) | def forward(self, query_feat, query_bbox, mlvl_feats, img_metas):

FILE: models/sparsebev_head.py
  class SparseBEVHead (line 14) | class SparseBEVHead(DETRHead):
    method __init__ (line 15) | def __init__(self,
    method _init_layers (line 48) | def _init_layers(self):
    method init_weights (line 65) | def init_weights(self):
    method forward (line 68) | def forward(self, mlvl_feats, img_metas):
    method prepare_for_dn_input (line 117) | def prepare_for_dn_input(self, batch_size, init_query_bbox, label_enc,...
    method prepare_for_dn_loss (line 218) | def prepare_for_dn_loss(self, mask_dict):
    method dn_loss_single (line 233) | def dn_loss_single(self,
    method calc_dn_loss (line 272) | def calc_dn_loss(self, loss_dict, preds_dicts, num_dec_layers):
    method _get_target_single (line 295) | def _get_target_single(self,
    method get_targets (line 323) | def get_targets(self,
    method loss_single (line 343) | def loss_single(self,
    method loss (line 399) | def loss(self,
    method get_bboxes (line 457) | def get_bboxes(self, preds_dicts, img_metas, rescale=False):

FILE: models/sparsebev_sampling.py
  function make_sample_points_from_bbox (line 7) | def make_sample_points_from_bbox(query_bbox, offset, pc_range):
  function make_sample_points_from_mask (line 32) | def make_sample_points_from_mask(valid_map, pc_range, occ_size, num_poin...
  function sampling_4d (line 83) | def sampling_4d(sample_points, mlvl_feats, scale_weights, lidar2img, ima...

FILE: models/sparsebev_transformer.py
  class SparseBEVTransformer (line 16) | class SparseBEVTransformer(BaseModule):
    method __init__ (line 17) | def __init__(self, embed_dims, num_frames=8, num_points=4, num_layers=...
    method init_weights (line 28) | def init_weights(self):
    method forward (line 31) | def forward(self, query_bbox, query_feat, mlvl_feats, attn_mask, img_m...
  class SparseBEVTransformerDecoder (line 40) | class SparseBEVTransformerDecoder(BaseModule):
    method __init__ (line 41) | def __init__(self, embed_dims, num_frames=8, num_points=4, num_layers=...
    method init_weights (line 51) | def init_weights(self):
    method forward (line 54) | def forward(self, query_bbox, query_feat, mlvl_feats, attn_mask, img_m...
  class SparseBEVTransformerDecoderLayer (line 93) | class SparseBEVTransformerDecoderLayer(BaseModule):
    method __init__ (line 94) | def __init__(self, embed_dims, num_frames=8, num_points=4, num_levels=...
    method init_weights (line 136) | def init_weights(self):
    method refine_bbox (line 144) | def refine_bbox(self, bbox_proposal, bbox_delta):
    method forward (line 151) | def forward(self, query_bbox, query_feat, mlvl_feats, attn_mask, img_m...
  class SparseBEVSelfAttention (line 184) | class SparseBEVSelfAttention(BaseModule):
    method __init__ (line 185) | def __init__(self, embed_dims=256, num_heads=8, dropout=0.1, pc_range=...
    method init_weights (line 197) | def init_weights(self):
    method inner_forward (line 202) | def inner_forward(self, query_bbox, query_feat, pre_attn_mask=None):
    method forward (line 224) | def forward(self, query_bbox, query_feat, pre_attn_mask=None):
    method calc_bbox_dists (line 231) | def calc_bbox_dists(self, bboxes):
  class SparseBEVSampling (line 245) | class SparseBEVSampling(BaseModule):
    method __init__ (line 246) | def __init__(self, embed_dims=256, num_frames=4, num_groups=4, num_poi...
    method init_weights (line 258) | def init_weights(self):
    method inner_forward (line 263) | def inner_forward(self, query_bbox, query_feat, mlvl_feats, img_metas):
    method forward (line 307) | def forward(self, query_bbox, query_feat, mlvl_feats, img_metas):
  class AdaptiveMixing (line 314) | class AdaptiveMixing(nn.Module):
    method __init__ (line 315) | def __init__(self, in_dim, in_points, n_groups=1, query_dim=None, out_...
    method init_weights (line 341) | def init_weights(self):
    method inner_forward (line 344) | def inner_forward(self, x, query):
    method forward (line 376) | def forward(self, x, query):
  class AdaptiveMixingPointOnly (line 383) | class AdaptiveMixingPointOnly(nn.Module):
    method __init__ (line 384) | def __init__(self, in_dim, in_points, n_groups=1, query_dim=None, out_...
    method init_weights (line 409) | def init_weights(self):
    method inner_forward (line 412) | def inner_forward(self, x, query):
    method forward (line 437) | def forward(self, x, query):
  class DeformAggregation (line 444) | class DeformAggregation(nn.Module):
    method __init__ (line 445) | def __init__(self, in_dim, in_points, n_groups=1, query_dim=None, out_...
    method init_weights (line 467) | def init_weights(self):
    method inner_forward (line 470) | def inner_forward(self, x, query):
    method forward (line 488) | def forward(self, x, query):

FILE: models/sparseocc.py
  class SparseOcc (line 13) | class SparseOcc(MVXTwoStageDetector):
    method __init__ (line 14) | def __init__(self,
    method extract_img_feat (line 48) | def extract_img_feat(self, img):
    method extract_feat (line 60) | def extract_feat(self, img, img_metas=None):
    method forward_pts_train (line 109) | def forward_pts_train(self, mlvl_feats, voxel_semantics, voxel_instanc...
    method forward (line 119) | def forward(self, return_loss=True, **kwargs):
    method forward_train (line 126) | def forward_train(self, img_metas=None, img=None, voxel_semantics=None...
    method forward_test (line 130) | def forward_test(self, img_metas, img=None, **kwargs):
    method simple_test_pts (line 155) | def simple_test_pts(self, x, img_metas, rescale=False):
    method simple_test (line 160) | def simple_test(self, img_metas, img=None, rescale=False):
    method simple_test_offline (line 167) | def simple_test_offline(self, img_metas, img=None, rescale=False):
    method simple_test_online (line 171) | def simple_test_online(self, img_metas, img=None, rescale=False):

FILE: models/sparseocc_head.py
  class SparseOccHead (line 19) | class SparseOccHead(nn.Module):
    method __init__ (line 20) | def __init__(self,
    method init_weights (line 45) | def init_weights(self):
    method forward (line 49) | def forward(self, mlvl_feats, img_metas):
    method loss (line 59) | def loss(self, voxel_semantics, voxel_instances, instance_class_ids, p...
    method loss_single (line 62) | def loss_single(self, voxel_semantics, voxel_instances, instance_class...
    method merge_occ_pred (line 131) | def merge_occ_pred(self, outs):
    method merge_semseg (line 148) | def merge_semseg(self, mask_cls, mask_pred):
    method merge_panoseg (line 160) | def merge_panoseg(self, mask_cls, mask_pred):
    method merge_panoseg_single (line 176) | def merge_panoseg_single(self, mask_cls, mask_pred):

FILE: models/sparseocc_transformer.py
  class SparseOccTransformer (line 16) | class SparseOccTransformer(BaseModule):
    method __init__ (line 17) | def __init__(self,
    method init_weights (line 60) | def init_weights(self):
    method forward (line 64) | def forward(self, mlvl_feats, img_metas):
  class MaskFormerOccDecoder (line 87) | class MaskFormerOccDecoder(BaseModule):
    method __init__ (line 88) | def __init__(self,
    method init_weights (line 121) | def init_weights(self):
    method forward (line 124) | def forward(self, occ_preds, mlvl_feats, img_metas):
  class MaskFormerOccDecoderLayer (line 146) | class MaskFormerOccDecoderLayer(BaseModule):
    method __init__ (line 147) | def __init__(self,
    method init_weights (line 175) | def init_weights(self):
    method forward (line 181) | def forward(self, query_feat, valid_map, mask_pred, occ_preds, mlvl_fe...
    method pred_segmentation (line 202) | def pred_segmentation(self, query_feat, mask_feat):
    method inner_pred_segmentation (line 208) | def inner_pred_segmentation(self, query_feat, mask_feat):
  class MaskFormerSelfAttention (line 217) | class MaskFormerSelfAttention(BaseModule):
    method __init__ (line 218) | def __init__(self, embed_dims, num_heads, dropout=0.0):
    method init_weights (line 224) | def init_weights(self):
    method with_pos_embed (line 229) | def with_pos_embed(self, tensor, pos=None):
    method inner_forward (line 232) | def inner_forward(self, query, mask = None, key_padding_mask = None,qu...
    method forward (line 238) | def forward(self, query, mask = None, key_padding_mask = None,query_po...
  class MaskFormerSampling (line 245) | class MaskFormerSampling(BaseModule):
    method __init__ (line 246) | def __init__(self, embed_dims=256, num_frames=4, num_groups=4, num_poi...
    method init_weights (line 259) | def init_weights(self, ):
    method inner_forward (line 263) | def inner_forward(self, query_feat, valid_map, occ_loc, mlvl_feats, im...
    method forward (line 293) | def forward(self, query_feat, valid_map, occ_loc,  mlvl_feats, img_met...

FILE: models/utils.py
  function conv3d_gn_relu (line 8) | def conv3d_gn_relu(in_channels, out_channels, kernel_size=1, stride=1):
  function deconv3d_gn_relu (line 16) | def deconv3d_gn_relu(in_channels, out_channels, kernel_size=2, stride=2):
  function sparse2dense (line 24) | def sparse2dense(indices, value, dense_shape, empty_value=0):
  function generate_grid (line 38) | def generate_grid(n_vox, interval):
  function batch_indexing (line 46) | def batch_indexing(batched_data: torch.Tensor, batched_indices: torch.Te...
  function rotation_3d_in_axis (line 92) | def rotation_3d_in_axis(points, angles):
  function inverse_sigmoid (line 123) | def inverse_sigmoid(x, eps=1e-5):
  function pad_multiple (line 141) | def pad_multiple(inputs, img_metas, size_divisor=32):
  function rgb_to_hsv (line 160) | def rgb_to_hsv(image: torch.Tensor, eps: float = 1e-8) -> torch.Tensor:
  function hsv_to_rgb (line 215) | def hsv_to_rgb(image: torch.Tensor) -> torch.Tensor:
  class GpuPhotoMetricDistortion (line 256) | class GpuPhotoMetricDistortion:
    method __init__ (line 275) | def __init__(self,
    method __call__ (line 285) | def __call__(self, imgs):
  class DumpConfig (line 345) | class DumpConfig:
    method __init__ (line 346) | def __init__(self):

FILE: old_metrics.py
  function main (line 10) | def main(args):

FILE: ray_metrics.py
  function main (line 14) | def main(args):

FILE: timing.py
  function main (line 17) | def main():

FILE: train.py
  function main (line 20) | def main():

FILE: utils.py
  function init_logging (line 16) | def init_logging(filename=None, debug=False):
  function backup_code (line 30) | def backup_code(work_dir, verbose=False):
  class MyTextLoggerHook (line 45) | class MyTextLoggerHook(TextLoggerHook):
    method _log_info (line 46) | def _log_info(self, log_dict, runner):
    method log (line 84) | def log(self, runner):
    method after_train_epoch (line 121) | def after_train_epoch(self, runner):
  class MyTensorboardLoggerHook (line 132) | class MyTensorboardLoggerHook(LoggerHook):
    method __init__ (line 133) | def __init__(self, log_dir=None, interval=10, ignore_last=True, reset_...
    method before_run (line 139) | def before_run(self, runner):
    method log (line 146) | def log(self, runner):
    method after_run (line 179) | def after_run(self, runner):
  class MyWandbLoggerHook (line 185) | class MyWandbLoggerHook(LoggerHook):
    method __init__ (line 225) | def __init__(self, log_dir=None, project_name=None, team_name=None, ex...
    method import_wandb (line 244) | def import_wandb(self) -> None:
    method before_run (line 253) | def before_run(self, runner) -> None:
    method log (line 276) | def log(self, runner) -> None:
    method after_run (line 296) | def after_run(self, runner) -> None:

FILE: val.py
  function evaluate (line 18) | def evaluate(dataset, results):
  function main (line 28) | def main():

FILE: viz_prediction.py
  function occ2img (line 42) | def occ2img(semantics):
  function main (line 59) | def main():