SYMBOL INDEX (528 symbols across 59 files)

FILE: projects/mmdet3d_plugin/core/apis/mmdet_train.py
  function custom_train_detector (line 31) | def custom_train_detector(model,

FILE: projects/mmdet3d_plugin/core/apis/test.py
  function custom_encode_mask_results (line 25) | def custom_encode_mask_results(mask_results):
  function custom_multi_gpu_test (line 45) | def custom_multi_gpu_test(model, data_loader, tmpdir=None, gpu_collect=F...
  function collect_results_cpu (line 116) | def collect_results_cpu(result_part, size, tmpdir=None):
  function collect_results_gpu (line 163) | def collect_results_gpu(result_part, size):

FILE: projects/mmdet3d_plugin/core/apis/train.py
  function custom_train_model (line 14) | def custom_train_model(model,
  function train_model (line 41) | def train_model(model,

FILE: projects/mmdet3d_plugin/core/bbox/assigners/hungarian_assigner_2d.py
  class HungarianAssigner2D (line 20) | class HungarianAssigner2D(BaseAssigner):
    method __init__ (line 48) | def __init__(self,
    method assign (line 58) | def assign(self,

FILE: projects/mmdet3d_plugin/core/bbox/assigners/hungarian_assigner_3d.py
  class HungarianAssigner3D (line 18) | class HungarianAssigner3D(BaseAssigner):
    method __init__ (line 19) | def __init__(self,
    method assign (line 29) | def assign(self,
  class HungarianAssigner3DPolar (line 94) | class HungarianAssigner3DPolar(BaseAssigner):
    method __init__ (line 95) | def __init__(self,
    method assign (line 105) | def assign(self,

FILE: projects/mmdet3d_plugin/core/bbox/coders/nms_free_coder.py
  class NMSFreeCoder (line 9) | class NMSFreeCoder(BaseBBoxCoder):
    method __init__ (line 21) | def __init__(self,
    method encode (line 36) | def encode(self):
    method decode_single (line 39) | def decode_single(self, cls_scores, bbox_preds):
    method decode (line 93) | def decode(self, preds_dicts):
  class NMSFreeCoderPolar (line 116) | class NMSFreeCoderPolar(BaseBBoxCoder):
    method __init__ (line 128) | def __init__(self,
    method encode (line 143) | def encode(self):
    method decode_single (line 146) | def decode_single(self, cls_scores, bbox_preds):
    method decode (line 199) | def decode(self, preds_dicts):

FILE: projects/mmdet3d_plugin/core/bbox/match_costs/match_cost.py
  class BBox3DL1Cost (line 5) | class BBox3DL1Cost(object):
    method __init__ (line 11) | def __init__(self, weight=1.):
    method __call__ (line 14) | def __call__(self, bbox_pred, gt_bboxes):

FILE: projects/mmdet3d_plugin/core/bbox/util.py
  function normalize_bbox (line 4) | def normalize_bbox(bboxes, pc_range):
  function denormalize_bbox (line 25) | def denormalize_bbox(normalized_bboxes, pc_range):
  function normalize_bbox_polar (line 54) | def normalize_bbox_polar(bboxes, pc_range):
  function denormalize_bbox_polar (line 82) | def denormalize_bbox_polar(normalized_bboxes, pc_range):

FILE: projects/mmdet3d_plugin/core/evaluation/eval_hooks.py
  function _calc_dynamic_intervals (line 17) | def _calc_dynamic_intervals(start_interval, dynamic_interval_list):
  class CustomDistEvalHook (line 29) | class CustomDistEvalHook(BaseDistEvalHook):
    method __init__ (line 31) | def __init__(self, *args, dynamic_intervals=None,  **kwargs):
    method _decide_interval (line 38) | def _decide_interval(self, runner):
    method before_train_epoch (line 45) | def before_train_epoch(self, runner):
    method before_train_iter (line 50) | def before_train_iter(self, runner):
    method _do_evaluate (line 54) | def _do_evaluate(self, runner):

FILE: projects/mmdet3d_plugin/datasets/argoverse2_dataset.py
  class Argoverse2Dataset (line 20) | class Argoverse2Dataset(Custom3DDataset):
    method __init__ (line 23) | def __init__(self,
    method get_cat_ids (line 61) | def get_cat_ids(self, idx):
    method load_annotations (line 85) | def load_annotations(self, ann_file):
    method get_data_info (line 99) | def get_data_info(self, index):
    method get_ann_info (line 182) | def get_ann_info(self, index, input_dict):
    method evaluate (line 224) | def evaluate(self,
    method format_results (line 267) | def format_results(self,
    method box_to_av2 (line 333) | def box_to_av2(self, boxes):

FILE: projects/mmdet3d_plugin/datasets/argoverse2_dataset_t.py
  class Argoverse2DatasetT (line 16) | class Argoverse2DatasetT(Argoverse2Dataset):
    method __init__ (line 19) | def __init__(self, collect_keys, seq_mode=False, seq_split_num=1, num_...
    method _set_sequence_group_flag (line 40) | def _set_sequence_group_flag(self):
    method prepare_train_data (line 82) | def prepare_train_data(self, index):
    method prepare_test_data (line 122) | def prepare_test_data(self, index):
    method union2one (line 136) | def union2one(self, queue):
    method get_data_info (line 143) | def get_data_info(self, index):
    method __getitem__ (line 243) | def __getitem__(self, idx):
    method _rand_another (line 258) | def _rand_another(self, idx):
  function invert_matrix_egopose_numpy (line 266) | def invert_matrix_egopose_numpy(egopose):
  function convert_egopose_to_matrix_numpy (line 276) | def convert_egopose_to_matrix_numpy(rotation, translation):

FILE: projects/mmdet3d_plugin/datasets/av2_eval_util.py
  function evaluate (line 60) | def evaluate(
  function load_mapped_avm_and_egoposes (line 158) | def load_mapped_avm_and_egoposes(log_ids: List[str], dataset_dir: Path
  function read_feather_remote (line 178) | def read_feather_remote(path: Path, columns: Optional[Tuple[str, ...]] =...
  class ArgoverseStaticMapRemote (line 195) | class ArgoverseStaticMapRemote(ArgoverseStaticMap):
    method from_json_remote (line 200) | def from_json_remote(cls, static_map_path_s3, static_map_path: Path) -...
    method from_map_dir_remote (line 235) | def from_map_dir_remote(cls, log_map_dirpath: Path, build_raster: bool...
  class GroundHeightLayerRemote (line 276) | class GroundHeightLayerRemote(GroundHeightLayer):
    method from_file_remote (line 283) | def from_file_remote(cls, log_map_dirpath: Path, log_map_dirpath_s3) -...

FILE: projects/mmdet3d_plugin/datasets/av2_utils.py
  class DetectionCfg (line 35) | class DetectionCfg:
    method metrics_defaults (line 50) | def metrics_defaults(self) -> Tuple[float, ...]:
    method tp_normalization_terms (line 62) | def tp_normalization_terms(self) -> Tuple[float, ...]:
  function accumulate (line 70) | def accumulate(
  function assign (line 115) | def assign(dts: NDArrayFloat, gts: NDArrayFloat, cfg: DetectionCfg) -> T...
  function distance (line 159) | def distance(dts: NDArrayFloat, gts: NDArrayFloat, metric: DistanceType)...
  function compute_affinity_matrix (line 175) | def compute_affinity_matrix(dts: NDArrayFloat, gts: NDArrayFloat, metric...
  function compute_evaluated_dts_mask (line 185) | def compute_evaluated_dts_mask(
  function compute_evaluated_gts_mask (line 205) | def compute_evaluated_gts_mask(
  function compute_objects_in_roi_mask (line 222) | def compute_objects_in_roi_mask(cuboids_ego: NDArrayFloat, city_SE3_ego:...
  function xyz_to_quat (line 241) | def xyz_to_quat(xyz_rad: Tensor) -> Tensor:
  function yaw_to_quat (line 270) | def yaw_to_quat(yaw_rad: Tensor) -> Tensor:

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

FILE: projects/mmdet3d_plugin/datasets/eval_recall.py
  function bbox_cxcywh_to_xyxy (line 7) | def bbox_cxcywh_to_xyxy(bbox):
  function bbox_overlaps (line 25) | def bbox_overlaps(bboxes1,
  function eval_recall (line 86) | def eval_recall(gt_2dbboxes, gt_2d_depth, pred_bboxes, thr, range_m = [0...
  function cal_recall_2d (line 121) | def cal_recall_2d(results_2d):

FILE: projects/mmdet3d_plugin/datasets/nuscenes_dataset.py
  class CustomNuScenesDataset (line 23) | class CustomNuScenesDataset(NuScenesDataset):
    method __init__ (line 29) | def __init__(self, collect_keys, seq_mode=False, seq_split_num=1, num_...
    method _set_sequence_group_flag (line 43) | def _set_sequence_group_flag(self):
    method prepare_train_data (line 82) | def prepare_train_data(self, index):
    method prepare_test_data (line 118) | def prepare_test_data(self, index):
    method union2one (line 132) | def union2one(self, queue):
    method get_data_info (line 148) | def get_data_info(self, index):
    method __getitem__ (line 243) | def __getitem__(self, idx):
  function invert_matrix_egopose_numpy (line 258) | def invert_matrix_egopose_numpy(egopose):
  function convert_egopose_to_matrix_numpy (line 268) | def convert_egopose_to_matrix_numpy(rotation, translation):

FILE: projects/mmdet3d_plugin/datasets/pipelines/custom_pipeline.py
  class AV2LoadMultiViewImageFromFiles (line 18) | class AV2LoadMultiViewImageFromFiles(object):
    method __init__ (line 20) | def __init__(self, to_float32=False, color_type='unchanged'):
    method __call__ (line 24) | def __call__(self, results):
    method __repr__ (line 40) | def __repr__(self):
  class AV2ResizeCropFlipRotImageV2 (line 48) | class AV2ResizeCropFlipRotImageV2():
    method __init__ (line 49) | def __init__(self, data_aug_conf=None, multi_stamps=False):
    method __call__ (line 54) | def __call__(self, results):
    method _bboxes_transform (line 186) | def _bboxes_transform(self, img, bboxes, centers2d, gt_labels, depths,...
    method offline_2d_transform (line 224) | def offline_2d_transform(self, img, bboxes, resize, crop, flip=False):
    method _filter_invisible (line 243) | def _filter_invisible(self, img, bboxes, centers2d, gt_labels, depths):
    method _get_rot (line 269) | def _get_rot(self, h):
    method _img_transform (line 277) | def _img_transform(self, img, resize, resize_dims, crop, flip=False, r...
    method _sample_augmentation (line 313) | def _sample_augmentation(self, img):
    method _sample_augmentation_f (line 328) | def _sample_augmentation_f(self, img):
  class AV2PadMultiViewImage (line 341) | class AV2PadMultiViewImage():
    method __init__ (line 351) | def __init__(self, size=None, size_divisor=None, pad_val=0):
    method _pad_img (line 358) | def _pad_img(self, results):
    method __call__ (line 380) | def __call__(self, results):
    method __repr__ (line 390) | def __repr__(self):
  class AV2DownsampleQuantizeDepthmap (line 398) | class AV2DownsampleQuantizeDepthmap():
    method __init__ (line 404) | def __init__(self, downsample=16, grid_config=None):
    method __call__ (line 412) | def __call__(self, results):
    method _get_downsampled_gt_depth (line 422) | def _get_downsampled_gt_depth(self, gt_depths):
    method _quantize_gt_depth (line 439) | def _quantize_gt_depth(self, gt_depths):
  class AV2DownsampleQuantizeInstanceDepthmap (line 446) | class AV2DownsampleQuantizeInstanceDepthmap():
    method __init__ (line 452) | def __init__(self, downsample=8, depth_config={}):
    method __call__ (line 460) | def __call__(self, results):
    method build_target_depth_from_3dcenter_argo (line 478) | def build_target_depth_from_3dcenter_argo(self, gt_boxes2d, gt_center_...
    method bin_depths (line 501) | def bin_depths(self, depth_map, mode="LID", depth_min=1e-3, depth_max=...

FILE: projects/mmdet3d_plugin/datasets/pipelines/formating.py
  class PETRFormatBundle3D (line 15) | class PETRFormatBundle3D(DefaultFormatBundle):
    method __init__ (line 30) | def __init__(self, class_names, collect_keys, with_gt=True, with_label...
    method __call__ (line 36) | def __call__(self, results):
    method __repr__ (line 108) | def __repr__(self):

FILE: projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py
  class PadMultiViewImage (line 23) | class PadMultiViewImage():
    method __init__ (line 33) | def __init__(self, size=None, size_divisor=None, pad_val=0):
    method _pad_img (line 40) | def _pad_img(self, results):
    method __call__ (line 54) | def __call__(self, results):
    method __repr__ (line 65) | def __repr__(self):
  class NormalizeMultiviewImage (line 74) | class NormalizeMultiviewImage(object):
    method __init__ (line 84) | def __init__(self, mean, std, to_rgb=True):
    method __call__ (line 89) | def __call__(self, results):
    method __repr__ (line 103) | def __repr__(self):
  class ResizeCropFlipRotImage (line 110) | class ResizeCropFlipRotImage():
    method __init__ (line 111) | def __init__(self, data_aug_conf=None, with_2d=True, filter_invisible=...
    method __call__ (line 118) | def __call__(self, results):
    method _bboxes_transform (line 176) | def _bboxes_transform(self, bboxes, centers2d, gt_labels, depths,resiz...
    method _filter_invisible (line 213) | def _filter_invisible(self, bboxes, centers2d, gt_labels, depths):
    method _get_rot (line 240) | def _get_rot(self, h):
    method _img_transform (line 248) | def _img_transform(self, img, resize, resize_dims, crop, flip, rotate):
    method _sample_augmentation (line 276) | def _sample_augmentation(self):

FILE: projects/mmdet3d_plugin/datasets/samplers/distributed_sampler.py
  class DistributedSampler (line 15) | class DistributedSampler(_DistributedSampler):
    method __init__ (line 17) | def __init__(self,
    method __iter__ (line 28) | def __iter__(self):

FILE: projects/mmdet3d_plugin/datasets/samplers/group_sampler.py
  class DistributedGroupSampler (line 22) | class DistributedGroupSampler(Sampler):
    method __init__ (line 40) | def __init__(self,
    method __iter__ (line 69) | def __iter__(self):
    method __len__ (line 112) | def __len__(self):
    method set_epoch (line 115) | def set_epoch(self, epoch):
  function sync_random_seed (line 119) | def sync_random_seed(seed=None, device='cuda'):
  class InfiniteGroupEachSampleInBatchSampler (line 154) | class InfiniteGroupEachSampleInBatchSampler(Sampler):
    method __init__ (line 162) | def __init__(self,
    method _infinite_group_indices (line 204) | def _infinite_group_indices(self):
    method _group_indices_per_global_sample_idx (line 210) | def _group_indices_per_global_sample_idx(self, global_sample_idx):
    method __iter__ (line 216) | def __iter__(self):
    method __len__ (line 231) | def __len__(self):
    method set_epoch (line 235) | def set_epoch(self, epoch):

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

FILE: projects/mmdet3d_plugin/datasets/summarize_metrics_av2.py
  class TruePositiveErrorNames (line 25) | class TruePositiveErrorNames(str, Enum):
  class MetricNames (line 33) | class MetricNames(str, Enum):
  function summarize_metrics (line 44) | def summarize_metrics(
  function compute_average_precision (line 132) | def compute_average_precision(
  function interpolate_precision (line 165) | def interpolate_precision(precision: NDArrayFloat, interpolation_method:...

FILE: projects/mmdet3d_plugin/models/backbones/vovnet.py
  function dw_conv3x3 (line 100) | def dw_conv3x3(in_channels, out_channels, module_name, postfix, stride=1...
  function conv3x3 (line 124) | def conv3x3(in_channels, out_channels, module_name, postfix, stride=1, g...
  function conv1x1 (line 144) | def conv1x1(in_channels, out_channels, module_name, postfix, stride=1, g...
  class Hsigmoid (line 164) | class Hsigmoid(nn.Module):
    method __init__ (line 165) | def __init__(self, inplace=True):
    method forward (line 169) | def forward(self, x):
  class eSEModule (line 173) | class eSEModule(nn.Module):
    method __init__ (line 174) | def __init__(self, channel, reduction=4):
    method forward (line 180) | def forward(self, x):
  class _OSA_module (line 188) | class _OSA_module(nn.Module):
    method __init__ (line 189) | def __init__(
    method forward (line 218) | def forward(self, x):
  class _OSA_stage (line 241) | class _OSA_stage(nn.Sequential):
    method __init__ (line 242) | def __init__(
  class VoVNet (line 277) | class VoVNet(BaseModule):
    method __init__ (line 278) | def __init__(self, spec_name, input_ch=3, out_features=None,
    method _initialize_weights (line 344) | def _initialize_weights(self):
    method forward (line 349) | def forward(self, x):
    method _freeze_stages (line 362) | def _freeze_stages(self):
    method train (line 375) | def train(self, mode=True):

FILE: projects/mmdet3d_plugin/models/backbones/vovnetcp.py
  function dw_conv3x3 (line 101) | def dw_conv3x3(in_channels, out_channels, module_name, postfix, stride=1...
  function conv3x3 (line 125) | def conv3x3(in_channels, out_channels, module_name, postfix, stride=1, g...
  function conv1x1 (line 145) | def conv1x1(in_channels, out_channels, module_name, postfix, stride=1, g...
  class Hsigmoid (line 165) | class Hsigmoid(nn.Module):
    method __init__ (line 166) | def __init__(self, inplace=True):
    method forward (line 170) | def forward(self, x):
  class eSEModule (line 174) | class eSEModule(nn.Module):
    method __init__ (line 175) | def __init__(self, channel, reduction=4):
    method forward (line 181) | def forward(self, x):
  class _OSA_module (line 189) | class _OSA_module(nn.Module):
    method __init__ (line 190) | def __init__(
    method _forward (line 220) | def _forward(self, x):
    method forward (line 242) | def forward(self, x):
  class _OSA_stage (line 252) | class _OSA_stage(nn.Sequential):
    method __init__ (line 253) | def __init__(
  class VoVNetCP (line 288) | class VoVNetCP(BaseModule):
    method __init__ (line 289) | def __init__(self, spec_name, input_ch=3, out_features=None,
    method _initialize_weights (line 355) | def _initialize_weights(self):
    method forward (line 372) | def forward(self, x):
    method _freeze_stages (line 386) | def _freeze_stages(self):
    method train (line 399) | def train(self, mode=True):

FILE: projects/mmdet3d_plugin/models/dense_heads/farhead.py
  class FarHead (line 21) | class FarHead(AnchorFreeHead):
    method __init__ (line 52) | def __init__(self,
    method _init_layers (line 228) | def _init_layers(self):
    method temporal_alignment (line 284) | def temporal_alignment(self, query_pos, tgt, reference_points):
    method prepare_for_dn (line 315) | def prepare_for_dn(self, batch_size, reference_points, img_metas):
    method init_weights (line 432) | def init_weights(self):
    method reset_memory (line 446) | def reset_memory(self):
    method pre_update_memory (line 453) | def pre_update_memory(self, data):
    method post_update_memory (line 479) | def post_update_memory(self, data, rec_ego_pose, all_cls_scores, all_b...
    method _get_gt_depth (line 510) | def _get_gt_depth(self, img_metas, device, BNHW):
    method _convert_bin_depth_to_specific (line 521) | def _convert_bin_depth_to_specific(self, pred_indices, mode='LID', inv...
    method forward (line 533) | def forward(self, img_metas, outs_roi=None, **data):
    method split_offline_pred2d (line 695) | def split_offline_pred2d(self, pred_bbox_list, device):
    method build_query2d_proposal (line 711) | def build_query2d_proposal(self, pred_bbox_list, pred_depth, data, bn,...
    method prepare_for_loss (line 830) | def prepare_for_loss(self, mask_dict):
    method _get_target_single (line 874) | def _get_target_single(self,
    method get_targets (line 933) | def get_targets(self,
    method loss_single (line 984) | def loss_single(self,
    method dn_loss_single (line 1053) | def dn_loss_single(self,
    method loss (line 1114) | def loss(self,
    method get_bboxes (line 1225) | def get_bboxes(self, preds_dicts, img_metas, rescale=False):

FILE: projects/mmdet3d_plugin/models/dense_heads/yolox_head.py
  class YOLOXHeadCustom (line 26) | class YOLOXHeadCustom(BaseDenseHead, BBoxTestMixin):
    method __init__ (line 56) | def __init__(self,
    method _init_layers (line 164) | def _init_layers(self):
    method _build_stacked_convs (line 197) | def _build_stacked_convs(self):
    method _build_predictor (line 221) | def _build_predictor(self):
    method init_weights (line 229) | def init_weights(self):
    method forward_single (line 241) | def forward_single(self, x, cls_convs, reg_convs, conv_cls, conv_reg,
    method forward (line 261) | def forward(self, locations, **data):
    method depth_func (line 343) | def depth_func(self, feat, net):
    method get_2dpred_height (line 347) | def get_2dpred_height(self, bbox_preds, level_idx, HW):
    method get_bboxes (line 356) | def get_bboxes(self, preds_dicts,
    method _bbox_decode (line 491) | def _bbox_decode(self, priors, bbox_preds):
    method _centers2d_decode (line 503) | def _centers2d_decode(self, priors, centers2d):
    method _bboxes_nms (line 507) | def _bboxes_nms(self, cls_scores, bboxes, score_factor, cfg):
    method loss (line 522) | def loss(self,
    method _get_target_single (line 679) | def _get_target_single(self, cls_preds, objectness, priors, decoded_bb...
    method _get_l1_target (line 751) | def _get_l1_target(self, l1_target, gt_bboxes, priors, eps=1e-8):
    method _get_centers2d_target (line 758) | def _get_centers2d_target(self, centers2d_target, centers2d_labels, pr...

FILE: projects/mmdet3d_plugin/models/depth_predictor/ddn_loss/balancer.py
  class Balancer (line 8) | class Balancer(nn.Module):
    method __init__ (line 9) | def __init__(self, fg_weight, bg_weight, downsample_factor=1):
    method forward (line 22) | def forward(self, loss, gt_boxes2d, num_gt_per_img, fg_mask=None):
  function compute_fg_mask (line 55) | def compute_fg_mask(gt_boxes2d, shape, num_gt_per_img, downsample_factor...

FILE: projects/mmdet3d_plugin/models/depth_predictor/ddn_loss/ddn_loss.py
  class DDNLoss (line 14) | class DDNLoss(nn.Module):
    method __init__ (line 16) | def __init__(self,
    method build_target_depth_from_3dcenter (line 48) | def build_target_depth_from_3dcenter(self, depth_logits, gt_boxes2d, g...
    method build_target_depth_from_3dcenter_argo (line 71) | def build_target_depth_from_3dcenter_argo(self, depth_logits, gt_boxes...
    method bin_depths (line 96) | def bin_depths(self, depth_map, mode="LID", depth_min=1e-3, depth_max=...
    method forward (line 134) | def forward(self, depth_logits, gt_boxes2d, gt_center_depth, type='arg...

FILE: projects/mmdet3d_plugin/models/depth_predictor/ddn_loss/focalloss.py
  function one_hot (line 12) | def one_hot(
  function focal_loss (line 55) | def focal_loss(
  class FocalLoss (line 138) | class FocalLoss(nn.Module):
    method __init__ (line 169) | def __init__(self, alpha: float, gamma: float = 2.0, reduction: str = ...
    method forward (line 176) | def forward(self, input: torch.Tensor, target: torch.Tensor) -> torch....
  function binary_focal_loss_with_logits (line 180) | def binary_focal_loss_with_logits(
  class BinaryFocalLossWithLogits (line 248) | class BinaryFocalLossWithLogits(nn.Module):
    method __init__ (line 275) | def __init__(self, alpha: float, gamma: float = 2.0, reduction: str = ...
    method forward (line 281) | def forward(self, input: torch.Tensor, target: torch.Tensor) -> torch....

FILE: projects/mmdet3d_plugin/models/depth_predictor/depth_predictor.py
  class DepthPredictor (line 6) | class DepthPredictor(nn.Module):
    method __init__ (line 8) | def __init__(self, model_cfg):
    method forward (line 62) | def forward(self, feature):

FILE: projects/mmdet3d_plugin/models/detectors/far3d.py
  class Far3D (line 22) | class Far3D(MVXTwoStageDetector):
    method __init__ (line 25) | def __init__(self,
    method extract_img_feat (line 64) | def extract_img_feat(self, img, return_depth=False):
    method extract_feat (line 102) | def extract_feat(self, img, return_depth=False):
    method prepare_location (line 111) | def prepare_location(self, img_metas, **data):
    method forward_roi_head (line 122) | def forward_roi_head(self, location, **data):
    method forward_pts_train (line 128) | def forward_pts_train(self,
    method forward (line 167) | def forward(self, return_loss=True, **data):
    method forward_train (line 182) | def forward_train(self,
    method forward_test (line 232) | def forward_test(self, img_metas, rescale, **data):
    method simple_test_pts (line 244) | def simple_test_pts(self, img_metas, **data):
    method simple_test (line 268) | def simple_test(self, img_metas, **data):

FILE: projects/mmdet3d_plugin/models/necks/cp_fpn.py
  class CPFPN (line 16) | class CPFPN(BaseModule):
    method __init__ (line 67) | def __init__(self,
    method forward (line 157) | def forward(self, inputs):

FILE: projects/mmdet3d_plugin/models/necks/second_fpn.py
  class CustomSECONDFPN (line 12) | class CustomSECONDFPN(BaseModule):
    method __init__ (line 26) | def __init__(self,
    method forward (line 87) | def forward(self, x):

FILE: projects/mmdet3d_plugin/models/utils/attention.py
  function _in_projection_packed (line 20) | def _in_projection_packed(q, k, v, w, b = None):
  class FlashAttention (line 29) | class FlashAttention(nn.Module):
    method __init__ (line 39) | def __init__(self, softmax_scale=None, attention_dropout=0.0, device=N...
    method forward (line 46) | def forward(self, q, kv,
  class FlashMHA (line 94) | class FlashMHA(nn.Module):
    method __init__ (line 96) | def __init__(self, embed_dim, num_heads, bias=True, batch_first=True, ...
    method _reset_parameters (line 119) | def _reset_parameters(self) -> None:
    method forward (line 125) | def forward(self, q, k, v, key_padding_mask=None):

FILE: projects/mmdet3d_plugin/models/utils/deformable_aggregation.py
  class DeformableAggregationFunction (line 7) | class DeformableAggregationFunction(Function):
    method forward (line 9) | def forward(
    method backward (line 41) | def backward(ctx, grad_output):
    method feature_maps_format (line 78) | def feature_maps_format(feature_maps, inverse=False):

FILE: projects/mmdet3d_plugin/models/utils/detr3d_transformer.py
  function get_global_pos (line 27) | def get_global_pos(points, pc_range):
  class Detr3DTransformer (line 32) | class Detr3DTransformer(BaseModule):
    method __init__ (line 43) | def __init__(self,
    method init_weights (line 49) | def init_weights(self):
    method forward (line 58) | def forward(self,
  class Detr3DTransformerDecoder (line 127) | class Detr3DTransformerDecoder(TransformerLayerSequence):
    method __init__ (line 135) | def __init__(self, embed_dims, *args, **kwargs):
    method forward (line 138) | def forward(self,
  class Detr3DTemporalDecoderLayer (line 193) | class Detr3DTemporalDecoderLayer(BaseModule):
    method __init__ (line 229) | def __init__(self,
    method _forward (line 311) | def _forward(self,
    method forward (line 424) | def forward(self,
  class DeformableFeatureAggregationCuda (line 484) | class DeformableFeatureAggregationCuda(BaseModule):
    method __init__ (line 485) | def __init__(
    method init_weight (line 517) | def init_weight(self):
    method forward (line 522) | def forward(self, instance_feature, query_pos,feat_flatten, reference_...
    method _get_weights (line 535) | def _get_weights(self, instance_feature, anchor_embed, lidar2img_mat):
    method feature_sampling (line 544) | def feature_sampling(self, feat_flatten, spatial_flatten, level_start_...

FILE: projects/mmdet3d_plugin/models/utils/grid_mask.py
  class Grid (line 6) | class Grid(object):
    method __init__ (line 7) | def __init__(self, use_h, use_w, rotate = 1, offset=False, ratio = 0.5...
    method set_prob (line 17) | def set_prob(self, epoch, max_epoch):
    method __call__ (line 20) | def __call__(self, img, label):
  class GridMask (line 69) | class GridMask(nn.Module):
    method __init__ (line 70) | def __init__(self, use_h, use_w, rotate = 1, offset=False, ratio = 0.5...
    method set_prob (line 81) | def set_prob(self, epoch, max_epoch):
    method forward (line 84) | def forward(self, x):
  class CustomGridMask (line 125) | class CustomGridMask(nn.Module):
    method __init__ (line 126) | def __init__(self, use_h, use_w, rotate=1, offset=False, ratio_range=(...
    method set_prob (line 140) | def set_prob(self, epoch, max_epoch):
    method forward (line 143) | def forward(self, x):

FILE: projects/mmdet3d_plugin/models/utils/hook.py
  class UseGtDepthHook (line 5) | class UseGtDepthHook(Hook):
    method __init__ (line 6) | def __init__(
    method before_train_iter (line 14) | def before_train_iter(self, runner):

FILE: projects/mmdet3d_plugin/models/utils/layer_decay_optimizer_constructor.py
  function get_vit_lr_decay_rate (line 17) | def get_vit_lr_decay_rate(name, backbone_lr_decay_rate=1.0, head_lr_deca...
  function get_num_layer_layer_wise (line 38) | def get_num_layer_layer_wise(var_name, num_max_layer=12):
  function get_num_layer_stage_wise (line 68) | def get_num_layer_stage_wise(var_name, num_max_layer):
  class LearningRateDecayOptimizerConstructor (line 81) | class LearningRateDecayOptimizerConstructor(DefaultOptimizerConstructor):
    method add_params (line 82) | def add_params(self, params, module, prefix='', is_dcn_module=None):

FILE: projects/mmdet3d_plugin/models/utils/misc.py
  function memory_refresh (line 7) | def memory_refresh(memory, prev_exist):
  function topk_gather (line 13) | def topk_gather(feat, topk_indexes):
  function apply_ltrb (line 26) | def apply_ltrb(locations, pred_ltrb):
  function apply_center_offset (line 44) | def apply_center_offset(locations, center_offset):
  function locations (line 58) | def locations(features, stride, pad_h, pad_w):
  function gaussian_2d (line 88) | def gaussian_2d(shape, sigma=1.0):
  function draw_heatmap_gaussian (line 107) | def draw_heatmap_gaussian(heatmap, center, radius, k=1):
  class SELayer_Linear (line 138) | class SELayer_Linear(nn.Module):
    method __init__ (line 139) | def __init__(self, channels, act_layer=nn.ReLU, gate_layer=nn.Sigmoid):
    method forward (line 146) | def forward(self, x, x_se):
  class MLN (line 153) | class MLN(nn.Module):
    method __init__ (line 160) | def __init__(self, c_dim, f_dim=256, use_ln=True):
    method init_weight (line 176) | def init_weight(self):
    method forward (line 182) | def forward(self, x, c):
  function transform_reference_points (line 193) | def transform_reference_points(reference_points, egopose, reverse=False,...

FILE: projects/mmdet3d_plugin/models/utils/petr_transformer.py
  class PETRMultiheadFlashAttention (line 24) | class PETRMultiheadFlashAttention(BaseModule):
    method __init__ (line 44) | def __init__(self,
    method forward (line 76) | def forward(self,
  class MultiheadAttentionWrapper (line 165) | class MultiheadAttentionWrapper(nn.MultiheadAttention):
    method __init__ (line 166) | def __init__(self, *args, **kwargs):
    method forward_fp16 (line 171) | def forward_fp16(self, *args, **kwargs):
    method forward_fp32 (line 174) | def forward_fp32(self, *args, **kwargs):
    method forward (line 177) | def forward(self, *args, **kwargs):
  class PETRMultiheadAttention (line 184) | class PETRMultiheadAttention(BaseModule):
    method __init__ (line 204) | def __init__(self,
    method forward (line 239) | def forward(self,
  class PETRTransformerEncoder (line 331) | class PETRTransformerEncoder(TransformerLayerSequence):
    method __init__ (line 338) | def __init__(self, *args, post_norm_cfg=dict(type='LN'), **kwargs):
    method forward (line 349) | def forward(self, *args, **kwargs):
  class PETRTransformerDecoder (line 361) | class PETRTransformerDecoder(TransformerLayerSequence):
    method __init__ (line 369) | def __init__(self,
    method forward (line 383) | def forward(self, query, *args, **kwargs):
  class PETRTemporalTransformer (line 412) | class PETRTemporalTransformer(BaseModule):
    method __init__ (line 430) | def __init__(self, encoder=None, decoder=None, init_cfg=None, cross=Fa...
    method init_weights (line 440) | def init_weights(self):
    method forward (line 448) | def forward(self, memory, tgt, query_pos, pos_embed, attn_masks, temp_...
  class PETRTemporalDecoderLayer (line 502) | class PETRTemporalDecoderLayer(BaseModule):
    method __init__ (line 538) | def __init__(self,
    method _forward (line 634) | def _forward(self,
    method forward (line 741) | def forward(self,
  class PETRTransformer (line 789) | class PETRTransformer(BaseModule):
    method __init__ (line 807) | def __init__(self, encoder=None, decoder=None, init_cfg=None, cross=Fa...
    method init_weights (line 817) | def init_weights(self):
    method forward (line 825) | def forward(self, x, mask, query_embed, pos_embed, reg_branch=None):
  class PETRTransformerDecoderLayer (line 868) | class PETRTransformerDecoderLayer(BaseTransformerLayer):
    method __init__ (line 889) | def __init__(self,
    method _forward (line 913) | def _forward(self,
    method forward (line 942) | def forward(self,
  class FlattenMHSelfAttention (line 987) | class FlattenMHSelfAttention(MultiheadAttention):
    method forward (line 989) | def forward(self,

FILE: projects/mmdet3d_plugin/models/utils/positional_encoding.py
  function pos2posemb3d (line 13) | def pos2posemb3d(pos, num_pos_feats=128, temperature=10000):
  function pos2posemb1d (line 27) | def pos2posemb1d(pos, num_pos_feats=256, temperature=10000):
  function nerf_positional_encoding (line 38) | def nerf_positional_encoding(
  class PE (line 82) | class PE(nn.Module):
    method __init__ (line 83) | def __init__(self, positional_encoding, strides, position_range, depth...
    method position_encoding (line 115) | def position_encoding(self, img_feats, img_metas, masks=None):
    method forward (line 168) | def forward(self, mlvl_feats, img_metas):
  class SELayer (line 202) | class SELayer(nn.Module):
    method __init__ (line 203) | def __init__(self, channels, act_layer=nn.ReLU, gate_layer=nn.Sigmoid):
    method forward (line 210) | def forward(self, x, x_se):
  class SinePositionalEncoding3D (line 217) | class SinePositionalEncoding3D(BaseModule):
    method __init__ (line 240) | def __init__(self,
    method forward (line 260) | def forward(self, mask, stride=0):
    method __repr__ (line 300) | def __repr__(self):

FILE: projects/mmdet3d_plugin/models/utils/sparse_blocks.py
  class SparseBox3DRefinementModule (line 18) | class SparseBox3DRefinementModule(BaseModule):
    method __init__ (line 19) | def __init__(
    method init_weight (line 61) | def init_weight(self):
    method forward (line 66) | def forward(self, instance_feature, reference_points, anchor_embed):
    method cls_forward (line 73) | def cls_forward(self, instance_feature):
  class DepthReweightModule (line 78) | class DepthReweightModule(BaseModule):
    method __init__ (line 79) | def __init__(
    method forward (line 109) | def forward(self, features, points_3d, output_conf=False):
  class SparseBox3DKeyPointsGenerator (line 130) | class SparseBox3DKeyPointsGenerator(BaseModule):
    method __init__ (line 131) | def __init__(
    method init_weight (line 153) | def init_weight(self):
    method forward (line 157) | def forward(self, anchor, instance_feature):
  class DeformableFeatureAggregation (line 177) | class DeformableFeatureAggregation(BaseModule):
    method __init__ (line 178) | def __init__(
    method init_weight (line 200) | def init_weight(self):
    method forward (line 206) | def forward(self, instance_feature, reference_points, anchor_embed, fe...
    method _get_weights (line 223) | def _get_weights(self, instance_feature, anchor_embed):
    method multi_view_level_fusion (line 228) | def multi_view_level_fusion(self, features, weights):
    method feature_sampling (line 235) | def feature_sampling(feature_maps, key_points, lidar2img_mat, img_metas):
  class DeformableFeatureAggregationCuda (line 258) | class DeformableFeatureAggregationCuda(BaseModule):
    method __init__ (line 259) | def __init__(
    method init_weight (line 283) | def init_weight(self):
    method forward (line 288) | def forward(self, instance_feature, reference_points, anchor_embed, fe...
    method _get_weights (line 299) | def _get_weights(self, instance_feature, anchor_embed):
    method multi_view_level_fusion (line 305) | def multi_view_level_fusion(self, features, weights):
    method feature_sampling (line 311) | def feature_sampling(self, feat_flatten, spatial_flatten, level_start_...
  class SparseBox3DEncoder (line 343) | class SparseBox3DEncoder(BaseModule):
    method __init__ (line 344) | def __init__(self, embed_dims=256, vel_dims=0):
    method forward (line 363) | def forward(self, box_3d):
  function get_global_pos (line 374) | def get_global_pos(points, pc_range):

FILE: projects/mmdet3d_plugin/models/utils/warmup_fp16_optimizer.py
  class WarmupFp16OptimizerHook (line 19) | class WarmupFp16OptimizerHook(OptimizerHook):
    method __init__ (line 44) | def __init__(self,
    method before_run (line 72) | def before_run(self, runner) -> None:
    method copy_grads_to_fp32 (line 81) | def copy_grads_to_fp32(self, fp16_net: nn.Module,
    method copy_params_to_fp16 (line 92) | def copy_params_to_fp16(self, fp16_net: nn.Module,
    method after_train_iter (line 99) | def after_train_iter(self, runner) -> None:

FILE: tools/analysis_tools/benchmark.py
  function parse_args (line 18) | def parse_args():
  function main (line 29) | def main():

FILE: tools/create_data_nusc.py
  function nuscenes_data_prep (line 9) | def nuscenes_data_prep(root_path,

FILE: tools/create_infos_av2/create_av2_infos.py
  function create_av2_infos (line 38) | def create_av2_infos(dataset_dir, split, out_dir):
  function _load_annotations (line 116) | def _load_annotations(split, log_id, timestamp_ns):
  function intrinsics_matrix (line 126) | def intrinsics_matrix(fx, fy, cx, cy):
  function get_gt3d_data (line 134) | def get_gt3d_data(dataset_dir, split, log_id, timestamp_ns):
  function get_gt2d_data (line 168) | def get_gt2d_data(synchronized_imagery, cam_infos, timestamp_city_SE3_eg...
  function post_process_coords (line 260) | def post_process_coords(corner_coords, imsize = (2048, 1550)):

FILE: tools/data_converter/info2coco.py
  function ref (line 8) | def ref():
  function convert_info_to_coco (line 90) | def convert_info_to_coco(data_infos, name2nori, images, annotations):

FILE: tools/data_converter/nuscenes_converter.py
  function create_nuscenes_infos (line 27) | def create_nuscenes_infos(root_path,
  function get_available_scenes (line 106) | def get_available_scenes(nusc):
  function _fill_trainval_infos (line 146) | def _fill_trainval_infos(nusc,
  function obtain_sensor2top (line 355) | def obtain_sensor2top(nusc,
  function export_2d_annotation (line 416) | def export_2d_annotation(root_path, info_path, version, mono3d=True):
  function get_2d_boxes (line 481) | def get_2d_boxes(nusc,
  function post_process_coords (line 613) | def post_process_coords(
  function generate_record (line 646) | def generate_record(ann_rec: dict, x1: float, y1: float, x2: float, y2: ...

FILE: tools/filter_ckpt.py
  function filter_params (line 4) | def filter_params(ckpt, left_prefix, inverse=False):

FILE: tools/test.py
  function parse_args (line 27) | def parse_args():
  function main (line 110) | def main():

FILE: tools/test_and_visualize.py
  function parse_args (line 27) | def parse_args():
  function main (line 110) | def main():

FILE: tools/train.py
  function parse_args (line 30) | def parse_args():
  function main (line 99) | def main():

FILE: tools/visual/check_img_label.py
  function _bbox_decode (line 24) | def _bbox_decode(priors, bbox_preds):
  function _centers2d_decode (line 34) | def _centers2d_decode(priors, pred_centers2d):
  class NormalizeInverse (line 38) | class NormalizeInverse(torchvision.transforms.Normalize):
    method __init__ (line 40) | def __init__(self, mean, std):
    method __call__ (line 47) | def __call__(self, tensor):

FILE: tools/visual/vis_3dpred_depth_stat.py
  function _convert_bin_depth_to_specific (line 118) | def _convert_bin_depth_to_specific(pred_indices):

FILE: tools/visual/vis_3dpred_depth_stat2.py
  function _convert_bin_depth_to_specific (line 34) | def _convert_bin_depth_to_specific(pred_indices):

FILE: tools/visual/vis_util.py
  function _bbox_decode (line 4) | def _bbox_decode(priors, bbox_preds):
  function _centers2d_decode (line 15) | def _centers2d_decode(priors, pred_centers2d):
  function bbox_xyxy_to_cxcywh (line 18) | def bbox_xyxy_to_cxcywh(bbox):

FILE: tools/visual_nuscenes.py
  class NuScenes (line 59) | class NuScenes:
    method __init__ (line 64) | def __init__(self,
    method table_root (line 155) | def table_root(self) -> str:
    method __load_table__ (line 159) | def __load_table__(self, table_name, pred=False) -> dict:
    method load_lidarseg_cat_name_mapping (line 179) | def load_lidarseg_cat_name_mapping(self):
    method __make_reverse_index__ (line 189) | def __make_reverse_index__(self, verbose: bool) -> None:
    method get (line 248) | def get(self, table_name: str, token: str) -> dict:
    method getind (line 259) | def getind(self, table_name: str, token: str) -> int:
    method field2token (line 268) | def field2token(self, table_name: str, field: str, query) -> List[str]:
    method get_sample_data_path (line 283) | def get_sample_data_path(self, sample_data_token: str) -> str:
    method get_sample_data (line 289) | def get_sample_data(self, sample_data_token: str,
    method get_box (line 351) | def get_box(self, sample_annotation_token: str) -> Box:
    method get_boxes (line 360) | def get_boxes(self, sample_data_token: str) -> List[Box]:
    method box_velocity (line 418) | def box_velocity(self, sample_annotation_token: str, max_time_diff: fl...
    method get_sample_lidarseg_stats (line 465) | def get_sample_lidarseg_stats(self,
    method list_categories (line 538) | def list_categories(self) -> None:
    method list_lidarseg_categories (line 541) | def list_lidarseg_categories(self, sort_by: str = 'count', gt_from: st...
    method list_panoptic_instances (line 544) | def list_panoptic_instances(self, sort_by: str = 'count', get_hist: bo...
    method list_attributes (line 547) | def list_attributes(self) -> None:
    method list_scenes (line 550) | def list_scenes(self) -> None:
    method list_sample (line 553) | def list_sample(self, sample_token: str) -> None:
    method render_pointcloud_in_image (line 556) | def render_pointcloud_in_image(self, sample_token: str, dot_size: int ...
    method render_sample (line 575) | def render_sample(self, sample_token: str,
    method render_sample_data (line 589) | def render_sample_data(self, sample_data_token: str, with_anns: bool =...
    method render_annotation (line 609) | def render_annotation(self, sample_annotation_token: str, margin: floa...
    method render_instance (line 614) | def render_instance(self, instance_token: str, margin: float = 10, vie...
    method render_scene (line 619) | def render_scene(self, scene_token: str, freq: float = 10, imsize: Tup...
    method render_scene_channel (line 623) | def render_scene_channel(self, scene_token: str, channel: str = 'CAM_F...
    method render_egoposes_on_map (line 627) | def render_egoposes_on_map(self, log_location: str, scene_tokens: List...
    method render_scene_channel_lidarseg (line 630) | def render_scene_channel_lidarseg(self, scene_token: str,
    method render_scene_lidarseg (line 655) | def render_scene_lidarseg(self, scene_token: str,
  class NuScenesExplorer (line 677) | class NuScenesExplorer:
    method __init__ (line 681) | def __init__(self, nusc: NuScenes):
    method get_color (line 684) | def get_color(self, category_name: str) -> Tuple[int, int, int]:
    method list_categories (line 692) | def list_categories(self) -> None:
    method list_lidarseg_categories (line 713) | def list_lidarseg_categories(self, sort_by: str = 'count', gt_from: st...
    method list_panoptic_instances (line 766) | def list_panoptic_instances(self, sort_by: str = 'count', get_hist: bo...
    method list_attributes (line 820) | def list_attributes(self) -> None:
    method list_scenes (line 833) | def list_scenes(self) -> None:
    method list_sample (line 861) | def list_sample(self, sample_token: str) -> None:
    method map_pointcloud_to_image (line 875) | def map_pointcloud_to_image(self,
    method render_pointcloud_in_image (line 1015) | def render_pointcloud_in_image(self,
    method render_sample (line 1105) | def render_sample(self,
    method render_ego_centric_map (line 1203) | def render_ego_centric_map(self,
    method render_sample_data (line 1262) | def render_sample_data(self,
    method render_annotation (line 1515) | def render_annotation(self,
    method render_instance (line 1604) | def render_instance(self,
    method render_scene (line 1634) | def render_scene(self,
    method render_scene_channel (line 1748) | def render_scene_channel(self,
    method render_egoposes_on_map (line 1825) | def render_egoposes_on_map(self,
    method _plot_points_and_bboxes (line 1918) | def _plot_points_and_bboxes(self,
    method render_scene_channel_lidarseg (line 1990) | def render_scene_channel_lidarseg(self,
    method render_scene_lidarseg (line 2139) | def render_scene_lidarseg(self,