SYMBOL INDEX (3065 symbols across 326 files)

FILE: mmdet3d/__init__.py
  function digit_version (line 9) | def digit_version(version_str):

FILE: mmdet3d/apis/inference.py
  function convert_SyncBN (line 22) | def convert_SyncBN(config):
  function init_model (line 38) | def init_model(config, checkpoint=None, device='cuda:0'):
  function inference_detector (line 81) | def inference_detector(model, pcd):
  function inference_multi_modality_detector (line 155) | def inference_multi_modality_detector(model, pcd, image, ann_file):
  function inference_mono_3d_detector (line 230) | def inference_mono_3d_detector(model, image, ann_file):
  function inference_segmentor (line 289) | def inference_segmentor(model, pcd):
  function show_det_result_meshlab (line 328) | def show_det_result_meshlab(data,
  function show_seg_result_meshlab (line 371) | def show_seg_result_meshlab(data,
  function show_proj_det_result_meshlab (line 403) | def show_proj_det_result_meshlab(data,
  function show_result_meshlab (line 484) | def show_result_meshlab(data,

FILE: mmdet3d/apis/test.py
  function single_gpu_test (line 12) | def single_gpu_test(model,
  function custom_encode_mask_results (line 117) | def custom_encode_mask_results(mask_results):
  function custom_multi_gpu_test (line 137) | def custom_multi_gpu_test(model, data_loader, tmpdir=None, gpu_collect=F...
  function collect_results_cpu (line 210) | def collect_results_cpu(result_part, size, tmpdir=None):
  function collect_results_gpu (line 259) | def collect_results_gpu(result_part, size):

FILE: mmdet3d/apis/train.py
  function init_random_seed (line 29) | def init_random_seed(seed=None, device='cuda'):
  function set_random_seed (line 60) | def set_random_seed(seed, deterministic=False):
  function train_segmentor (line 79) | def train_segmentor(model,
  function train_detector (line 183) | def train_detector(model,
  function train_model (line 330) | def train_model(model,

FILE: mmdet3d/core/anchor/anchor_3d_generator.py
  class Anchor3DRangeGenerator (line 9) | class Anchor3DRangeGenerator(object):
    method __init__ (line 39) | def __init__(self,
    method __repr__ (line 67) | def __repr__(self):
    method num_base_anchors (line 78) | def num_base_anchors(self):
    method num_levels (line 85) | def num_levels(self):
    method grid_anchors (line 89) | def grid_anchors(self, featmap_sizes, device='cuda'):
    method single_level_grid_anchors (line 115) | def single_level_grid_anchors(self, featmap_size, scale, device='cuda'):
    method anchors_single_range (line 155) | def anchors_single_range(self,
  class AlignedAnchor3DRangeGenerator (line 225) | class AlignedAnchor3DRangeGenerator(Anchor3DRangeGenerator):
    method __init__ (line 251) | def __init__(self, align_corner=False, **kwargs):
    method anchors_single_range (line 255) | def anchors_single_range(self,
  class AlignedAnchor3DRangeGeneratorPerCls (line 344) | class AlignedAnchor3DRangeGeneratorPerCls(AlignedAnchor3DRangeGenerator):
    method __init__ (line 355) | def __init__(self, **kwargs):
    method grid_anchors (line 360) | def grid_anchors(self, featmap_sizes, device='cuda'):
    method multi_cls_grid_anchors (line 382) | def multi_cls_grid_anchors(self, featmap_sizes, scale, device='cuda'):

FILE: mmdet3d/core/bbox/box_np_ops.py
  function camera_to_lidar (line 13) | def camera_to_lidar(points, r_rect, velo2cam):
  function box_camera_to_lidar (line 36) | def box_camera_to_lidar(data, r_rect, velo2cam):
  function corners_nd (line 62) | def corners_nd(dims, origin=0.5):
  function center_to_corner_box2d (line 96) | def center_to_corner_box2d(centers, dims, angles=None, origin=0.5):
  function depth_to_points (line 123) | def depth_to_points(depth, trunc_pixel):
  function depth_to_lidar_points (line 147) | def depth_to_lidar_points(depth, trunc_pixel, P2, r_rect, velo2cam):
  function center_to_corner_box3d (line 171) | def center_to_corner_box3d(centers,
  function box2d_to_corner_jit (line 204) | def box2d_to_corner_jit(boxes):
  function corner_to_standup_nd_jit (line 235) | def corner_to_standup_nd_jit(boxes_corner):
  function corner_to_surfaces_3d_jit (line 256) | def corner_to_surfaces_3d_jit(corners):
  function rotation_points_single_angle (line 279) | def rotation_points_single_angle(points, angle, axis=0):
  function box3d_to_bbox (line 311) | def box3d_to_bbox(box3d, P2):
  function corner_to_surfaces_3d (line 331) | def corner_to_surfaces_3d(corners):
  function points_in_rbbox (line 353) | def points_in_rbbox(points, rbbox, z_axis=2, origin=(0.5, 0.5, 0)):
  function minmax_to_corner_2d (line 379) | def minmax_to_corner_2d(minmax_box):
  function create_anchors_3d_range (line 394) | def create_anchors_3d_range(feature_size,
  function center_to_minmax_2d (line 444) | def center_to_minmax_2d(centers, dims, origin=0.5):
  function rbbox2d_to_near_bbox (line 463) | def rbbox2d_to_near_bbox(rbboxes):
  function iou_jit (line 483) | def iou_jit(boxes, query_boxes, mode='iou', eps=0.0):
  function projection_matrix_to_CRT_kitti (line 526) | def projection_matrix_to_CRT_kitti(proj):
  function remove_outside_points (line 553) | def remove_outside_points(points, rect, Trv2c, P2, image_shape):
  function get_frustum (line 584) | def get_frustum(bbox_image, C, near_clip=0.001, far_clip=100):
  function surface_equ_3d (line 617) | def surface_equ_3d(polygon_surfaces):
  function _points_in_convex_polygon_3d_jit (line 642) | def _points_in_convex_polygon_3d_jit(points, polygon_surfaces, normal_ve...
  function points_in_convex_polygon_3d_jit (line 679) | def points_in_convex_polygon_3d_jit(points,
  function points_in_convex_polygon_jit (line 709) | def points_in_convex_polygon_jit(points, polygon, clockwise=False):
  function boxes3d_to_corners3d_lidar (line 753) | def boxes3d_to_corners3d_lidar(boxes3d, bottom_center=True):

FILE: mmdet3d/core/bbox/coders/anchor_free_bbox_coder.py
  class AnchorFreeBBoxCoder (line 10) | class AnchorFreeBBoxCoder(PartialBinBasedBBoxCoder):
    method __init__ (line 18) | def __init__(self, num_dir_bins, with_rot=True):
    method encode (line 24) | def encode(self, gt_bboxes_3d, gt_labels_3d):
    method decode (line 54) | def decode(self, bbox_out):
    method split_pred (line 88) | def split_pred(self, cls_preds, reg_preds, base_xyz):

FILE: mmdet3d/core/bbox/coders/centerpoint_bbox_coders.py
  class CenterPointBBoxCoder (line 9) | class CenterPointBBoxCoder(BaseBBoxCoder):
    method __init__ (line 24) | def __init__(self,
    method _gather_feat (line 41) | def _gather_feat(self, feats, inds, feat_masks=None):
    method _topk (line 63) | def _topk(self, scores, K=80):
    method _transpose_and_gather_feat (line 98) | def _transpose_and_gather_feat(self, feat, ind):
    method encode (line 114) | def encode(self):
    method decode (line 117) | def decode(self,

FILE: mmdet3d/core/bbox/coders/delta_xyzwhlr_bbox_coder.py
  class DeltaXYZWLHRBBoxCoder (line 9) | class DeltaXYZWLHRBBoxCoder(BaseBBoxCoder):
    method __init__ (line 16) | def __init__(self, code_size=7):
    method encode (line 21) | def encode(src_boxes, dst_boxes):
    method decode (line 58) | def decode(anchors, deltas):

FILE: mmdet3d/core/bbox/coders/fcos3d_bbox_coder.py
  class FCOS3DBBoxCoder (line 11) | class FCOS3DBBoxCoder(BaseBBoxCoder):
    method __init__ (line 24) | def __init__(self,
    method encode (line 35) | def encode(self, gt_bboxes_3d, gt_labels_3d, gt_bboxes, gt_labels):
    method decode (line 39) | def decode(self, bbox, scale, stride, training, cls_score=None):
    method decode_yaw (line 103) | def decode_yaw(bbox, centers2d, dir_cls, dir_offset, cam2img):

FILE: mmdet3d/core/bbox/coders/groupfree3d_bbox_coder.py
  class GroupFree3DBBoxCoder (line 10) | class GroupFree3DBBoxCoder(PartialBinBasedBBoxCoder):
    method __init__ (line 23) | def __init__(self,
    method encode (line 36) | def encode(self, gt_bboxes_3d, gt_labels_3d):
    method decode (line 68) | def decode(self, bbox_out, prefix=''):
    method split_pred (line 119) | def split_pred(self, cls_preds, reg_preds, base_xyz, prefix=''):

FILE: mmdet3d/core/bbox/coders/monoflex_bbox_coder.py
  class MonoFlexCoder (line 11) | class MonoFlexCoder(BaseBBoxCoder):
    method __init__ (line 37) | def __init__(self,
    method encode (line 74) | def encode(self, gt_bboxes_3d):
    method decode (line 108) | def decode(self, bbox, base_centers2d, labels, downsample_ratio, cam2i...
    method decode_direct_depth (line 214) | def decode_direct_depth(self, depth_offsets):
    method decode_location (line 241) | def decode_location(self,
    method keypoints2depth (line 286) | def keypoints2depth(self,
    method decode_dims (line 386) | def decode_dims(self, labels, dims_offset):
    method decode_orientation (line 414) | def decode_orientation(self, ori_vector, locations):
    method decode_bboxes2d (line 470) | def decode_bboxes2d(self, reg_bboxes2d, base_centers2d):
    method combine_depths (line 495) | def combine_depths(self, depth, depth_uncertainty):

FILE: mmdet3d/core/bbox/coders/partial_bin_based_bbox_coder.py
  class PartialBinBasedBBoxCoder (line 10) | class PartialBinBasedBBoxCoder(BaseBBoxCoder):
    method __init__ (line 20) | def __init__(self, num_dir_bins, num_sizes, mean_sizes, with_rot=True):
    method encode (line 28) | def encode(self, gt_bboxes_3d, gt_labels_3d):
    method decode (line 59) | def decode(self, bbox_out, suffix=''):
    method decode_corners (line 102) | def decode_corners(self, center, size_res, size_class):
    method split_pred (line 140) | def split_pred(self, cls_preds, reg_preds, base_xyz):
    method angle2class (line 204) | def angle2class(self, angle):
    method class2angle (line 225) | def class2angle(self, angle_cls, angle_res, limit_period=True):

FILE: mmdet3d/core/bbox/coders/pgd_bbox_coder.py
  class PGDBBoxCoder (line 11) | class PGDBBoxCoder(FCOS3DBBoxCoder):
    method encode (line 14) | def encode(self, gt_bboxes_3d, gt_labels_3d, gt_bboxes, gt_labels):
    method decode_2d (line 18) | def decode_2d(self,
    method decode_prob_depth (line 73) | def decode_prob_depth(self, depth_cls_preds, depth_range, depth_unit,

FILE: mmdet3d/core/bbox/coders/point_xyzwhlr_bbox_coder.py
  class PointXYZWHLRBBoxCoder (line 10) | class PointXYZWHLRBBoxCoder(BaseBBoxCoder):
    method __init__ (line 21) | def __init__(self, code_size=7, use_mean_size=True, mean_size=None):
    method encode (line 31) | def encode(self, gt_bboxes_3d, points, gt_labels_3d=None):
    method decode (line 78) | def decode(self, box_encodings, points, pred_labels_3d=None):

FILE: mmdet3d/core/bbox/coders/smoke_bbox_coder.py
  class SMOKECoder (line 10) | class SMOKECoder(BaseBBoxCoder):
    method __init__ (line 20) | def __init__(self, base_depth, base_dims, code_size):
    method encode (line 26) | def encode(self, locations, dimensions, orientations, input_metas):
    method decode (line 52) | def decode(self,
    method _decode_depth (line 107) | def _decode_depth(self, depth_offsets):
    method _decode_location (line 114) | def _decode_location(self, points, centers2d_offsets, depths, cam2imgs,
    method _decode_dimension (line 155) | def _decode_dimension(self, labels, dims_offset):
    method _decode_orientation (line 171) | def _decode_orientation(self, ori_vector, locations):

FILE: mmdet3d/core/bbox/iou_calculators/iou3d_calculator.py
  class BboxOverlapsNearest3D (line 10) | class BboxOverlapsNearest3D(object):
    method __init__ (line 21) | def __init__(self, coordinate='lidar'):
    method __call__ (line 25) | def __call__(self, bboxes1, bboxes2, mode='iou', is_aligned=False):
    method __repr__ (line 50) | def __repr__(self):
  class BboxOverlaps3D (line 58) | class BboxOverlaps3D(object):
    method __init__ (line 66) | def __init__(self, coordinate):
    method __call__ (line 70) | def __call__(self, bboxes1, bboxes2, mode='iou'):
    method __repr__ (line 92) | def __repr__(self):
  function bbox_overlaps_nearest_3d (line 99) | def bbox_overlaps_nearest_3d(bboxes1,
  function bbox_overlaps_3d (line 148) | def bbox_overlaps_3d(bboxes1, bboxes2, mode='iou', coordinate='camera'):
  class AxisAlignedBboxOverlaps3D (line 180) | class AxisAlignedBboxOverlaps3D(object):
    method __call__ (line 183) | def __call__(self, bboxes1, bboxes2, mode='iou', is_aligned=False):
    method __repr__ (line 204) | def __repr__(self):
  function axis_aligned_bbox_overlaps_3d (line 210) | def axis_aligned_bbox_overlaps_3d(bboxes1,

FILE: mmdet3d/core/bbox/samplers/iou_neg_piecewise_sampler.py
  class IoUNegPiecewiseSampler (line 9) | class IoUNegPiecewiseSampler(RandomSampler):
    method __init__ (line 29) | def __init__(self,
    method _sample_pos (line 47) | def _sample_pos(self, assign_result, num_expected, **kwargs):
    method _sample_neg (line 57) | def _sample_neg(self, assign_result, num_expected, **kwargs):
    method sample (line 125) | def sample(self,

FILE: mmdet3d/core/bbox/structures/base_box3d.py
  class BaseInstance3DBoxes (line 12) | class BaseInstance3DBoxes(object):
    method __init__ (line 39) | def __init__(self, tensor, box_dim=7, with_yaw=True, origin=(0.5, 0.5,...
    method volume (line 71) | def volume(self):
    method dims (line 76) | def dims(self):
    method yaw (line 81) | def yaw(self):
    method height (line 86) | def height(self):
    method top_height (line 91) | def top_height(self):
    method bottom_height (line 97) | def bottom_height(self):
    method center (line 103) | def center(self):
    method bottom_center (line 122) | def bottom_center(self):
    method gravity_center (line 127) | def gravity_center(self):
    method corners (line 132) | def corners(self):
    method bev (line 138) | def bev(self):
    method nearest_bev (line 144) | def nearest_bev(self):
    method in_range_bev (line 164) | def in_range_bev(self, box_range):
    method rotate (line 186) | def rotate(self, angle, points=None):
    method flip (line 200) | def flip(self, bev_direction='horizontal'):
    method translate (line 210) | def translate(self, trans_vector):
    method in_range_3d (line 220) | def in_range_3d(self, box_range):
    method convert_to (line 245) | def convert_to(self, dst, rt_mat=None):
    method scale (line 263) | def scale(self, scale_factor):
    method limit_yaw (line 272) | def limit_yaw(self, offset=0.5, period=np.pi):
    method nonempty (line 281) | def nonempty(self, threshold=0.0):
    method __getitem__ (line 303) | def __getitem__(self, item):
    method __len__ (line 332) | def __len__(self):
    method __repr__ (line 336) | def __repr__(self):
    method cat (line 341) | def cat(cls, boxes_list):
    method to (line 363) | def to(self, device):
    method clone (line 379) | def clone(self):
    method device (line 391) | def device(self):
    method __iter__ (line 395) | def __iter__(self):
    method height_overlaps (line 404) | def height_overlaps(cls, boxes1, boxes2, mode='iou'):
    method overlaps (line 436) | def overlaps(cls, boxes1, boxes2, mode='iou'):
    method new_box (line 489) | def new_box(self, data):
    method points_in_boxes_part (line 508) | def points_in_boxes_part(self, points, boxes_override=None):
    method points_in_boxes_all (line 537) | def points_in_boxes_all(self, points, boxes_override=None):
    method points_in_boxes (line 568) | def points_in_boxes(self, points, boxes_override=None):
    method points_in_boxes_batch (line 574) | def points_in_boxes_batch(self, points, boxes_override=None):

FILE: mmdet3d/core/bbox/structures/box_3d_mode.py
  class Box3DMode (line 15) | class Box3DMode(IntEnum):
    method convert (line 66) | def convert(box, src, dst, rt_mat=None, with_yaw=True):

FILE: mmdet3d/core/bbox/structures/cam_box3d.py
  class CameraInstance3DBoxes (line 10) | class CameraInstance3DBoxes(BaseInstance3DBoxes):
    method __init__ (line 40) | def __init__(self,
    method height (line 76) | def height(self):
    method top_height (line 81) | def top_height(self):
    method bottom_height (line 88) | def bottom_height(self):
    method local_yaw (line 94) | def local_yaw(self):
    method gravity_center (line 108) | def gravity_center(self):
    method corners (line 117) | def corners(self):
    method bev (line 160) | def bev(self):
    method rotate (line 170) | def rotate(self, angle, points=None):
    method flip (line 218) | def flip(self, bev_direction='horizontal', points=None):
    method height_overlaps (line 253) | def height_overlaps(cls, boxes1, boxes2, mode='iou'):
    method convert_to (line 283) | def convert_to(self, dst, rt_mat=None):
    method points_in_boxes_part (line 303) | def points_in_boxes_part(self, points, boxes_override=None):
    method points_in_boxes_all (line 330) | def points_in_boxes_all(self, points, boxes_override=None):

FILE: mmdet3d/core/bbox/structures/coord_3d_mode.py
  class Coord3DMode (line 13) | class Coord3DMode(IntEnum):
    method convert (line 65) | def convert(input, src, dst, rt_mat=None, with_yaw=True, is_point=True):
    method convert_box (line 109) | def convert_box(box, src, dst, rt_mat=None, with_yaw=True):
    method convert_point (line 136) | def convert_point(point, src, dst, rt_mat=None):

FILE: mmdet3d/core/bbox/structures/custom_box.py
  class CustomBox (line 23) | class CustomBox(Box):
    method __init__ (line 26) | def __init__(self,
    method render (line 71) | def render(self,

FILE: mmdet3d/core/bbox/structures/depth_box3d.py
  class DepthInstance3DBoxes (line 10) | class DepthInstance3DBoxes(BaseInstance3DBoxes):
    method gravity_center (line 43) | def gravity_center(self):
    method corners (line 52) | def corners(self):
    method rotate (line 93) | def rotate(self, angle, points=None):
    method flip (line 153) | def flip(self, bev_direction='horizontal', points=None):
    method convert_to (line 188) | def convert_to(self, dst, rt_mat=None):
    method enlarged_box (line 208) | def enlarged_box(self, extra_width):
    method get_surface_line_center (line 223) | def get_surface_line_center(self):

FILE: mmdet3d/core/bbox/structures/lidar_box3d.py
  class LiDARInstance3DBoxes (line 10) | class LiDARInstance3DBoxes(BaseInstance3DBoxes):
    method gravity_center (line 41) | def gravity_center(self):
    method corners (line 50) | def corners(self):
    method rotate (line 91) | def rotate(self, angle, points=None):
    method flip (line 143) | def flip(self, bev_direction='horizontal', points=None):
    method convert_to (line 177) | def convert_to(self, dst, rt_mat=None):
    method enlarged_box (line 197) | def enlarged_box(self, extra_width):

FILE: mmdet3d/core/bbox/structures/utils.py
  function limit_period (line 11) | def limit_period(val, offset=0.5, period=np.pi):
  function rotation_3d_in_axis (line 29) | def rotation_3d_in_axis(points,
  function xywhr2xyxyr (line 121) | def xywhr2xyxyr(boxes_xywhr):
  function get_box_type (line 142) | def get_box_type(box_type):
  function points_cam2img (line 176) | def points_cam2img(points_3d, proj_mat, with_depth=False):
  function points_img2cam (line 218) | def points_img2cam(points, cam2img):
  function mono_cam_box2vis (line 251) | def mono_cam_box2vis(cam_box):
  function get_proj_mat_by_coord_type (line 297) | def get_proj_mat_by_coord_type(img_meta, coord_type):
  function yaw2local (line 314) | def yaw2local(yaw, loc):

FILE: mmdet3d/core/bbox/transforms.py
  function bbox3d_mapping_back (line 5) | def bbox3d_mapping_back(bboxes, scale_factor, flip_horizontal, flip_vert...
  function bbox3d2roi (line 27) | def bbox3d2roi(bbox_list):
  function bbox3d2result (line 50) | def bbox3d2result(bboxes, scores, labels, attrs=None):

FILE: mmdet3d/core/bbox/util.py
  function normalize_bbox (line 3) | def normalize_bbox(bboxes, pc_range):
  function denormalize_bbox (line 26) | def denormalize_bbox(normalized_bboxes, pc_range):

FILE: mmdet3d/core/evaluation/indoor_eval.py
  function average_precision (line 8) | def average_precision(recalls, precisions, mode='area'):
  function eval_det_cls (line 56) | def eval_det_cls(pred, gt, iou_thr=None):
  function eval_map_recall (line 164) | def eval_map_recall(pred, gt, ovthresh=None):
  function indoor_eval (line 203) | def indoor_eval(gt_annos,

FILE: mmdet3d/core/evaluation/instance_seg_eval.py
  function aggregate_predictions (line 9) | def aggregate_predictions(masks, labels, scores, valid_class_ids):
  function rename_gt (line 39) | def rename_gt(gt_semantic_masks, gt_instance_masks, valid_class_ids):
  function instance_seg_eval (line 70) | def instance_seg_eval(gt_semantic_masks,

FILE: mmdet3d/core/evaluation/kitti_utils/eval.py
  function get_thresholds (line 10) | def get_thresholds(scores: np.ndarray, num_gt, num_sample_pts=41):
  function clean_data (line 30) | def clean_data(gt_anno, dt_anno, current_class, difficulty):
  function image_box_overlap (line 86) | def image_box_overlap(boxes, query_boxes, criterion=-1):
  function bev_box_overlap (line 117) | def bev_box_overlap(boxes, qboxes, criterion=-1):
  function d3_box_overlap_kernel (line 124) | def d3_box_overlap_kernel(boxes, qboxes, rinc, criterion=-1):
  function d3_box_overlap (line 155) | def d3_box_overlap(boxes, qboxes, criterion=-1):
  function compute_statistics_jit (line 164) | def compute_statistics_jit(overlaps,
  function get_split_parts (line 284) | def get_split_parts(num, num_part):
  function fused_compute_statistics (line 294) | def fused_compute_statistics(overlaps,
  function calculate_iou_partly (line 343) | def calculate_iou_partly(gt_annos, dt_annos, metric, num_parts=50):
  function _prepare_data (line 421) | def _prepare_data(gt_annos, dt_annos, current_class, difficulty):
  function eval_class (line 452) | def eval_class(gt_annos,
  function get_mAP11 (line 573) | def get_mAP11(prec):
  function get_mAP40 (line 580) | def get_mAP40(prec):
  function print_str (line 587) | def print_str(value, *arg, sstream=None):
  function do_eval (line 596) | def do_eval(gt_annos,
  function do_coco_style_eval (line 642) | def do_coco_style_eval(gt_annos, dt_annos, current_classes, overlap_ranges,
  function kitti_eval (line 662) | def kitti_eval(gt_annos,
  function kitti_eval_coco_style (line 881) | def kitti_eval_coco_style(gt_annos, dt_annos, current_classes):

FILE: mmdet3d/core/evaluation/kitti_utils/rotate_iou.py
  function div_up (line 15) | def div_up(m, n):
  function trangle_area (line 20) | def trangle_area(a, b, c):
  function area (line 26) | def area(int_pts, num_of_inter):
  function sort_vertex_in_convex_polygon (line 36) | def sort_vertex_in_convex_polygon(int_pts, num_of_inter):
  function line_segment_intersection (line 76) | def line_segment_intersection(pts1, pts2, i, j, temp_pts):
  function line_segment_intersection_v1 (line 119) | def line_segment_intersection_v1(pts1, pts2, i, j, temp_pts):
  function point_in_quadrilateral (line 158) | def point_in_quadrilateral(pt_x, pt_y, corners):
  function quadrilateral_intersection (line 177) | def quadrilateral_intersection(pts1, pts2, int_pts):
  function rbbox_to_corners (line 201) | def rbbox_to_corners(corners, rbbox):
  function inter (line 227) | def inter(rbbox1, rbbox2):
  function devRotateIoUEval (line 253) | def devRotateIoUEval(rbox1, rbox2, criterion=-1):
  function rotate_iou_kernel_eval (line 283) | def rotate_iou_kernel_eval(N,
  function rotate_iou_gpu_eval (line 337) | def rotate_iou_gpu_eval(boxes, query_boxes, criterion=-1, device_id=0):

FILE: mmdet3d/core/evaluation/lyft_eval.py
  function load_lyft_gts (line 15) | def load_lyft_gts(lyft, data_root, eval_split, logger=None):
  function load_lyft_predictions (line 74) | def load_lyft_predictions(res_path):
  function lyft_eval (line 91) | def lyft_eval(lyft, data_root, res_path, eval_set, output_dir, logger=No...
  function get_classwise_aps (line 143) | def get_classwise_aps(gt, predictions, class_names, iou_thresholds):
  function get_single_class_aps (line 200) | def get_single_class_aps(gt, predictions, iou_thresholds):

FILE: mmdet3d/core/evaluation/scannet_utils/evaluate_semantic_instance.py
  function evaluate_matches (line 10) | def evaluate_matches(matches, class_labels, options):
  function compute_averages (line 189) | def compute_averages(aps, options, class_labels):
  function assign_instances_for_scan (line 221) | def assign_instances_for_scan(pred_info, gt_ids, options, valid_class_ids,
  function scannet_eval (line 297) | def scannet_eval(preds, gts, options, valid_class_ids, class_labels,
  function get_options (line 330) | def get_options(options=None):

FILE: mmdet3d/core/evaluation/scannet_utils/util_3d.py
  class Instance (line 8) | class Instance:
    method __init__ (line 21) | def __init__(self, mesh_vert_instances, instance_id):
    method get_label_id (line 30) | def get_label_id(instance_id):
    method get_instance_verts (line 34) | def get_instance_verts(mesh_vert_instances, instance_id):
    method to_json (line 37) | def to_json(self):
    method to_dict (line 41) | def to_dict(self):
    method from_json (line 50) | def from_json(self, data):
    method __str__ (line 58) | def __str__(self):
  function get_instances (line 62) | def get_instances(ids, class_ids, class_labels, id2label):

FILE: mmdet3d/core/evaluation/seg_eval.py
  function fast_hist (line 7) | def fast_hist(preds, labels, num_classes):
  function per_class_iou (line 28) | def per_class_iou(hist):
  function get_acc (line 42) | def get_acc(hist):
  function get_acc_cls (line 56) | def get_acc_cls(hist):
  function seg_eval (line 70) | def seg_eval(gt_labels, seg_preds, label2cat, ignore_index, logger=None):

FILE: mmdet3d/core/evaluation/waymo_utils/prediction_kitti_to_waymo.py
  class KITTI2Waymo (line 23) | class KITTI2Waymo(object):
    method __init__ (line 41) | def __init__(self,
    method get_file_names (line 79) | def get_file_names(self):
    method create_folder (line 85) | def create_folder(self):
    method parse_objects (line 89) | def parse_objects(self, kitti_result, T_k2w, context_name,
    method convert_one (line 171) | def convert_one(self, file_idx):
    method convert (line 211) | def convert(self):
    method __len__ (line 225) | def __len__(self):
    method transform (line 229) | def transform(self, T, x, y, z):
    method combine (line 245) | def combine(self, pathnames):

FILE: mmdet3d/core/hook/ema.py
  class ModelEMA (line 17) | class ModelEMA:
    method __init__ (line 31) | def __init__(self, model, decay=0.9999, updates=0):
    method update (line 48) | def update(self, trainer, model):
  class MEGVIIEMAHook (line 63) | class MEGVIIEMAHook(Hook):
    method __init__ (line 70) | def __init__(self, init_updates=0, decay=0.9990, resume=None, interval...
    method before_run (line 77) | def before_run(self, runner):
    method after_train_iter (line 100) | def after_train_iter(self, runner):
    method after_train_epoch (line 108) | def after_train_epoch(self, runner):
    method after_run (line 111) | def after_run(self, runner):
    method save_checkpoint (line 115) | def save_checkpoint(self, runner):
    method save_checkpoint_iter (line 128) | def save_checkpoint_iter(self, runner):

FILE: mmdet3d/core/hook/forge_load.py
  class Monitor (line 22) | class Monitor(threading.Thread):
    method __init__ (line 23) | def __init__(self):
    method update (line 30) | def update(self, ):
    method run (line 36) | def run(self):
    method get_current_load (line 42) | def get_current_load():
  class ForgeLoadWorker (line 49) | class ForgeLoadWorker(Hook):
    method __init__ (line 51) | def __init__(self, target=50):
    method after_run (line 59) | def after_run(self, runner):
    method __str__ (line 72) | def __str__(self):
    method my_kernel (line 78) | def my_kernel(io_array):
    method run_awhile (line 87) | def run_awhile(self, sec=10):
    method idle_awhile (line 93) | def idle_awhile(self, sec=5):
    method _boost (line 97) | def _boost(self, rate=1.2):
    method _slow_down (line 101) | def _slow_down(self, rate=1.5):
    method adjust_speed (line 105) | def adjust_speed(self, avg_load):
    method main (line 117) | def main(self, target=50):

FILE: mmdet3d/core/hook/sequentialsontrol.py
  class SequentialControlHook (line 9) | class SequentialControlHook(Hook):
    method __init__ (line 12) | def __init__(self, temporal_start_epoch=1, temporal_start_iter=-1):
    method set_temporal_flag (line 17) | def set_temporal_flag(self, runner, flag):
    method set_temporal_flag_v2 (line 23) | def set_temporal_flag_v2(self, runner, flag):
    method before_run (line 29) | def before_run(self, runner):
    method before_train_epoch (line 34) | def before_train_epoch(self, runner):
    method after_train_iter (line 38) | def after_train_iter(self, runner):

FILE: mmdet3d/core/hook/utils.py
  function is_parallel (line 7) | def is_parallel(model):

FILE: mmdet3d/core/points/__init__.py
  function get_points_type (line 10) | def get_points_type(points_type):

FILE: mmdet3d/core/points/base_points.py
  class BasePoints (line 11) | class BasePoints(object):
    method __init__ (line 30) | def __init__(self, tensor, points_dim=3, attribute_dims=None):
    method coord (line 50) | def coord(self):
    method coord (line 55) | def coord(self, tensor):
    method height (line 66) | def height(self):
    method height (line 76) | def height(self, tensor):
    method color (line 97) | def color(self):
    method color (line 107) | def color(self, tensor):
    method shape (line 131) | def shape(self):
    method shuffle (line 135) | def shuffle(self):
    method rotate (line 145) | def rotate(self, rotation, axis=None):
    method flip (line 174) | def flip(self, bev_direction='horizontal'):
    method translate (line 182) | def translate(self, trans_vector):
    method in_range_3d (line 203) | def in_range_3d(self, point_range):
    method bev (line 228) | def bev(self):
    method in_range_bev (line 232) | def in_range_bev(self, point_range):
    method convert_to (line 250) | def convert_to(self, dst, rt_mat=None):
    method scale (line 268) | def scale(self, scale_factor):
    method __getitem__ (line 276) | def __getitem__(self, item):
    method __len__ (line 348) | def __len__(self):
    method __repr__ (line 352) | def __repr__(self):
    method cat (line 357) | def cat(cls, points_list):
    method to (line 379) | def to(self, device):
    method clone (line 395) | def clone(self):
    method device (line 409) | def device(self):
    method __iter__ (line 413) | def __iter__(self):
    method new_point (line 421) | def new_point(self, data):

FILE: mmdet3d/core/points/cam_points.py
  class CameraPoints (line 5) | class CameraPoints(BasePoints):
    method __init__ (line 24) | def __init__(self, tensor, points_dim=3, attribute_dims=None):
    method flip (line 29) | def flip(self, bev_direction='horizontal'):
    method bev (line 41) | def bev(self):
    method convert_to (line 45) | def convert_to(self, dst, rt_mat=None):

FILE: mmdet3d/core/points/depth_points.py
  class DepthPoints (line 5) | class DepthPoints(BasePoints):
    method __init__ (line 24) | def __init__(self, tensor, points_dim=3, attribute_dims=None):
    method flip (line 29) | def flip(self, bev_direction='horizontal'):
    method convert_to (line 40) | def convert_to(self, dst, rt_mat=None):

FILE: mmdet3d/core/points/lidar_points.py
  class LiDARPoints (line 5) | class LiDARPoints(BasePoints):
    method __init__ (line 24) | def __init__(self, tensor, points_dim=3, attribute_dims=None):
    method flip (line 29) | def flip(self, bev_direction='horizontal'):
    method convert_to (line 40) | def convert_to(self, dst, rt_mat=None):

FILE: mmdet3d/core/post_processing/box3d_nms.py
  function box3d_multiclass_nms (line 8) | def box3d_multiclass_nms(mlvl_bboxes,
  function aligned_3d_nms (line 131) | def aligned_3d_nms(boxes, scores, classes, thresh):
  function circle_nms (line 182) | def circle_nms(dets, thresh, post_max_size=83):
  function nms_bev (line 231) | def nms_bev(boxes, scores, thresh, pre_max_size=None, post_max_size=None,
  function nms_normal_bev (line 276) | def nms_normal_bev(boxes, scores, thresh):

FILE: mmdet3d/core/post_processing/merge_augs.py
  function merge_aug_bboxes_3d (line 8) | def merge_aug_bboxes_3d(aug_results, img_metas, test_cfg):

FILE: mmdet3d/core/utils/array_converter.py
  function array_converter (line 9) | def array_converter(to_torch=True,
  class ArrayConverter (line 202) | class ArrayConverter:
    method __init__ (line 208) | def __init__(self, template_array=None):
    method set_template (line 212) | def set_template(self, array):
    method convert (line 253) | def convert(self, input_array, target_type=None, target_array=None):
    method recover (line 312) | def recover(self, input_array):

FILE: mmdet3d/core/utils/gaussian.py
  function gaussian_2d (line 6) | def gaussian_2d(shape, sigma=1):
  function draw_heatmap_gaussian (line 25) | def draw_heatmap_gaussian(heatmap, center, radius, k=1):
  function gaussian_radius (line 57) | def gaussian_radius(det_size, min_overlap=0.5):
  function get_ellip_gaussian_2D (line 89) | def get_ellip_gaussian_2D(heatmap, center, radius_x, radius_y, k=1):
  function ellip_gaussian2D (line 128) | def ellip_gaussian2D(radius,

FILE: mmdet3d/core/visualizer/image_vis.py
  function project_pts_on_img (line 10) | def project_pts_on_img(points,
  function plot_rect3d_on_img (line 61) | def plot_rect3d_on_img(img,
  function draw_lidar_bbox3d_on_img (line 89) | def draw_lidar_bbox3d_on_img(bboxes3d,
  function draw_depth_bbox3d_on_img (line 128) | def draw_depth_bbox3d_on_img(bboxes3d,
  function draw_camera_bbox3d_on_img (line 168) | def draw_camera_bbox3d_on_img(bboxes3d,

FILE: mmdet3d/core/visualizer/open3d_vis.py
  function _draw_points (line 15) | def _draw_points(points,
  function _draw_bboxes (line 60) | def _draw_bboxes(bbox3d,
  function show_pts_boxes (line 125) | def show_pts_boxes(points,
  function _draw_bboxes_ind (line 191) | def _draw_bboxes_ind(bbox3d,
  function show_pts_index_boxes (line 263) | def show_pts_index_boxes(points,
  class Visualizer (line 333) | class Visualizer(object):
    method __init__ (line 363) | def __init__(self,
    method add_bboxes (line 404) | def add_bboxes(self, bbox3d, bbox_color=None, points_in_box_color=None):
    method add_seg_mask (line 425) | def add_seg_mask(self, seg_mask_colors):
    method show (line 447) | def show(self, save_path=None):

FILE: mmdet3d/core/visualizer/show_result.py
  function _write_obj (line 12) | def _write_obj(points, out_filename):
  function _write_oriented_bbox (line 34) | def _write_oriented_bbox(scene_bbox, out_filename):
  function show_result (line 76) | def show_result(points,
  function show_seg_result (line 147) | def show_seg_result(points,
  function show_multi_modality_result (line 220) | def show_multi_modality_result(img,

FILE: mmdet3d/core/voxel/builder.py
  function build_voxel_generator (line 7) | def build_voxel_generator(cfg, **kwargs):

FILE: mmdet3d/core/voxel/voxel_generator.py
  class VoxelGenerator (line 6) | class VoxelGenerator(object):
    method __init__ (line 17) | def __init__(self,
    method generate (line 36) | def generate(self, points):
    method voxel_size (line 43) | def voxel_size(self):
    method max_num_points_per_voxel (line 48) | def max_num_points_per_voxel(self):
    method point_cloud_range (line 53) | def point_cloud_range(self):
    method grid_size (line 58) | def grid_size(self):
    method __repr__ (line 62) | def __repr__(self):
  function points_to_voxel (line 76) | def points_to_voxel(points,
  function _points_to_voxel_reverse_kernel (line 138) | def _points_to_voxel_reverse_kernel(points,
  function _points_to_voxel_kernel (line 212) | def _points_to_voxel_kernel(points,

FILE: mmdet3d/datasets/builder.py
  function build_dataset (line 33) | def build_dataset(cfg, default_args=None):
  function build_dataloader (line 61) | def build_dataloader(dataset,

FILE: mmdet3d/datasets/custom_3d.py
  class Custom3DDataset (line 24) | class Custom3DDataset(Dataset):
    method __init__ (line 70) | def __init__(self,
    method load_annotations (line 112) | def load_annotations(self, ann_file):
    method get_data_info (line 124) | def get_data_info(self, index):
    method get_ann_info (line 156) | def get_ann_info(self, index):
    method pre_pipeline (line 197) | def pre_pipeline(self, results):
    method prepare_train_data (line 224) | def prepare_train_data(self, index):
    method prepare_test_data (line 245) | def prepare_test_data(self, index):
    method get_classes (line 260) | def get_classes(cls, classes=None):
    method format_results (line 286) | def format_results(self,
    method evaluate (line 310) | def evaluate(self,
    method _build_default_pipeline (line 362) | def _build_default_pipeline(self):
    method _get_pipeline (line 367) | def _get_pipeline(self, pipeline):
    method _extract_data (line 384) | def _extract_data(self, index, pipeline, key, load_annos=False):
    method __len__ (line 418) | def __len__(self):
    method _rand_another (line 426) | def _rand_another(self, idx):
    method __getitem__ (line 435) | def __getitem__(self, idx):
    method _set_group_flag (line 450) | def _set_group_flag(self):

FILE: mmdet3d/datasets/custom_3d_seg.py
  class Custom3DSegDataset (line 18) | class Custom3DSegDataset(Dataset):
    method __init__ (line 56) | def __init__(self,
    method load_annotations (line 100) | def load_annotations(self, ann_file):
    method get_data_info (line 112) | def get_data_info(self, index):
    method pre_pipeline (line 141) | def pre_pipeline(self, results):
    method prepare_train_data (line 160) | def prepare_train_data(self, index):
    method prepare_test_data (line 176) | def prepare_test_data(self, index):
    method get_classes_and_palette (line 190) | def get_classes_and_palette(self, classes=None, palette=None):
    method get_scene_idxs (line 265) | def get_scene_idxs(self, scene_idxs):
    method format_results (line 286) | def format_results(self,
    method evaluate (line 310) | def evaluate(self,
    method _rand_another (line 364) | def _rand_another(self, idx):
    method _build_default_pipeline (line 373) | def _build_default_pipeline(self):
    method _get_pipeline (line 378) | def _get_pipeline(self, pipeline):
    method _extract_data (line 395) | def _extract_data(self, index, pipeline, key, load_annos=False):
    method __len__ (line 429) | def __len__(self):
    method __getitem__ (line 437) | def __getitem__(self, idx):
    method _set_group_flag (line 458) | def _set_group_flag(self):

FILE: mmdet3d/datasets/dataset_wrappers.py
  class CBGSDataset (line 8) | class CBGSDataset(object):
    method __init__ (line 19) | def __init__(self, dataset):
    method _get_sample_indices (line 30) | def _get_sample_indices(self):
    method __getitem__ (line 64) | def __getitem__(self, idx):
    method __len__ (line 73) | def __len__(self):

FILE: mmdet3d/datasets/evals/eval_utils.py
  function category_to_motion_name (line 18) | def category_to_motion_name(category_name: str):
  function detection_prediction_category_to_motion_name (line 47) | def detection_prediction_category_to_motion_name(category_name: str):
  class DetectionMotionMetrics (line 72) | class DetectionMotionMetrics(DetectionMetrics):
    method deserialize (line 76) | def deserialize(cls, content: dict):
  class DetectionMotionMetricDataList (line 93) | class DetectionMotionMetricDataList(DetectionMetricDataList):
    method deserialize (line 96) | def deserialize(cls, content: dict):
  class DetectionMotionMetricData (line 103) | class DetectionMotionMetricData(DetectionMetricData):
    method __init__ (line 108) | def __init__(self,
    method __eq__ (line 151) | def __eq__(self, other):
    method max_recall_ind (line 158) | def max_recall_ind(self):
    method max_recall (line 171) | def max_recall(self):
    method serialize (line 176) | def serialize(self):
    method deserialize (line 193) | def deserialize(cls, content: dict):
    method no_predictions (line 208) | def no_predictions(cls):
    method random_md (line 223) | def random_md(cls):
  class DetectionMotionBox (line 238) | class DetectionMotionBox(DetectionBox):
    method __init__ (line 239) | def __init__(self,
    method __eq__ (line 271) | def __eq__(self, other):
    method serialize (line 285) | def serialize(self) -> dict:
    method deserialize (line 303) | def deserialize(cls, content: dict):
  class DetectionMotionBox_modified (line 319) | class DetectionMotionBox_modified(DetectionMotionBox):
    method __init__ (line 320) | def __init__(self, *args, token=None, visibility=None, index=None, **k...
    method serialize (line 329) | def serialize(self) -> dict:
    method deserialize (line 350) | def deserialize(cls, content: dict):
  function load_prediction (line 371) | def load_prediction(result_path: str, max_boxes_per_sample: int, box_cls...
  function load_gt (line 406) | def load_gt(nusc: NuScenes, eval_split: str, box_cls, data_infos = None,...
  function prediction_metrics (line 537) | def prediction_metrics(gt_box_match, pred_box):
  function accumulate (line 558) | def accumulate(gt_boxes: EvalBoxes,
  function accumulate_motion (line 729) | def accumulate_motion(gt_boxes: EvalBoxes,

FILE: mmdet3d/datasets/evals/map_api.py
  class NuScenesMap (line 42) | class NuScenesMap:
    method __init__ (line 63) | def __init__(self,
    method _load_layer (line 112) | def _load_layer(self, layer_name: str) -> List[dict]:
    method _load_layer_dict (line 120) | def _load_layer_dict(self, layer_name: str) -> Dict[str, Union[dict, l...
    method _load_layers (line 128) | def _load_layers(self) -> None:
    method _make_token2ind (line 151) | def _make_token2ind(self) -> None:
    method _make_shortcuts (line 160) | def _make_shortcuts(self) -> None:
    method _get_stop_line_cue (line 190) | def _get_stop_line_cue(self, stop_line_record: dict) -> List[dict]:
    method get (line 203) | def get(self, layer_name: str, token: str) -> dict:
    method getind (line 214) | def getind(self, layer_name: str, token: str) -> int:
    method render_record (line 223) | def render_record(self,
    method render_layers (line 244) | def render_layers(self,
    method render_map_patch (line 262) | def render_map_patch(self,
    method render_map_in_image (line 285) | def render_map_in_image(self,
    method get_map_mask_in_image (line 319) | def get_map_mask_in_image(self,
    method render_egoposes_on_fancy_map (line 353) | def render_egoposes_on_fancy_map(self,
    method render_centerlines (line 381) | def render_centerlines(self,
    method render_map_mask (line 394) | def render_map_mask(self,
    method get_map_mask (line 415) | def get_map_mask(self,
    method get_map_geom (line 430) | def get_map_geom(self,
    method get_records_in_patch (line 445) | def get_records_in_patch(self,
    method is_record_in_patch (line 460) | def is_record_in_patch(self,
    method layers_on_point (line 476) | def layers_on_point(self, x: float, y: float, layer_names: List[str] =...
    method record_on_point (line 486) | def record_on_point(self, x: float, y: float, layer_name: str) -> str:
    method extract_polygon (line 496) | def extract_polygon(self, polygon_token: str) -> Polygon:
    method extract_line (line 504) | def extract_line(self, line_token: str) -> LineString:
    method get_bounds (line 512) | def get_bounds(self, layer_name: str, token: str) -> Tuple[float, floa...
    method get_records_in_radius (line 521) | def get_records_in_radius(self, x: float, y: float, radius: float,
    method discretize_centerlines (line 538) | def discretize_centerlines(self, resolution_meters: float) -> List[np....
    method discretize_lanes (line 553) | def discretize_lanes(self, tokens: List[str],
    method _get_connected_lanes (line 565) | def _get_connected_lanes(self, lane_token: str, incoming_outgoing: str...
    method get_outgoing_lane_ids (line 578) | def get_outgoing_lane_ids(self, lane_token: str) -> List[str]:
    method get_incoming_lane_ids (line 587) | def get_incoming_lane_ids(self, lane_token: str) -> List[str]:
    method get_arcline_path (line 596) | def get_arcline_path(self, lane_token: str) -> List[ArcLinePath]:
    method get_closest_lane (line 611) | def get_closest_lane(self, x: float, y: float, radius: float = 5) -> str:
    method render_next_roads (line 638) | def render_next_roads(self,
    method get_next_roads (line 654) | def get_next_roads(self, x: float, y: float) -> Dict[str, List[str]]:
  class NuScenesMapExplorer (line 707) | class NuScenesMapExplorer:
    method __init__ (line 709) | def __init__(self,
    method render_centerlines (line 742) | def render_centerlines(self,
    method render_map_mask (line 769) | def render_map_mask(self,
    method get_map_geom (line 813) | def get_map_geom(self,
    method map_geom_to_mask (line 840) | def map_geom_to_mask(self,
    method get_map_mask (line 861) | def get_map_mask(self,
    method render_record (line 911) | def render_record(self,
    method render_layers (line 992) | def render_layers(self,
    method render_map_patch (line 1024) | def render_map_patch(self,
    method render_map_in_image (line 1081) | def render_map_in_image(self,
    method points_transform (line 1239) | def points_transform(points, poserecord, cs_record, cam_intrinsic, im_...
    method get_map_mask_in_image (line 1293) | def get_map_mask_in_image(self,
    method render_egoposes_on_fancy_map (line 1476) | def render_egoposes_on_fancy_map(self,
    method render_next_roads (line 1579) | def render_next_roads(self,
    method _clip_points_behind_camera (line 1611) | def _clip_points_behind_camera(points, near_plane: float):
    method get_records_in_patch (line 1674) | def get_records_in_patch(self,
    method is_record_in_patch (line 1705) | def is_record_in_patch(self,
    method layers_on_point (line 1729) | def layers_on_point(self, x: float, y: float, layer_names: List[str] =...
    method record_on_point (line 1747) | def record_on_point(self, x: float, y: float, layer_name: str) -> str:
    method extract_polygon (line 1780) | def extract_polygon(self, polygon_token: str) -> Polygon:
    method extract_line (line 1800) | def extract_line(self, line_token: str) -> LineString:
    method get_bounds (line 1812) | def get_bounds(self, layer_name: str, token: str) -> Tuple[float, floa...
    method _get_polygon_bounds (line 1826) | def _get_polygon_bounds(self, layer_name: str, token: str) -> Tuple[fl...
    method _get_line_bounds (line 1862) | def _get_line_bounds(self, layer_name: str, token: str) -> Tuple[float...
    method _is_polygon_record_in_patch (line 1887) | def _is_polygon_record_in_patch(self,
    method _is_line_record_in_patch (line 1919) | def _is_line_record_in_patch(self,
    method _render_layer (line 1956) | def _render_layer(self, ax: Axes, layer_name: str, alpha: float, token...
    method _render_polygon_layer (line 1971) | def _render_polygon_layer(self, ax: Axes, layer_name: str, alpha: floa...
    method _render_line_layer (line 2011) | def _render_line_layer(self, ax: Axes, layer_name: str, alpha: float, ...
    method _get_layer_geom (line 2045) | def _get_layer_geom(self,
    method _layer_geom_to_mask (line 2063) | def _layer_geom_to_mask(self,
    method mask_for_polygons (line 2084) | def mask_for_polygons(polygons: MultiPolygon, mask: np.ndarray) -> np....
    method mask_for_lines (line 2104) | def mask_for_lines(lines: LineString, mask: np.ndarray) -> np.ndarray:
    method _polygon_geom_to_mask (line 2123) | def _polygon_geom_to_mask(self,
    method _line_geom_to_mask (line 2177) | def _line_geom_to_mask(self,
    method _get_layer_polygon (line 2221) | def _get_layer_polygon(self,
    method _get_layer_line (line 2275) | def _get_layer_line(self,
    method get_patch_coord (line 2315) | def get_patch_coord(patch_box: Tuple[float, float, float, float],
    method _get_figsize (line 2335) | def _get_figsize(self, figsize: Union[None, float, Tuple[float, float]...

FILE: mmdet3d/datasets/evals/metric_utils.py
  function min_ade (line 6) | def min_ade(traj: torch.Tensor, traj_gt: torch.Tensor,
  function min_fde (line 33) | def min_fde(traj: torch.Tensor, traj_gt: torch.Tensor,
  function miss_rate (line 64) | def miss_rate(
  function traj_fde (line 96) | def traj_fde(gt_box, pred_box, final_step):

FILE: mmdet3d/datasets/evals/nuscenes_eval_motion.py
  function class_tp_curve (line 78) | def class_tp_curve(md_list: DetectionMetricDataList,
  function center_in_image (line 152) | def center_in_image(box,
  function exist_corners_in_image_but_not_all (line 188) | def exist_corners_in_image_but_not_all(box,
  function filter_eval_boxes_by_id (line 220) | def filter_eval_boxes_by_id(nusc: NuScenes,
  function filter_eval_boxes_by_visibility (line 252) | def filter_eval_boxes_by_visibility(
  function filter_by_sample_token (line 284) | def filter_by_sample_token(
  function filter_eval_boxes_by_overlap (line 295) | def filter_eval_boxes_by_overlap(nusc: NuScenes,
  class MotionEval (line 379) | class MotionEval(NuScenesEval):
    method __init__ (line 384) | def __init__(self,
    method update_gt (line 483) | def update_gt(self, type_='vis', visibility='1', index=1):
    method evaluate (line 516) | def evaluate(self) -> Tuple[DetectionMotionMetrics,
    method evaluate_motion (line 582) | def evaluate_motion(
    method evaluate_epa (line 647) | def evaluate_epa(
    method main (line 717) | def main(self,
    method render (line 801) | def render(self, metrics: DetectionMetrics,
  function print_traj_metrics (line 855) | def print_traj_metrics(metrics):

FILE: mmdet3d/datasets/evaluation/AP.py
  function average_precision (line 7) | def average_precision(recalls, precisions, mode='area'):
  function instance_match (line 52) | def instance_match(pred_lines: NDArray,

FILE: mmdet3d/datasets/evaluation/distance.py
  function chamfer_distance (line 5) | def chamfer_distance(line1: NDArray, line2: NDArray) -> float:
  function frechet_distance (line 23) | def frechet_distance(line1: NDArray, line2: NDArray) -> float:
  function chamfer_distance_batch (line 37) | def chamfer_distance_batch(pred_lines, gt_lines):

FILE: mmdet3d/datasets/evaluation/raster_eval.py
  class RasterEvaluate (line 14) | class RasterEvaluate(object):
    method __init__ (line 22) | def __init__(self, dataset_cfg: Config, n_workers: int=N_WORKERS):
    method gts (line 31) | def gts(self) -> Dict[str, NDArray]:
    method evaluate (line 42) | def evaluate(self,

FILE: mmdet3d/datasets/evaluation/vector_eval.py
  class VectorEvaluate (line 24) | class VectorEvaluate(object):
    method __init__ (line 32) | def __init__(self, dataset_cfg: Config, n_workers: int=N_WORKERS) -> N...
    method gts (line 45) | def gts(self) -> Dict[str, Dict[int, List[NDArray]]]:
    method interp_fixed_num (line 77) | def interp_fixed_num(self,
    method interp_fixed_dist (line 96) | def interp_fixed_dist(self,
    method _evaluate_single (line 118) | def _evaluate_single(self,
    method evaluate (line 169) | def evaluate(self,

FILE: mmdet3d/datasets/kitti2d_dataset.py
  class Kitti2DDataset (line 10) | class Kitti2DDataset(CustomDataset):
    method load_annotations (line 81) | def load_annotations(self, ann_file):
    method _filter_imgs (line 97) | def _filter_imgs(self, min_size=32):
    method get_ann_info (line 105) | def get_ann_info(self, index):
    method prepare_train_img (line 137) | def prepare_train_img(self, idx):
    method prepare_test_img (line 158) | def prepare_test_img(self, idx):
    method drop_arrays_by_name (line 176) | def drop_arrays_by_name(self, gt_names, used_classes):
    method keep_arrays_by_name (line 190) | def keep_arrays_by_name(self, gt_names, used_classes):
    method reformat_bbox (line 204) | def reformat_bbox(self, outputs, out=None):
    method evaluate (line 222) | def evaluate(self, result_files, eval_types=None):

FILE: mmdet3d/datasets/kitti_dataset.py
  class KittiDataset (line 21) | class KittiDataset(Custom3DDataset):
    method __init__ (line 57) | def __init__(self,
    method _get_pts_filename (line 87) | def _get_pts_filename(self, idx):
    method get_data_info (line 100) | def get_data_info(self, index):
    method get_ann_info (line 143) | def get_ann_info(self, index):
    method drop_arrays_by_name (line 223) | def drop_arrays_by_name(self, gt_names, used_classes):
    method keep_arrays_by_name (line 237) | def keep_arrays_by_name(self, gt_names, used_classes):
    method remove_dontcare (line 251) | def remove_dontcare(self, ann_info):
    method format_results (line 270) | def format_results(self,
    method evaluate (line 325) | def evaluate(self,
    method bbox2result_kitti (line 394) | def bbox2result_kitti(self,
    method bbox2result_kitti2d (line 509) | def bbox2result_kitti2d(self,
    method convert_valid_bboxes (line 621) | def convert_valid_bboxes(self, box_dict, info):
    method _build_default_pipeline (line 703) | def _build_default_pipeline(self):
    method show (line 722) | def show(self, results, out_dir, show=True, pipeline=None):

FILE: mmdet3d/datasets/kitti_mono_dataset.py
  class KittiMonoDataset (line 17) | class KittiMonoDataset(NuScenesMonoDataset):
    method __init__ (line 35) | def __init__(self,
    method _parse_ann_info (line 57) | def _parse_ann_info(self, img_info, ann_info):
    method format_results (line 147) | def format_results(self,
    method evaluate (line 203) | def evaluate(self,
    method bbox2result_kitti (line 271) | def bbox2result_kitti(self,
    method bbox2result_kitti2d (line 386) | def bbox2result_kitti2d(self,
    method convert_valid_bboxes (line 497) | def convert_valid_bboxes(self, box_dict, info):

FILE: mmdet3d/datasets/lyft_dataset.py
  class LyftDataset (line 22) | class LyftDataset(Custom3DDataset):
    method __init__ (line 80) | def __init__(self,
    method load_annotations (line 112) | def load_annotations(self, ann_file):
    method get_data_info (line 129) | def get_data_info(self, index):
    method get_ann_info (line 188) | def get_ann_info(self, index):
    method _format_bbox (line 230) | def _format_bbox(self, results, jsonfile_prefix=None):
    method _evaluate_single (line 273) | def _evaluate_single(self,
    method format_results (line 315) | def format_results(self, results, jsonfile_prefix=None, csv_savepath=N...
    method evaluate (line 366) | def evaluate(self,
    method _build_default_pipeline (line 422) | def _build_default_pipeline(self):
    method show (line 443) | def show(self, results, out_dir, show=False, pipeline=None):
    method json2csv (line 475) | def json2csv(self, json_path, csv_savepath):
  function output_to_lyft_box (line 513) | def output_to_lyft_box(detection):
  function lidar_lyft_box_to_global (line 546) | def lidar_lyft_box_to_global(info, boxes):

FILE: mmdet3d/datasets/map_utils/mean_ap.py
  function average_precision (line 14) | def average_precision(recalls, precisions, mode='area'):
  function get_cls_results (line 60) | def get_cls_results(gen_results,
  function format_res_gt_by_classes (line 137) | def format_res_gt_by_classes(result_path,
  function eval_map (line 220) | def eval_map(gen_results,
  function print_map_summary (line 331) | def print_map_summary(mean_ap,

FILE: mmdet3d/datasets/map_utils/tpfp.py
  function tpfp_bbox (line 9) | def tpfp_bbox(det_bboxes,
  function tpfp_rbbox (line 79) | def tpfp_rbbox(det_bboxes,
  function tpfp_det (line 149) | def tpfp_det(det_bboxes,
  function tpfp_gen (line 216) | def tpfp_gen(gen_lines,
  function custom_tpfp_gen (line 290) | def custom_tpfp_gen(gen_lines,

FILE: mmdet3d/datasets/map_utils/tpfp_chamfer.py
  function vec_iou (line 15) | def vec_iou(pred_lines, gt_lines):
  function convex_iou (line 52) | def convex_iou(pred_lines, gt_lines, gt_mask):
  function rbbox_iou (line 86) | def rbbox_iou(pred_lines, gt_lines, gt_mask):
  function polyline_score (line 121) | def polyline_score(pred_lines, gt_lines, linewidth=1., metric='POR'):
  function custom_polyline_score (line 217) | def custom_polyline_score(pred_lines, gt_lines, linewidth=1., metric='ch...

FILE: mmdet3d/datasets/nuscenes_dataset.py
  class NuScenesDataset (line 45) | class NuScenesDataset(Custom3DDataset):
    method __init__ (line 156) | def __init__(self,
    method get_cat_ids (line 232) | def get_cat_ids(self, idx):
    method load_annotations (line 256) | def load_annotations(self, ann_file):
    method _set_sequence_group_flag (line 276) | def _set_sequence_group_flag(self):
    method get_data_info (line 311) | def get_data_info(self, index):
    method get_fut_bbox_info (line 447) | def get_fut_bbox_info(self, info, index):
    method get_adj_info (line 463) | def get_adj_info(self, info, index):
    method get_ann_info (line 476) | def get_ann_info(self, index):
    method format_map_results (line 526) | def format_map_results(self, results, jsonfile_prefix=None):
    method get_map_classes (line 574) | def get_map_classes(cls, map_classes=None):
    method _format_map (line 600) | def _format_map(self, results, jsonfile_prefix=None, score_thresh=0.2):
    method vectormap_pipeline (line 661) | def vectormap_pipeline(self, location, ego2global_translation, patch_a...
    method _format_map_gt (line 700) | def _format_map_gt(self):
    method _evaluate_single (line 760) | def _evaluate_single(self,
    method format_results (line 819) | def format_results(self, results, jsonfile_prefix=None):
    method evaluate (line 909) | def evaluate(self, results,
    method evaluate_ego_traj (line 968) | def evaluate_ego_traj(self, results, jsonfile_prefix=None, logger=None):
    method smoothness (line 1083) | def smoothness(self, data):
    method _format_bbox (line 1161) | def _format_bbox(self, results, jsonfile_prefix=None):
    method evaluate_tracking (line 1251) | def evaluate_tracking(self,
    method format_tracking_results (line 1307) | def format_tracking_results(self, results, jsonfile_prefix=None):
    method _format_tracking_bbox (line 1353) | def _format_tracking_bbox(self, results, jsonfile_prefix=None):
    method _evaluate_motion_single (line 1463) | def _evaluate_motion_single(self,
    method _evaluate_tracking_single (line 1549) | def _evaluate_tracking_single(self,
    method evaluate_occupancy (line 1601) | def evaluate_occupancy(self, occ_results, runner=None, show_dir=None, ...
    method evaluate_mask (line 1675) | def evaluate_mask(self, results):
    method evaluate_map (line 1686) | def evaluate_map(self,
    method world2bev_vis (line 1735) | def world2bev_vis(self, x, y):
    method __map_visual__ (line 1738) | def __map_visual__(self, gt_map, pred_map, index=0):
    method _evaluate_map_single (line 1764) | def _evaluate_map_single(self,
    method evaluate_bbox (line 1848) | def evaluate_bbox(self,
    method _build_default_pipeline (line 1898) | def _build_default_pipeline(self):
    method show (line 1919) | def show(self, results, out_dir, show=False, pipeline=None):
  function output_to_nusc_box (line 1953) | def output_to_nusc_box(detection, with_velocity=True):
  class NuscenesOccupancy (line 1999) | class NuscenesOccupancy(NuScenesDataset):
    method __init__ (line 2021) | def __init__(self, occupancy_info='data/nuscenes/occupancy_category.js...
    method get_cat_ids (line 2046) | def get_cat_ids(self, idx):
  function lidar_nusc_box_to_global (line 2069) | def lidar_nusc_box_to_global(info,
  function invert_matrix_egopose_numpy (line 2108) | def invert_matrix_egopose_numpy(egopose):
  function convert_egopose_to_matrix_numpy (line 2118) | def convert_egopose_to_matrix_numpy(rotation, translation):
  function output_to_vecs (line 2126) | def output_to_vecs(detection):

FILE: mmdet3d/datasets/nuscenes_eval.py
  function class_tp_curve (line 74) | def class_tp_curve(md_list: DetectionMetricDataList,
  class DetectionBox_modified (line 137) | class DetectionBox_modified(DetectionBox):
    method __init__ (line 138) | def __init__(self, *args, token=None, visibility=None, index=None, **k...
    method serialize (line 147) | def serialize(self) -> dict:
    method deserialize (line 167) | def deserialize(cls, content: dict):
  function center_in_image (line 187) | def center_in_image(box, intrinsic: np.ndarray, imsize: Tuple[int, int],...
  function exist_corners_in_image_but_not_all (line 217) | def exist_corners_in_image_but_not_all(box, intrinsic: np.ndarray, imsiz...
  function load_gt (line 244) | def load_gt(nusc: NuScenes, eval_split: str, box_cls, verbose: bool = Fa...
  function filter_eval_boxes_by_id (line 362) | def filter_eval_boxes_by_id(nusc: NuScenes,
  function filter_eval_boxes_by_visibility (line 394) | def filter_eval_boxes_by_visibility(
  function filter_by_sample_token (line 426) | def filter_by_sample_token(ori_eval_boxes, valid_sample_tokens=[],  verb...
  function filter_eval_boxes_by_overlap (line 434) | def filter_eval_boxes_by_overlap(nusc: NuScenes,
  function filter_eval_boxes_by_range (line 509) | def filter_eval_boxes_by_range(nusc: NuScenes,
  class NuScenesEval_custom (line 569) | class NuScenesEval_custom(NuScenesEval):
    method __init__ (line 574) | def __init__(self,
    method update_gt (line 662) | def update_gt(self, type_='vis', visibility='1', index=1):
    method evaluate (line 691) | def evaluate(self) -> Tuple[DetectionMetrics, DetectionMetricDataList]:
    method render (line 741) | def render(self, metrics: DetectionMetrics, md_list: DetectionMetricDa...
    method evaluate_mask (line 767) | def evaluate_mask(self, preds, HDMap, Bbox_mask):

FILE: mmdet3d/datasets/nuscenes_mono_dataset.py
  class NuScenesMonoDataset (line 23) | class NuScenesMonoDataset(CocoDataset):
    method __init__ (line 78) | def __init__(self,
    method pre_pipeline (line 148) | def pre_pipeline(self, results):
    method _parse_ann_info (line 177) | def _parse_ann_info(self, img_info, ann_info):
    method get_attr_name (line 276) | def get_attr_name(self, attr_idx, label_name):
    method _format_bbox (line 323) | def _format_bbox(self, results, jsonfile_prefix=None):
    method _evaluate_single (line 430) | def _evaluate_single(self,
    method format_results (line 488) | def format_results(self, results, jsonfile_prefix=None, **kwargs):
    method evaluate (line 537) | def evaluate(self,
    method _extract_data (line 588) | def _extract_data(self, index, pipeline, key, load_annos=False):
    method _get_pipeline (line 621) | def _get_pipeline(self, pipeline):
    method _build_default_pipeline (line 638) | def _build_default_pipeline(self):
    method show (line 650) | def show(self, results, out_dir, show=False, pipeline=None):
  function output_to_nusc_box (line 686) | def output_to_nusc_box(detection):
  function cam_nusc_box_to_global (line 732) | def cam_nusc_box_to_global(info,
  function global_nusc_box_to_cam (line 773) | def global_nusc_box_to_cam(info,
  function nusc_box_to_cam_box3d (line 812) | def nusc_box_to_cam_box3d(boxes):

FILE: mmdet3d/datasets/occ_metrics.py
  function pcolor (line 20) | def pcolor(string, color, on_color=None, attrs=None):
  function getCellCoordinates (line 41) | def getCellCoordinates(points, voxelSize):
  function getNumUniqueCells (line 45) | def getNumUniqueCells(cells):
  class Metric_mIoU (line 50) | class Metric_mIoU():
    method __init__ (line 51) | def __init__(self,
    method hist_info (line 80) | def hist_info(self, n_cl, pred, gt):
    method per_class_iu (line 107) | def per_class_iu(self, hist):
    method compute_mIoU (line 111) | def compute_mIoU(self, pred, label, n_classes):
    method add_batch (line 122) | def add_batch(self,semantics_pred,semantics_gt,mask_lidar,mask_camera):
    method count_miou (line 166) | def count_miou(self):
  class Metric_FScore (line 182) | class Metric_FScore():
    method __init__ (line 183) | def __init__(self,
    method voxel2points (line 210) | def voxel2points(self, voxel):
    method add_batch (line 222) | def add_batch(self,semantics_pred,semantics_gt,mask_lidar,mask_camera ):
    method count_fscore (line 277) | def count_fscore(self,):
  function parse_args (line 291) | def parse_args():
  function eval (line 305) | def eval(args):

FILE: mmdet3d/datasets/occupancy_eval.py
  function parse_args (line 7) | def parse_args():
  function eval (line 19) | def eval(args):

FILE: mmdet3d/datasets/pipelines/compose.py
  class Compose (line 11) | class Compose:
    method __init__ (line 22) | def __init__(self, transforms):
    method __call__ (line 38) | def __call__(self, data):
    method __repr__ (line 54) | def __repr__(self):

FILE: mmdet3d/datasets/pipelines/data_augment_utils.py
  function _rotation_box2d_jit_ (line 14) | def _rotation_box2d_jit_(corners, angle, rot_mat_T):
  function box_collision_test (line 32) | def box_collision_test(boxes, qboxes, clockwise=True):
  function noise_per_box (line 129) | def noise_per_box(boxes, valid_mask, loc_noises, rot_noises):
  function noise_per_box_v2_ (line 170) | def noise_per_box_v2_(boxes, valid_mask, loc_noises, rot_noises,
  function _select_transform (line 236) | def _select_transform(transform, indices):
  function _rotation_matrix_3d_ (line 255) | def _rotation_matrix_3d_(rot_mat_T, angle, axis):
  function points_transform_ (line 284) | def points_transform_(points, centers, point_masks, loc_transform,
  function box3d_transform_ (line 314) | def box3d_transform_(boxes, loc_transform, rot_transform, valid_mask):
  function noise_per_object_v3_ (line 330) | def noise_per_object_v3_(gt_boxes,

FILE: mmdet3d/datasets/pipelines/dbsampler.py
  class BatchSampler (line 14) | class BatchSampler:
    method __init__ (line 25) | def __init__(self,
    method _sample (line 43) | def _sample(self, num):
    method _reset (line 60) | def _reset(self):
    method sample (line 68) | def sample(self, num):
  class DataBaseSampler (line 82) | class DataBaseSampler(object):
    method __init__ (line 98) | def __init__(self,
    method filter_by_difficulty (line 177) | def filter_by_difficulty(db_infos, removed_difficulty):
    method filter_by_min_points (line 198) | def filter_by_min_points(db_infos, min_gt_points_dict):
    method sample_all (line 219) | def sample_all(self, gt_bboxes, gt_labels, img=None, ground_plane=None):
    method sample_class_v2 (line 317) | def sample_class_v2(self, name, num, gt_bboxes):

FILE: mmdet3d/datasets/pipelines/formating.py
  class DefaultFormatBundle (line 12) | class DefaultFormatBundle(object):
    method __init__ (line 29) | def __init__(self, ):
    method __call__ (line 32) | def __call__(self, results):
    method __repr__ (line 103) | def __repr__(self):
  class Collect3D (line 108) | class Collect3D(object):
    method __init__ (line 156) | def __init__(
    method __call__ (line 172) | def __call__(self, results):
    method __repr__ (line 196) | def __repr__(self):
  class DefaultFormatBundle3D (line 203) | class DefaultFormatBundle3D(DefaultFormatBundle):
    method __init__ (line 218) | def __init__(self, class_names, with_gt=True, with_label=True):
    method __call__ (line 224) | def __call__(self, results):
    method __repr__ (line 290) | def __repr__(self):

FILE: mmdet3d/datasets/pipelines/loading.py
  class LoadVectorMap (line 37) | class LoadVectorMap(object):
    method __init__ (line 39) | def __init__(self, data_root, point_cloud_range, map_fixed_ptsnum_per_...
    method vectormap_pipeline (line 47) | def vectormap_pipeline(self, location, ego2global_translation, patch_a...
    method __call__ (line 103) | def __call__(self, results):
  class LoadVectorMap2 (line 122) | class LoadVectorMap2(object):
    method __init__ (line 124) | def __init__(self, data_root, point_cloud_range, map_fixed_ptsnum_per_...
    method vectormap_pipeline (line 133) | def vectormap_pipeline(self, location, ego2global_translation, patch_a...
    method __call__ (line 191) | def __call__(self, results):
  class LoadGTPlaner (line 210) | class LoadGTPlaner(object):
    method __init__ (line 211) | def __init__(self):
    method __call__ (line 214) | def __call__(self, results):
  class LoadGTMotion (line 236) | class LoadGTMotion(object):
    method __init__ (line 237) | def __init__(self, with_ego_as_agent=False):
    method __call__ (line 240) | def __call__(self, results):
  class LoadFutBoxInfo (line 270) | class LoadFutBoxInfo(object):
    method __init__ (line 271) | def __init__(self, add_boundary=True):
    method gen_dx_bx (line 300) | def gen_dx_bx(self, xbound, ybound, zbound):
    method calculate_birds_eye_view_parameters (line 307) | def calculate_birds_eye_view_parameters(self, x_bounds, y_bounds, z_bo...
    method get_label (line 328) | def get_label(
    method world2bev_vis (line 340) | def world2bev_vis(self, x, y):
    method get_birds_eye_view_label (line 343) | def get_birds_eye_view_label(self, boxes_in_cur_ego_list, labels_in_cu...
    method __call__ (line 361) | def __call__(self, results):
  class LoadSemanticImageMask (line 423) | class LoadSemanticImageMask(object):
    method __init__ (line 424) | def __init__(self, mask_file_path='./data/nus_sem'):
    method __call__ (line 427) | def __call__(self, results):
    method img_transform_core (line 444) | def img_transform_core(self, img, resize_dims, crop, flip, rotate):
  class LoadMultiViewImageFromFiles (line 455) | class LoadMultiViewImageFromFiles(object):
    method __init__ (line 467) | def __init__(self, to_float32=False, color_type='unchanged'):
    method __call__ (line 471) | def __call__(self, results):
    method __repr__ (line 511) | def __repr__(self):
  class LoadImageFromFileMono3D (line 520) | class LoadImageFromFileMono3D(object):
    method __call__ (line 530) | def __call__(self, results):
  class LoadOccupancy (line 544) | class LoadOccupancy(object):
    method __init__ (line 553) | def __init__(self, occupancy_path='/mount/dnn_data/occupancy_2023/gts',
    method __call__ (line 569) | def __call__(self, results):
  class LoadPointsFromMultiSweeps (line 627) | class LoadPointsFromMultiSweeps(object):
    method __init__ (line 653) | def __init__(self,
    method _load_points (line 678) | def _load_points(self, pts_filename):
    method _remove_close (line 700) | def _remove_close(self, points, radius=1.0):
    method __call__ (line 722) | def __call__(self, results):
    method __repr__ (line 781) | def __repr__(self):
  class PointsFromLidartoEgo (line 787) | class PointsFromLidartoEgo(object):
    method __init__ (line 789) | def __init__(self, translate2ego=True, ego_cam='CAM_FRONT'):
    method __call__ (line 793) | def __call__(self, results):
  class PointSegClassMapping (line 826) | class PointSegClassMapping(object):
    method __init__ (line 838) | def __init__(self, valid_cat_ids, max_cat_id=40):
    method __call__ (line 852) | def __call__(self, results):
    method __repr__ (line 872) | def __repr__(self):
  class NormalizePointsColor (line 881) | class NormalizePointsColor(object):
    method __init__ (line 888) | def __init__(self, color_mean):
    method __call__ (line 891) | def __call__(self, results):
    method __repr__ (line 914) | def __repr__(self):
  class LoadPointsFromFile (line 922) | class LoadPointsFromFile(object):
    method __init__ (line 948) | def __init__(self,
    method _load_points (line 979) | def _load_points(self, pts_filename):
    method __call__ (line 1003) | def __call__(self, results):
    method __repr__ (line 1058) | def __repr__(self):
  class LoadPointsFromDict (line 1072) | class LoadPointsFromDict(LoadPointsFromFile):
    method __call__ (line 1075) | def __call__(self, results):
  class LoadAnnotations3D (line 1081) | class LoadAnnotations3D(LoadAnnotations):
    method __init__ (line 1117) | def __init__(self,
    method _load_bboxes_3d (line 1146) | def _load_bboxes_3d(self, results):
    method _load_bboxes_depth (line 1159) | def _load_bboxes_depth(self, results):
    method _load_labels_3d (line 1172) | def _load_labels_3d(self, results):
    method _load_attr_labels (line 1184) | def _load_attr_labels(self, results):
    method _load_masks_3d (line 1196) | def _load_masks_3d(self, results):
    method _load_semantic_seg_3d (line 1221) | def _load_semantic_seg_3d(self, results):
    method __call__ (line 1248) | def __call__(self, results):
    method __repr__ (line 1278) | def __repr__(self):
  class PointToMultiViewDepth (line 1297) | class PointToMultiViewDepth(object):
    method __init__ (line 1299) | def __init__(self, grid_config, downsample=1):
    method points2depthmap (line 1303) | def points2depthmap(self, points, height, width):
    method __call__ (line 1329) | def __call__(self, results):
  function mmlabNormalize (line 1389) | def mmlabNormalize(img, mean=[123.675, 116.28, 103.53], std=[58.395, 57....
  class PrepareImageInputs (line 1404) | class PrepareImageInputs(object):
    method __init__ (line 1415) | def __init__(
    method get_rot (line 1434) | def get_rot(self, h):
    method img_transform (line 1441) | def img_transform(self, img, post_rot, post_tran, resize, resize_dims,
    method img_transform_core (line 1462) | def img_transform_core(self, img, resize_dims, crop, flip, rotate):
    method choose_cams (line 1471) | def choose_cams(self):
    method sample_augmentation (line 1482) | def sample_augmentation(self, H, W, flip=None, scale=None):
    method get_sensor2ego_transformation (line 1512) | def get_sensor2ego_transformation(self,
    method get_sensor_transforms (line 1584) | def get_sensor_transforms(self, cam_info, cam_name):
    method get_inputs (line 1607) | def get_inputs(self, results, scale=None):
    method __call__ (line 1744) | def __call__(self, results):
  class LoadAnnotationsBEVDepth (line 1750) | class LoadAnnotationsBEVDepth(object):
    method __init__ (line 1753) | def __init__(self, bda_aug_conf, classes, with_2d_bbox=False, with_ego...
    method sample_bda_augmentation (line 1761) | def sample_bda_augmentation(self, tta_config=None):
    method bev_transform (line 1783) | def bev_transform(self, gt_boxes, rotate_angle, scale_ratio, flip_dx,
    method _bboxes_transform (line 1815) | def _bboxes_transform(self, bboxes, centers2d, gt_labels, depths, resi...
    method _filter_invisible (line 1851) | def _filter_invisible(self, bboxes, centers2d, gt_labels, depths, fH, ...
    method __call__ (line 1877) | def __call__(self, results):

FILE: mmdet3d/datasets/pipelines/test_time_aug.py
  class MultiScaleFlipAug (line 12) | class MultiScaleFlipAug:
    method __init__ (line 49) | def __init__(self,
    method __call__ (line 80) | def __call__(self, results):
    method __repr__ (line 110) | def __repr__(self):
  class MultiScaleFlipAug3D (line 119) | class MultiScaleFlipAug3D(object):
    method __init__ (line 142) | def __init__(self,
    method __call__ (line 175) | def __call__(self, results):
    method __repr__ (line 223) | def __repr__(self):
  class CustomMultiScaleFlipAug3D (line 233) | class CustomMultiScaleFlipAug3D(object):
    method __init__ (line 256) | def __init__(self,
    method __call__ (line 266) | def __call__(self, results):
    method __repr__ (line 311) | def __repr__(self):
  class CustomDistMultiScaleFlipAug3D (line 319) | class CustomDistMultiScaleFlipAug3D(object):
    method __init__ (line 342) | def __init__(self,
    method __call__ (line 348) | def __call__(self, results):
    method __repr__ (line 395) | def __repr__(self):

FILE: mmdet3d/datasets/pipelines/transforms_3d.py
  class VisualInputsAndGT (line 28) | class VisualInputsAndGT(object):
    method __init__ (line 32) | def __init__(self, max=20):
    method _draw_point_cloud_ (line 35) | def _draw_point_cloud_(self, point_cloud, img_size):
    method world2bev_vis (line 63) | def world2bev_vis(self, x, y):
    method world2bev_vis2 (line 66) | def world2bev_vis2(self, x, y):
    method __visual__ (line 70) | def __visual__(self, results):
    method _render_traj (line 192) | def _render_traj(self, future_traj, points_per_step=10):
    method __call__ (line 203) | def __call__(self, results):
  class GridMask (line 213) | class GridMask:
    method __init__ (line 214) | def __init__(
    method __call__ (line 232) | def __call__(self, results):
  class AugPoints (line 288) | class AugPoints(object):
    method __call__ (line 289) | def __call__(self, results):
  class ToEgo (line 308) | class ToEgo(object):
    method __init__ (line 309) | def __init__(self, ego_cam='CAM_FRONT',):
    method __call__ (line 312) | def __call__(self, results):
  class PadMultiViewImage (line 343) | class PadMultiViewImage(object):
    method __init__ (line 354) | def __init__(self, size=None, size_divisor=None, pad_val=0):
    method _pad_img (line 362) | def _pad_img(self, results):
    method __call__ (line 386) | def __call__(self, results):
    method __repr__ (line 398) | def __repr__(self):
  class RandomDropPointsColor (line 408) | class RandomDropPointsColor(object):
    method __init__ (line 420) | def __init__(self, drop_ratio=0.2):
    method __call__ (line 425) | def __call__(self, input_dict):
    method __repr__ (line 450) | def __repr__(self):
  class RandomFlip3D (line 458) | class RandomFlip3D(RandomFlip):
    method __init__ (line 476) | def __init__(self,
    method random_flip_data_3d (line 494) | def random_flip_data_3d(self, input_dict, direction='horizontal'):
    method __call__ (line 535) | def __call__(self, input_dict):
    method __repr__ (line 574) | def __repr__(self):
  class MultiViewWrapper (line 583) | class MultiViewWrapper(object):
    method __init__ (line 603) | def __init__(self,
    method __call__ (line 611) | def __call__(self, input_dict):
  class RangeLimitedRandomCrop (line 629) | class RangeLimitedRandomCrop(RandomCrop):
    method __init__ (line 639) | def __init__(self,
    method _crop_data (line 649) | def _crop_data(self, results, crop_size, allow_negative_crop):
  class RandomRotate (line 722) | class RandomRotate(Rotate):
    method __init__ (line 733) | def __init__(self, range, **kwargs):
    method __call__ (line 737) | def __call__(self, results):
  class RandomJitterPoints (line 745) | class RandomJitterPoints(object):
    method __init__ (line 766) | def __init__(self,
    method __call__ (line 783) | def __call__(self, input_dict):
    method __repr__ (line 804) | def __repr__(self):
  class ObjectSample (line 813) | class ObjectSample(object):
    method __init__ (line 825) | def __init__(self, db_sampler, sample_2d=False, use_ground_plane=False):
    method remove_points_in_boxes (line 834) | def remove_points_in_boxes(points, boxes):
    method __call__ (line 848) | def __call__(self, input_dict):
    method __repr__ (line 918) | def __repr__(self):
  class ObjectNoise (line 932) | class ObjectNoise(object):
    method __init__ (line 947) | def __init__(self,
    method __call__ (line 957) | def __call__(self, input_dict):
    method __repr__ (line 986) | def __repr__(self):
  class GlobalAlignment (line 997) | class GlobalAlignment(object):
    method __init__ (line 1011) | def __init__(self, rotation_axis):
    method _trans_points (line 1014) | def _trans_points(self, input_dict, trans_factor):
    method _rot_points (line 1026) | def _rot_points(self, input_dict, rot_mat):
    method _check_rot_mat (line 1039) | def _check_rot_mat(self, rot_mat):
    method __call__ (line 1052) | def __call__(self, input_dict):
    method __repr__ (line 1077) | def __repr__(self):
  class GlobalRotScaleTrans (line 1084) | class GlobalRotScaleTrans(object):
    method __init__ (line 1101) | def __init__(self,
    method _trans_bbox_points (line 1128) | def _trans_bbox_points(self, input_dict):
    method _rot_bbox_points (line 1147) | def _rot_bbox_points(self, input_dict):
    method _scale_bbox_points (line 1177) | def _scale_bbox_points(self, input_dict):
    method _random_scale (line 1199) | def _random_scale(self, input_dict):
    method __call__ (line 1213) | def __call__(self, input_dict):
    method __repr__ (line 1239) | def __repr__(self):
  class RotScaleTransPoints (line 1252) | class RotScaleTransPoints(object):
    method _trans_bbox_points (line 1271) | def _trans_bbox_points(self, input_dict):
    method _rot_bbox_points (line 1290) | def _rot_bbox_points(self, input_dict):
    method _scale_bbox_points (line 1320) | def _scale_bbox_points(self, input_dict):
    method _random_scale (line 1342) | def _random_scale(self, input_dict):
    method __call__ (line 1356) | def __call__(self, input_dict):
    method __repr__ (line 1377) | def __repr__(self):
  class PointShuffle (line 1388) | class PointShuffle(object):
    method __call__ (line 1391) | def __call__(self, input_dict):
    method __repr__ (line 1415) | def __repr__(self):
  class ObjectRangeFilter (line 1420) | class ObjectRangeFilter(object):
    method __init__ (line 1427) | def __init__(self, point_cloud_range):
    method __call__ (line 1430) | def __call__(self, input_dict):
    method __repr__ (line 1470) | def __repr__(self):
  class PointsRangeFilter (line 1478) | class PointsRangeFilter(object):
    method __init__ (line 1485) | def __init__(self, point_cloud_range):
    method __call__ (line 1488) | def __call__(self, input_dict):
    method __repr__ (line 1516) | def __repr__(self):
  class ObjectNameFilter (line 1524) | class ObjectNameFilter(object):
    method __init__ (line 1531) | def __init__(self, classes):
    method __call__ (line 1535) | def __call__(self, input_dict):
    method __repr__ (line 1559) | def __repr__(self):
  class PointSample (line 1567) | class PointSample(object):
    method __init__ (line 1581) | def __init__(self, num_points, sample_range=None, replace=False):
    method _points_random_sampling (line 1586) | def _points_random_sampling(self,
    method __call__ (line 1634) | def __call__(self, results):
    method __repr__ (line 1665) | def __repr__(self):
  class IndoorPointSample (line 1676) | class IndoorPointSample(PointSample):
    method __init__ (line 1686) | def __init__(self, *args, **kwargs):
  class IndoorPatchPointSample (line 1693) | class IndoorPatchPointSample(object):
    method __init__ (line 1731) | def __init__(self,
    method _input_generation (line 1755) | def _input_generation(self, coords, patch_center, coord_max, attributes,
    method _patch_points_sampling (line 1796) | def _patch_points_sampling(self, points, sem_mask):
    method __call__ (line 1896) | def __call__(self, results):
    method __repr__ (line 1923) | def __repr__(self):
  class BackgroundPointsFilter (line 1938) | class BackgroundPointsFilter(object):
    method __init__ (line 1945) | def __init__(self, bbox_enlarge_range):
    method __call__ (line 1956) | def __call__(self, input_dict):
    method __repr__ (line 1995) | def __repr__(self):
  class VoxelBasedPointSampler (line 2003) | class VoxelBasedPointSampler(object):
    method __init__ (line 2015) | def __init__(self, cur_sweep_cfg, prev_sweep_cfg=None, time_dim=3):
    method _sample_points (line 2028) | def _sample_points(self, points, sampler, point_dim):
    method __call__ (line 2054) | def __call__(self, results):
    method __repr__ (line 2122) | def __repr__(self):
  class AffineResize (line 2145) | class AffineResize(object):
    method __init__ (line 2162) | def __init__(self, img_scale, down_ratio, bbox_clip_border=True):
    method __call__ (line 2168) | def __call__(self, results):
    method _affine_bboxes (line 2245) | def _affine_bboxes(self, results, matrix):
    method _affine_transform (line 2268) | def _affine_transform(self, points, matrix):
    method _get_transform_matrix (line 2287) | def _get_transform_matrix(self, center, scale, output_scale):
    method _get_ref_point (line 2322) | def _get_ref_point(self, ref_point1, ref_point2):
    method __repr__ (line 2333) | def __repr__(self):
  class RandomShiftScale (line 2341) | class RandomShiftScale(object):
    method __init__ (line 2354) | def __init__(self, shift_scale, aug_prob):
    method __call__ (line 2359) | def __call__(self, results):
    method __repr__ (line 2392) | def __repr__(self):

FILE: mmdet3d/datasets/s3dis_dataset.py
  class S3DISDataset (line 16) | class S3DISDataset(Custom3DDataset):
    method __init__ (line 49) | def __init__(self,
    method get_ann_info (line 70) | def get_ann_info(self, index):
    method get_data_info (line 114) | def get_data_info(self, index):
    method _build_default_pipeline (line 139) | def _build_default_pipeline(self):
  class _S3DISSegDataset (line 157) | class _S3DISSegDataset(Custom3DSegDataset):
    method __init__ (line 201) | def __init__(self,
    method get_ann_info (line 225) | def get_ann_info(self, index):
    method _build_default_pipeline (line 245) | def _build_default_pipeline(self):
    method show (line 273) | def show(self, results, out_dir, show=True, pipeline=None):
    method get_scene_idxs (line 297) | def get_scene_idxs(self, scene_idxs):
  class S3DISSegDataset (line 312) | class S3DISSegDataset(_S3DISSegDataset):
    method __init__ (line 345) | def __init__(self,
    method concat_data_infos (line 396) | def concat_data_infos(self, data_infos):
    method concat_scene_idxs (line 406) | def concat_scene_idxs(self, scene_idxs):
    method _duplicate_to_list (line 422) | def _duplicate_to_list(x, num):
    method _check_ann_files (line 426) | def _check_ann_files(self, ann_file):
    method _check_scene_idxs (line 433) | def _check_scene_idxs(self, scene_idx, num):

FILE: mmdet3d/datasets/samplers/d_sampler.py
  class CustomDistributedSampler (line 6) | class CustomDistributedSampler(_DistributedSampler):
    method __init__ (line 8) | def __init__(self,
    method __iter__ (line 19) | def __iter__(self):

FILE: mmdet3d/datasets/samplers/infinite_group_each_sample_in_batch_sampler.py
  function sync_random_seed (line 14) | def sync_random_seed(seed=None, device='cuda'):
  class InfiniteGroupEachSampleInBatchSampler (line 48) | class InfiniteGroupEachSampleInBatchSampler(Sampler):
    method __init__ (line 56) | def __init__(self,
    method _infinite_group_indices (line 99) | def _infinite_group_indices(self):
    method _group_indices_per_global_sample_idx (line 105) | def _group_indices_per_global_sample_idx(self, global_sample_idx):
    method __iter__ (line 111) | def __iter__(self):
    method __len__ (line 126) | def __len__(self):
    method set_epoch (line 130) | def set_epoch(self, epoch):
  class InfiniteGroupEachSampleInBatchSamplerEval (line 134) | class InfiniteGroupEachSampleInBatchSamplerEval(Sampler):
    method __init__ (line 142) | def __init__(self,
    method _infinite_group_indices (line 184) | def _infinite_group_indices(self):
    method _group_indices_per_global_sample_idx (line 190) | def _group_indices_per_global_sample_idx(self, global_sample_idx):
    method __iter__ (line 196) | def __iter__(self):
    method __len__ (line 214) | def __len__(self):
    method set_epoch (line 218) | def set_epoch(self, epoch):
  class TTADistributedSampler (line 223) | class TTADistributedSampler(Sampler):
    method __init__ (line 225) | def __init__(self,
    method __iter__ (line 246) | def __iter__(self):
    method __len__ (line 251) | def __len__(self):

FILE: mmdet3d/datasets/scannet_dataset.py
  class ScanNetDataset (line 18) | class ScanNetDataset(Custom3DDataset):
    method __init__ (line 53) | def __init__(self,
    method get_data_info (line 77) | def get_data_info(self, index):
    method get_ann_info (line 126) | def get_ann_info(self, index):
    method prepare_test_data (line 175) | def prepare_test_data(self, index):
    method _get_axis_align_matrix (line 197) | def _get_axis_align_matrix(info):
    method _build_default_pipeline (line 214) | def _build_default_pipeline(self):
    method show (line 232) | def show(self, results, out_dir, show=True, pipeline=None):
  class ScanNetSegDataset (line 257) | class ScanNetSegDataset(Custom3DSegDataset):
    method __init__ (line 318) | def __init__(self,
    method get_ann_info (line 342) | def get_ann_info(self, index):
    method _build_default_pipeline (line 362) | def _build_default_pipeline(self):
    method show (line 390) | def show(self, results, out_dir, show=True, pipeline=None):
    method get_scene_idxs (line 414) | def get_scene_idxs(self, scene_idxs):
    method format_results (line 426) | def format_results(self, results, txtfile_prefix=None):
  class ScanNetInstanceSegDataset (line 471) | class ScanNetInstanceSegDataset(Custom3DSegDataset):
    method get_ann_info (line 482) | def get_ann_info(self, index):
    method get_classes_and_palette (line 506) | def get_classes_and_palette(self, classes=None, palette=None):
    method _build_default_pipeline (line 525) | def _build_default_pipeline(self):
    method evaluate (line 555) | def evaluate(self,

FILE: mmdet3d/datasets/semantickitti_dataset.py
  class SemanticKITTIDataset (line 9) | class SemanticKITTIDataset(Custom3DDataset):
    method __init__ (line 44) | def __init__(self,
    method get_data_info (line 63) | def get_data_info(self, index):
    method get_ann_info (line 92) | def get_ann_info(self, index):

FILE: mmdet3d/datasets/sunrgbd_dataset.py
  class SUNRGBDDataset (line 16) | class SUNRGBDDataset(Custom3DDataset):
    method __init__ (line 49) | def __init__(self,
    method get_data_info (line 73) | def get_data_info(self, index):
    method get_ann_info (line 122) | def get_ann_info(self, index):
    method _build_default_pipeline (line 164) | def _build_default_pipeline(self):
    method show (line 183) | def show(self, results, out_dir, show=True, pipeline=None):
    method evaluate (line 230) | def evaluate(self,

FILE: mmdet3d/datasets/utils.py
  function is_loading_function (line 18) | def is_loading_function(transform):
  function get_loading_pipeline (line 53) | def get_loading_pipeline(pipeline):
  function extract_result_dict (line 116) | def extract_result_dict(results, key):
  function nuscenes_get_rt_matrix (line 145) | def nuscenes_get_rt_matrix(

FILE: mmdet3d/datasets/vector_map.py
  class LiDARInstanceLines (line 29) | class LiDARInstanceLines(object):
    method __init__ (line 33) | def __init__(self,
    method start_end_points (line 57) | def start_end_points(self):
    method bbox (line 79) | def bbox(self):
    method fixed_num_sampled_points (line 99) | def fixed_num_sampled_points(self):
    method fixed_num_sampled_points_ambiguity (line 119) | def fixed_num_sampled_points_ambiguity(self):
    method fixed_num_sampled_points_torch (line 140) | def fixed_num_sampled_points_torch(self):
    method shift_fixed_num_sampled_points (line 165) | def shift_fixed_num_sampled_points(self):
    method shift_fixed_num_sampled_points_v1 (line 203) | def shift_fixed_num_sampled_points_v1(self):
    method shift_fixed_num_sampled_points_v2 (line 250) | def shift_fixed_num_sampled_points_v2(self):
    method shift_fixed_num_sampled_points_v3 (line 308) | def shift_fixed_num_sampled_points_v3(self):
    method shift_fixed_num_sampled_points_v4 (line 377) | def shift_fixed_num_sampled_points_v4(self):
    method shift_fixed_num_sampled_points_torch (line 426) | def shift_fixed_num_sampled_points_torch(self):
  class VectorizedLocalMap (line 472) | class VectorizedLocalMap(object):
    method __init__ (line 480) | def __init__(self,
    method gen_vectorized_samples (line 517) | def gen_vectorized_samples(self, location, lidar2global_translation, p...
    method get_map_geom (line 575) | def get_map_geom(self, patch_box, patch_angle, layer_names, location, ...
    method _one_type_line_geom_to_vectors (line 594) | def _one_type_line_geom_to_vectors(self, line_geom):
    method _one_type_line_geom_to_instances (line 608) | def _one_type_line_geom_to_instances(self, line_geom):
    method poly_geoms_to_vectors (line 622) | def poly_geoms_to_vectors(self, polygon_geom):
    method ped_poly_geoms_to_instances (line 659) | def ped_poly_geoms_to_instances(self, ped_geom):
    method poly_geoms_to_instances (line 696) | def poly_geoms_to_instances(self, polygon_geom):
    method line_geoms_to_vectors (line 733) | def line_geoms_to_vectors(self, line_geom):
    method line_geoms_to_instances (line 740) | def line_geoms_to_instances(self, line_geom):
    method ped_geoms_to_vectors (line 748) | def ped_geoms_to_vectors(self, ped_geom):
    method get_contour_line (line 768) | def get_contour_line(self,patch_box,patch_angle,layer_name,location, f...
    method get_divider_line (line 820) | def get_divider_line(self,patch_box,patch_angle,layer_name,location, f...
    method get_ped_crossing_line (line 860) | def get_ped_crossing_line(self, patch_box, patch_angle, location, flip...
    method sample_pts_from_line (line 887) | def sample_pts_from_line(self, line):

FILE: mmdet3d/datasets/waymo_dataset.py
  class WaymoDataset (line 17) | class WaymoDataset(KittiDataset):
    method __init__ (line 57) | def __init__(self,
    method _get_pts_filename (line 90) | def _get_pts_filename(self, idx):
    method get_data_info (line 95) | def get_data_info(self, index):
    method format_results (line 138) | def format_results(self,
    method evaluate (line 220) | def evaluate(self,
    method bbox2result_kitti (line 360) | def bbox2result_kitti(self,
    method convert_valid_bboxes (line 475) | def convert_valid_bboxes(self, box_dict, info):

FILE: mmdet3d/models/backbones/base_pointnet.py
  class BasePointNet (line 8) | class BasePointNet(BaseModule, metaclass=ABCMeta):
    method __init__ (line 11) | def __init__(self, init_cfg=None, pretrained=None):
    method _split_point_feats (line 22) | def _split_point_feats(points):

FILE: mmdet3d/models/backbones/convnext.py
  class BaseBackbone (line 23) | class BaseBackbone(BaseModule, metaclass=ABCMeta):
    method __init__ (line 29) | def __init__(self, init_cfg=None):
    method forward (line 33) | def forward(self, x):
    method train (line 41) | def train(self, mode=True):
  class LayerNorm2d (line 50) | class LayerNorm2d(nn.LayerNorm):
    method __init__ (line 61) | def __init__(self, num_channels: int, **kwargs) -> None:
    method forward (line 65) | def forward(self, x):
  class ConvNeXtBlock (line 73) | class ConvNeXtBlock(BaseModule):
    method __init__ (line 99) | def __init__(self,
    method forward (line 139) | def forward(self, x):
  class ConvNeXt (line 171) | class ConvNeXt(BaseBackbone):
    method __init__ (line 232) | def __init__(self,
    method forward (line 340) | def forward(self, x):
    method _freeze_stages (line 358) | def _freeze_stages(self):
    method train (line 368) | def train(self, mode=True):

FILE: mmdet3d/models/backbones/dgcnn.py
  class DGCNNBackbone (line 10) | class DGCNNBackbone(BaseModule):
    method __init__ (line 32) | def __init__(self,
    method forward (line 75) | def forward(self, points):

FILE: mmdet3d/models/backbones/dla.py
  function dla_build_norm_layer (line 12) | def dla_build_norm_layer(cfg, num_features):
  class BasicBlock (line 39) | class BasicBlock(BaseModule):
    method __init__ (line 55) | def __init__(self,
    method forward (line 87) | def forward(self, x, identity=None):
  class Root (line 103) | class Root(BaseModule):
    method __init__ (line 119) | def __init__(self,
    method forward (line 140) | def forward(self, feat_list):
  class Tree (line 157) | class Tree(BaseModule):
    method __init__ (line 183) | def __init__(self,
    method forward (line 261) | def forward(self, x, identity=None, children=None):
  class DLANet (line 279) | class DLANet(BaseModule):
    method __init__ (line 304) | def __init__(self,
    method _make_conv_level (line 382) | def _make_conv_level(self,
    method _freeze_stages (line 421) | def _freeze_stages(self):
    method forward (line 439) | def forward(self, x):

FILE: mmdet3d/models/backbones/load.py
  function save_checkpoint (line 19) | def save_checkpoint(model, filename, optimizer=None, meta=None):

FILE: mmdet3d/models/backbones/mink_resnet.py
  class MinkResNet (line 18) | class MinkResNet(nn.Module):
    method __init__ (line 37) | def __init__(self, depth, in_channels, num_stages=4, pool=True):
    method init_weights (line 62) | def init_weights(self):
    method _make_layer (line 72) | def _make_layer(self, block, planes, blocks, stride):
    method forward (line 96) | def forward(self, x):

FILE: mmdet3d/models/backbones/multi_backbone.py
  class MultiBackbone (line 14) | class MultiBackbone(BaseModule):
    method __init__ (line 29) | def __init__(self,
    method forward (line 93) | def forward(self, points):

FILE: mmdet3d/models/backbones/nostem_regnet.py
  class NoStemRegNet (line 7) | class NoStemRegNet(RegNet):
    method __init__ (line 61) | def __init__(self, arch, init_cfg=None, **kwargs):
    method _make_stem_layer (line 64) | def _make_stem_layer(self, in_channels, base_channels):
    method forward (line 69) | def forward(self, x):

FILE: mmdet3d/models/backbones/pointnet2_sa_msg.py
  class PointNet2SAMSG (line 13) | class PointNet2SAMSG(BasePointNet):
    method __init__ (line 41) | def __init__(self,
    method forward (line 128) | def forward(self, points):

FILE: mmdet3d/models/backbones/pointnet2_sa_ssg.py
  class PointNet2SASSG (line 12) | class PointNet2SASSG(BasePointNet):
    method __init__ (line 34) | def __init__(self,
    method forward (line 90) | def forward(self, points):

FILE: mmdet3d/models/backbones/resnet.py
  class CustomResNet (line 11) | class CustomResNet(nn.Module):
    method __init__ (line 13) | def __init__(
    method forward (line 74) | def forward(self, x):

FILE: mmdet3d/models/backbones/second.py
  class SECOND (line 12) | class SECOND(BaseModule):
    method __init__ (line 24) | def __init__(self,
    method forward (line 78) | def forward(self, x):

FILE: mmdet3d/models/backbones/swin.py
  function swin_convert (line 25) | def swin_convert(ckpt):
  class PatchEmbed (line 79) | class PatchEmbed(BaseModule):
    method __init__ (line 100) | def __init__(self,
    method forward (line 150) | def forward(self, x):
  class PatchMerging (line 174) | class PatchMerging(BaseModule):
    method __init__ (line 192) | def __init__(self,
    method forward (line 216) | def forward(self, x, hw_shape):
  class WindowMSA (line 244) | class WindowMSA(BaseModule):
    method __init__ (line 263) | def __init__(self,
    method init_weights (line 300) | def init_weights(self):
    method forward (line 303) | def forward(self, x, mask=None):
    method double_step_seq (line 346) | def double_step_seq(step1, len1, step2, len2):
  class ShiftWindowMSA (line 353) | class ShiftWindowMSA(BaseModule):
    method __init__ (line 376) | def __init__(self,
    method forward (line 405) | def forward(self, query, hw_shape):
    method window_reverse (line 482) | def window_reverse(self, windows, H, W):
    method window_partition (line 499) | def window_partition(self, x):
  class SwinBlock (line 516) | class SwinBlock(BaseModule):
    method __init__ (line 538) | def __init__(self,
    method forward (line 581) | def forward(self, x, hw_shape):
  class SwinBlockSequence (line 595) | class SwinBlockSequence(BaseModule):
    method __init__ (line 620) | def __init__(self,
    method forward (line 665) | def forward(self, x, hw_shape):
  class SwinTransformer (line 680) | class SwinTransformer(BaseModule):
    method __init__ (line 730) | def __init__(self,
    method _freeze_stages (line 859) | def _freeze_stages(self):
    method init_weights (line 876) | def init_weights(self):
    method forward (line 946) | def forward(self, x):
    method train (line 973) | def train(self, mode=True):

FILE: mmdet3d/models/backbones/vovnet.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 384) | def _freeze_stages(self):
    method train (line 397) | def train(self, mode=True):

FILE: mmdet3d/models/backbones/vovnet2.py
  function dw_conv3x3 (line 92) | def dw_conv3x3(in_channels, out_channels, module_name, postfix, stride=1...
  function conv3x3 (line 116) | def conv3x3(in_channels, out_channels, module_name, postfix, stride=1, g...
  function conv1x1 (line 136) | def conv1x1(in_channels, out_channels, module_name, postfix, stride=1, g...
  class Hsigmoid (line 156) | class Hsigmoid(nn.Module):
    method __init__ (line 157) | def __init__(self, inplace=True):
    method forward (line 161) | def forward(self, x):
  class eSEModule (line 165) | class eSEModule(nn.Module):
    method __init__ (line 166) | def __init__(self, channel, reduction=4):
    method forward (line 172) | def forward(self, x):
  class _OSA_module (line 180) | class _OSA_module(nn.Module):
    method __init__ (line 181) | def __init__(
    method _forward (line 211) | def _forward(self, x):
    method forward (line 233) | def forward(self, x):
  class _OSA_stage (line 243) | class _OSA_stage(nn.Sequential):
    method __init__ (line 244) | def __init__(
  class VoVNet2 (line 279) | class VoVNet2(BaseModule):
    method __init__ (line 280) | def __init__(self, spec_name, input_ch=3, out_features=None,
    method _initialize_weights (line 346) | def _initialize_weights(self):
    method forward (line 351) | def forward(self, x):
    method _freeze_stages (line 363) | def _freeze_stages(self):
    method train (line 376) | def train(self, mode=True):

FILE: mmdet3d/models/builder.py
  function build_backbone (line 31) | def build_backbone(cfg):
  function build_neck (line 39) | def build_neck(cfg):
  function build_roi_extractor (line 47) | def build_roi_extractor(cfg):
  function build_shared_head (line 55) | def build_shared_head(cfg):
  function build_head (line 63) | def build_head(cfg):
  function build_loss (line 71) | def build_loss(cfg):
  function build_detector (line 81) | def build_detector(cfg, train_cfg=None, test_cfg=None):
  function build_segmentor (line 99) | def build_segmentor(cfg, train_cfg=None, test_cfg=None):
  function build_model (line 113) | def build_model(cfg, train_cfg=None, test_cfg=None):
  function build_voxel_encoder (line 125) | def build_voxel_encoder(cfg):
  function build_middle_encoder (line 130) | def build_middle_encoder(cfg):
  function build_fusion_layer (line 135) | def build_fusion_layer(cfg):

FILE: mmdet3d/models/decode_heads/decode_head.py
  class Base3DDecodeHead (line 11) | class Base3DDecodeHead(BaseModule, metaclass=ABCMeta):
    method __init__ (line 31) | def __init__(self,
    method init_weights (line 62) | def init_weights(self):
    method forward (line 69) | def forward(self, inputs):
    method forward_train (line 73) | def forward_train(self, inputs, img_metas, pts_semantic_mask, train_cfg):
    method forward_test (line 90) | def forward_test(self, inputs, img_metas, test_cfg):
    method cls_seg (line 103) | def cls_seg(self, feat):
    method losses (line 111) | def losses(self, seg_logit, seg_label):

FILE: mmdet3d/models/decode_heads/dgcnn_head.py
  class DGCNNHead (line 10) | class DGCNNHead(Base3DDecodeHead):
    method __init__ (line 22) | def __init__(self, fp_channels=(1216, 512), **kwargs):
    method _extract_input (line 38) | def _extract_input(self, feat_dict):
    method forward (line 51) | def forward(self, feat_dict):

FILE: mmdet3d/models/decode_heads/paconv_head.py
  class PAConvHead (line 9) | class PAConvHead(PointNet2Head):
    method __init__ (line 21) | def __init__(self,
    method forward (line 40) | def forward(self, feat_dict):

FILE: mmdet3d/models/decode_heads/pointnet2_head.py
  class PointNet2Head (line 11) | class PointNet2Head(Base3DDecodeHead):
    method __init__ (line 23) | def __init__(self,
    method _extract_input (line 46) | def _extract_input(self, feat_dict):
    method forward (line 62) | def forward(self, feat_dict):

FILE: mmdet3d/models/dense_heads/anchor3d_head.py
  class Anchor3DHead (line 16) | class Anchor3DHead(BaseModule, AnchorTrainMixin):
    method __init__ (line 42) | def __init__(self,
    method _init_assigner_sampler (line 118) | def _init_assigner_sampler(self):
    method _init_layers (line 134) | def _init_layers(self):
    method forward_single (line 144) | def forward_single(self, x):
    method forward (line 161) | def forward(self, feats):
    method get_anchors (line 174) | def get_anchors(self, featmap_sizes, input_metas, device='cuda'):
    method loss_single (line 194) | def loss_single(self, cls_score, bbox_pred, dir_cls_preds, labels,
    method add_sin_difference (line 281) | def add_sin_difference(boxes1, boxes2):
    method loss (line 305) | def loss(self,
    method get_bboxes (line 376) | def get_bboxes(self,
    method get_bboxes_single (line 427) | def get_bboxes_single(self,

FILE: mmdet3d/models/dense_heads/anchor_free_mono3d_head.py
  class AnchorFreeMono3DHead (line 15) | class AnchorFreeMono3DHead(BaseMono3DDenseHead):
    method __init__ (line 82) | def __init__(
    method _init_layers (line 181) | def _init_layers(self):
    method _init_cls_convs (line 187) | def _init_cls_convs(self):
    method _init_reg_convs (line 207) | def _init_reg_convs(self):
    method _init_branch (line 227) | def _init_branch(self, conv_channels=(64), conv_strides=(1)):
    method _init_predictor (line 250) | def _init_predictor(self):
    method init_weights (line 284) | def init_weights(self):
    method forward (line 318) | def forward(self, feats):
    method forward_single (line 343) | def forward_single(self, x):
    method loss (line 397) | def loss(self,
    method get_bboxes (line 446) | def get_bboxes(self,
    method get_targets (line 476) | def get_targets(self, points, gt_bboxes_list, gt_labels_list,
    method _get_points_single (line 502) | def _get_points_single(self,
    method get_points (line 518) | def get_points(self, featmap_sizes, dtype, device, flatten=False):

FILE: mmdet3d/models/dense_heads/base_conv_bbox_head.py
  class BaseConvBboxHead (line 11) | class BaseConvBboxHead(BaseModule):
    method __init__ (line 22) | def __init__(self,
    method _add_conv_branch (line 85) | def _add_conv_branch(self, in_channels, conv_channels):
    method forward (line 105) | def forward(self, feats):

FILE: mmdet3d/models/dense_heads/base_mono3d_dense_head.py
  class BaseMono3DDenseHead (line 7) | class BaseMono3DDenseHead(BaseModule, metaclass=ABCMeta):
    method __init__ (line 10) | def __init__(self, init_cfg=None):
    method loss (line 14) | def loss(self, **kwargs):
    method get_bboxes (line 19) | def get_bboxes(self, **kwargs):
    method forward_train (line 23) | def forward_train(self,

FILE: mmdet3d/models/dense_heads/centerpoint_head.py
  class SeparateHead (line 20) | class SeparateHead(BaseModule):
    method __init__ (line 38) | def __init__(self,
    method init_weights (line 90) | def init_weights(self):
    method forward (line 98) | def forward(self, x):
  class DCNSeparateHead (line 130) | class DCNSeparateHead(BaseModule):
    method __init__ (line 155) | def __init__(self,
    method init_weights (line 211) | def init_weights(self):
    method forward (line 216) | def forward(self, x):
  class CenterHead (line 251) | class CenterHead(BaseModule):
    method __init__ (line 281) | def __init__(self,
    method forward_single (line 347) | def forward_single(self, x):
    method forward (line 368) | def forward(self, input_dict, *args, **kwargs):
    method _gather_feat (line 389) | def _gather_feat(self, feat, ind, mask=None):
    method get_targets (line 414) | def get_targets(self, gt_bboxes_3d, gt_labels_3d):
    method get_targets_single (line 461) | def get_targets_single(self, gt_bboxes_3d, gt_labels_3d):
    method loss (line 611) | def loss(self,  gt_bboxes_3d, gt_labels_3d, preds_dicts, img_metas=Non...
    method loss_ (line 618) | def loss_(self, heatmaps, anno_boxes, inds, masks, preds_dicts, **kwar...
    method get_bboxes (line 700) | def get_bboxes(self, preds_dicts, img_metas, img=None, rescale=False):
    method get_task_detections (line 797) | def get_task_detections(self, num_class_with_bg, batch_cls_preds,

FILE: mmdet3d/models/dense_heads/centerpoint_head_single_task.py
  class SeparateHead (line 19) | class SeparateHead(BaseModule):
    method __init__ (line 37) | def __init__(self,
    method init_weights (line 87) | def init_weights(self):
    method forward (line 94) | def forward(self, x):
  class DCNSeparateHead (line 125) | class DCNSeparateHead(BaseModule):
    method __init__ (line 150) | def __init__(self,
    method init_weights (line 205) | def init_weights(self):
    method forward (line 210) | def forward(self, x):
  class CenterHead (line 244) | class CenterHead(BaseModule):
    method __init__ (line 274) | def __init__(self,
    method forward_single (line 334) | def forward_single(self, x):
    method forward (line 353) | def forward(self, feats):
    method _gather_feat (line 365) | def _gather_feat(self, feat, ind, mask=None):
    method get_targets (line 390) | def get_targets(self, gt_bboxes_3d, gt_labels_3d):
    method get_targets_single (line 437) | def get_targets_single(self, gt_bboxes_3d, gt_labels_3d):
    method loss (line 587) | def loss(self, gt_bboxes_3d, gt_labels_3d, preds_dicts, **kwargs):
    method get_bboxes (line 670) | def get_bboxes(self, preds_dicts, img_metas, img=None, rescale=False):
    method get_task_detections (line 765) | def get_task_detections(self, batch_cls_preds,

FILE: mmdet3d/models/dense_heads/fcaf3d_head.py
  class FCAF3DHead (line 21) | class FCAF3DHead(BaseModule):
    method __init__ (line 49) | def __init__(self,
    method _make_block (line 77) | def _make_block(in_channels, out_channels):
    method _make_up_block (line 93) | def _make_up_block(in_channels, out_channels):
    method _init_layers (line 115) | def _init_layers(self, in_channels, out_channels, n_reg_outs, n_classes):
    method init_weights (line 144) | def init_weights(self):
    method forward (line 151) | def forward(self, x):
    method forward_train (line 180) | def forward_train(self, x, gt_bboxes, gt_labels, input_metas):
    method forward_test (line 197) | def forward_test(self, x, input_metas):
    method _prune (line 211) | def _prune(self, x, scores):
    method _forward_single (line 236) | def _forward_single(self, x, scale):
    method _loss_single (line 270) | def _loss_single(self, center_preds, bbox_preds, cls_preds, points,
    method _loss (line 324) | def _loss(self, center_preds, bbox_preds, cls_preds, points, gt_bboxes,
    method _get_bboxes_single (line 362) | def _get_bboxes_single(self, center_preds, bbox_preds, cls_preds, points,
    method _get_bboxes (line 399) | def _get_bboxes(self, center_preds, bbox_preds, cls_preds, points,
    method _bbox_to_loss (line 429) | def _bbox_to_loss(bbox):
    method _bbox_pred_to_bbox (line 450) | def _bbox_pred_to_bbox(points, bbox_pred):
    method _get_face_distances (line 495) | def _get_face_distances(points, boxes):
    method _get_centerness (line 523) | def _get_centerness(face_distances):
    method _get_targets (line 542) | def _get_targets(self, points, gt_bboxes, gt_labels):
    method _single_scene_multiclass_nms (line 620) | def _single_scene_multiclass_nms(self, bboxes, scores, input_meta):

FILE: mmdet3d/models/dense_heads/fcos_mono3d_head.py
  class FCOSMono3DHead (line 21) | class FCOSMono3DHead(AnchorFreeMono3DHead):
    method __init__ (line 50) | def __init__(self,
    method _init_layers (line 102) | def _init_layers(self):
    method init_weights (line 115) | def init_weights(self):
    method forward (line 128) | def forward(self, feats):
    method forward_single (line 156) | def forward_single(self, x, scale, stride):
    method add_sin_difference (line 192) | def add_sin_difference(boxes1, boxes2):
    method get_direction_target (line 216) | def get_direction_target(reg_targets,
    method loss (line 255) | def loss(self,
    method get_bboxes (line 482) | def get_bboxes(self,
    method _get_bboxes_single (line 567) | def _get_bboxes_single(self,
    method pts2Dto3D (line 696) | def pts2Dto3D(points, view):
    method _get_points_single (line 732) | def _get_points_single(self,
    method get_targets (line 744) | def get_targets(self, points, gt_bboxes_list, gt_labels_list,
    method _get_target_single (line 852) | def _get_target_single(self, gt_bboxes, gt_labels, gt_bboxes_3d,

FILE: mmdet3d/models/dense_heads/free_anchor3d_head.py
  class FreeAnchor3DHead (line 13) | class FreeAnchor3DHead(Anchor3DHead):
    method __init__ (line 31) | def __init__(self,
    method loss (line 45) | def loss(self,
    method positive_bag_loss (line 247) | def positive_bag_loss(self, matched_cls_prob, matched_box_prob):
    method negative_bag_loss (line 269) | def negative_bag_loss(self, cls_prob, box_prob):

FILE: mmdet3d/models/dense_heads/groupfree3d_head.py
  class PointsObjClsModule (line 24) | class PointsObjClsModule(BaseModule):
    method __init__ (line 39) | def __init__(self,
    method forward (line 67) | def forward(self, seed_features):
  class GeneralSamplingModule (line 81) | class GeneralSamplingModule(nn.Module):
    method forward (line 87) | def forward(self, xyz, features, sample_inds):
  class GroupFree3DHead (line 110) | class GroupFree3DHead(BaseModule):
    method __init__ (line 140) | def __init__(self,
    method init_weights (line 253) | def init_weights(self):
    method _get_cls_out_channels (line 266) | def _get_cls_out_channels(self):
    method _get_reg_out_channels (line 271) | def _get_reg_out_channels(self):
    method _extract_input (line 281) | def _extract_input(self, feat_dict):
    method forward (line 299) | def forward(self, feat_dict, sample_mod):
    method loss (line 394) | def loss(self,
    method get_targets (line 537) | def get_targets(self,
    method get_targets_single (line 646) | def get_targets_single(self,
    method get_bboxes (line 865) | def get_bboxes(self,
    method multiclass_nms_single (line 936) | def multiclass_nms_single(self, obj_scores, sem_scores, bbox, points,

FILE: mmdet3d/models/dense_heads/monoflex_head.py
  class MonoFlexHead (line 21) | class MonoFlexHead(AnchorFreeMono3DHead):
    method __init__ (line 82) | def __init__(self,
    method _init_edge_module (line 128) | def _init_edge_module(self):
    method init_weights (line 138) | def init_weights(self):
    method _init_predictor (line 149) | def _init_predictor(self):
    method _init_layers (line 181) | def _init_layers(self):
    method forward_train (line 187) | def forward_train(self, x, input_metas, gt_bboxes, gt_labels, gt_bboxe...
    method forward (line 234) | def forward(self, feats, input_metas):
    method forward_single (line 255) | def forward_single(self, x, input_metas):
    method get_bboxes (line 315) | def get_bboxes(self, cls_scores, bbox_preds, input_metas):
    method decode_heatmap (line 360) | def decode_heatmap(self,
    method get_predictions (line 421) | def get_predictions(self, pred_reg, labels3d, centers2d, reg_mask,
    method get_targets (line 460) | def get_targets(self, gt_bboxes_list, gt_labels_list, gt_bboxes_3d_list,
    method loss (line 639) | def loss(self,

FILE: mmdet3d/models/dense_heads/parta2_rpn_head.py
  class PartA2RPNHead (line 13) | class PartA2RPNHead(Anchor3DHead):
    method __init__ (line 50) | def __init__(self,
    method loss (line 86) | def loss(self,
    method get_bboxes_single (line 126) | def get_bboxes_single(self,
    method class_agnostic_nms (line 222) | def class_agnostic_nms(self, mlvl_bboxes, mlvl_bboxes_for_nms,

FILE: mmdet3d/models/dense_heads/pgd_head.py
  class PGDHead (line 17) | class PGDHead(FCOSMono3DHead):
    method __init__ (line 58) | def __init__(self,
    method _init_layers (line 116) | def _init_layers(self):
    method _init_predictor (line 128) | def _init_predictor(self):
    method init_weights (line 159) | def init_weights(self):
    method forward (line 185) | def forward(self, feats):
    method forward_single (line 218) | def forward_single(self, x, scale, stride):
    method get_proj_bbox2d (line 265) | def get_proj_bbox2d(self,
    method get_pos_predictions (line 443) | def get_pos_predictions(self, bbox_preds, dir_cls_preds, depth_cls_preds,
    method loss (line 526) | def loss(self,
    method get_bboxes (line 795) | def get_bboxes(self,
    method _get_bboxes_single (line 912) | def _get_bboxes_single(self,
    method get_targets (line 1111) | def get_targets(self, points, gt_bboxes_list, gt_labels_list,

FILE: mmdet3d/models/dense_heads/point_rpn_head.py
  class PointRPNHead (line 15) | class PointRPNHead(BaseModule):
    method __init__ (line 35) | def __init__(self,
    method _make_fc_layers (line 69) | def _make_fc_layers(self, fc_cfg, input_channels, output_channels):
    method _get_cls_out_channels (line 92) | def _get_cls_out_channels(self):
    method _get_reg_out_channels (line 97) | def _get_reg_out_channels(self):
    method forward (line 104) | def forward(self, feat_dict):
    method loss (line 127) | def loss(self,
    method get_targets (line 172) | def get_targets(self, points, gt_bboxes_3d, gt_labels_3d):
    method get_targets_single (line 205) | def get_targets_single(self, points, gt_bboxes_3d, gt_labels_3d):
    method get_bboxes (line 246) | def get_bboxes(self,
    method class_agnostic_nms (line 284) | def class_agnostic_nms(self, obj_scores, sem_scores, bbox, points,
    method _assign_targets_by_points_inside (line 354) | def _assign_targets_by_points_inside(self, bboxes_3d, points):

FILE: mmdet3d/models/dense_heads/shape_aware_head.py
  class BaseShapeHead (line 17) | class BaseShapeHead(BaseModule):
    method __init__ (line 46) | def __init__(self,
    method forward (line 124) | def forward(self, x):
  class ShapeAwareHead (line 168) | class ShapeAwareHead(Anchor3DHead):
    method __init__ (line 179) | def __init__(self, tasks, assign_per_class=True, init_cfg=None, **kwar...
    method init_weights (line 185) | def init_weights(self):
    method _init_layers (line 195) | def _init_layers(self):
    method forward_single (line 216) | def forward_single(self, x):
    method loss_single (line 249) | def loss_single(self, cls_score, bbox_pred, dir_cls_preds, labels,
    method loss (line 311) | def loss(self,
    method get_bboxes (line 378) | def get_bboxes(self,
    method get_bboxes_single (line 430) | def get_bboxes_single(self,

FILE: mmdet3d/models/dense_heads/smoke_mono3d_head.py
  class SMOKEMono3DHead (line 16) | class SMOKEMono3DHead(AnchorFreeMono3DHead):
    method __init__ (line 49) | def __init__(self,
    method forward (line 76) | def forward(self, feats):
    method forward_single (line 94) | def forward_single(self, x):
    method get_bboxes (line 115) | def get_bboxes(self, cls_scores, bbox_preds, img_metas, rescale=None):
    method decode_heatmap (line 166) | def decode_heatmap(self,
    method get_predictions (line 224) | def get_predictions(self, labels3d, centers2d, gt_locations, gt_dimens...
    method get_targets (line 294) | def get_targets(self, gt_bboxes, gt_labels, gt_bboxes_3d, gt_labels_3d,
    method loss (line 430) | def loss(self,

FILE: mmdet3d/models/dense_heads/ssd_3d_head.py
  class SSD3DHead (line 16) | class SSD3DHead(VoteHead):
    method __init__ (line 42) | def __init__(self,
    method _get_cls_out_channels (line 85) | def _get_cls_out_channels(self):
    method _get_reg_out_channels (line 90) | def _get_reg_out_channels(self):
    method _extract_input (line 97) | def _extract_input(self, feat_dict):
    method loss (line 115) | def loss(self,
    method get_targets (line 221) | def get_targets(self,
    method get_targets_single (line 309) | def get_targets_single(self,
    method get_bboxes (line 442) | def get_bboxes(self, points, bbox_preds, input_metas, rescale=False):
    method multiclass_nms_single (line 475) | def multiclass_nms_single(self, obj_scores, sem_scores, bbox, points,
    method _assign_targets_by_points_inside (line 540) | def _assign_targets_by_points_inside(self, bboxes_3d, points):

FILE: mmdet3d/models/dense_heads/train_mixins.py
  class AnchorTrainMixin (line 9) | class AnchorTrainMixin(object):
    method anchor_target_3d (line 12) | def anchor_target_3d(self,
    method anchor_target_3d_single (line 102) | def anchor_target_3d_single(self,
    method anchor_target_single_assigner (line 238) | def anchor_target_single_assigner(self,
  function get_direction_target (line 319) | def get_direction_target(anchors,

FILE: mmdet3d/models/dense_heads/vote_head.py
  class VoteHead (line 18) | class VoteHead(BaseModule):
    method __init__ (line 42) | def __init__(self,
    method _get_cls_out_channels (line 96) | def _get_cls_out_channels(self):
    method _get_reg_out_channels (line 101) | def _get_reg_out_channels(self):
    method _extract_input (line 108) | def _extract_input(self, feat_dict):
    method forward (line 135) | def forward(self, feat_dict, sample_mod):
    method loss (line 222) | def loss(self,
    method get_targets (line 350) | def get_targets(self,
    method get_targets_single (line 439) | def get_targets_single(self,
    method get_bboxes (line 563) | def get_bboxes(self,
    method multiclass_nms_single (line 605) | def multiclass_nms_single(self, obj_scores, sem_scores, bbox, points,

FILE: mmdet3d/models/detectors/base.py
  class Base3DDetector (line 13) | class Base3DDetector(BaseDetector):
    method forward_test (line 16) | def forward_test(self, points, img_metas, img=None, **kwargs):
    method forward (line 48) | def forward(self, return_loss=True, **kwargs):
    method show_results (line 64) | def show_results(self, data, result, out_dir, show=False, score_thr=No...

FILE: mmdet3d/models/detectors/bevdet.py
  function convert_color (line 18) | def convert_color(img_path):
  function save_tensor (line 25) | def save_tensor(tensor, path, pad_value=254.0,normalize=False):
  class BEVDet (line 40) | class BEVDet(CenterPoint):
    method __init__ (line 42) | def __init__(self, img_view_transformer, img_bev_encoder_backbone,
    method image_encoder (line 50) | def image_encoder(self, img):
    method bev_encoder (line 64) | def bev_encoder(self, x):
    method extract_img_feat (line 71) | def extract_img_feat(self, img, img_metas, **kwargs):
    method extract_feat (line 82) | def extract_feat(self, points, img, img_metas, **kwargs):
    method forward_train (line 88) | def forward_train(self,
    method forward_test (line 133) | def forward_test(self,
    method aug_test (line 171) | def aug_test(self, points, img_metas, img=None, rescale=False):
    method simple_test (line 175) | def simple_test(self,
    method forward_dummy (line 216) | def forward_dummy(self,
  class BEVDetTRT (line 230) | class BEVDetTRT(BEVDet):
    method result_serialize (line 232) | def result_serialize(self, outs):
    method result_deserialize (line 239) | def result_deserialize(self, outs):
    method forward (line 249) | def forward(
    method get_bev_pool_input (line 275) | def get_bev_pool_input(self, input):
  class BEVDet4D (line 281) | class BEVDet4D(BEVDet):
    method __init__ (line 283) | def __init__(self,
    method shift_feature (line 300) | def shift_feature(self, input, trans, rots, bda, bda_adj=None):
    method prepare_bev_feat (line 371) | def prepare_bev_feat(self, img, rot, tran, intrin, post_rot, post_tran,
    method extract_img_feat_sequential (line 380) | def extract_img_feat_sequential(self, inputs, feat_prev):
    method prepare_inputs (line 406) | def prepare_inputs(self, inputs):
    method extract_img_feat (line 426) | def extract_img_feat(self,
  class BEVDepth4D (line 487) | class BEVDepth4D(BEVDet4D):
    method forward_train (line 489) | def forward_train(self,

FILE: mmdet3d/models/detectors/centerpoint.py
  class CenterPoint (line 10) | class CenterPoint(MVXTwoStageDetector):
    method __init__ (line 13) | def __init__(self,
    method with_velocity (line 37) | def with_velocity(self):
    method extract_pts_feat (line 42) | def extract_pts_feat(self, pts, img_feats, img_metas):
    method forward_pts_train (line 57) | def forward_pts_train(self,
    method simple_test_pts (line 83) | def simple_test_pts(self, x, img_metas, rescale=False):
    method aug_test_pts (line 95) | def aug_test_pts(self, feats, img_metas, rescale=False):
    method aug_test (line 197) | def aug_test(self, points, img_metas, imgs=None, rescale=False):

FILE: mmdet3d/models/detectors/dynamic_voxelnet.py
  class DynamicVoxelNet (line 11) | class DynamicVoxelNet(VoxelNet):
    method __init__ (line 16) | def __init__(self,
    method extract_feat (line 39) | def extract_feat(self, points, img_metas):
    method voxelize (line 52) | def voxelize(self, points):

FILE: mmdet3d/models/detectors/fcos_mono3d.py
  class FCOSMono3D (line 7) | class FCOSMono3D(SingleStageMono3DDetector):
    method __init__ (line 14) | def __init__(self,

FILE: mmdet3d/models/detectors/groupfree3dnet.py
  class GroupFree3DNet (line 10) | class GroupFree3DNet(SingleStage3DDetector):
    method __init__ (line 13) | def __init__(self,
    method forward_train (line 26) | def forward_train(self,
    method simple_test (line 62) | def simple_test(self, points, img_metas, imgs=None, rescale=False):
    method aug_test (line 84) | def aug_test(self, points, img_metas, imgs=None, rescale=False):

FILE: mmdet3d/models/detectors/h3dnet.py
  class H3DNet (line 10) | class H3DNet(TwoStage3DDetector):
    method __init__ (line 16) | def __init__(self,
    method forward_train (line 35) | def forward_train(self,
    method simple_test (line 101) | def simple_test(self, points, img_metas, imgs=None, rescale=False):
    method aug_test (line 133) | def aug_test(self, points, img_metas, imgs=None, rescale=False):
    method extract_feats (line 171) | def extract_feats(self, points, img_metas):

FILE: mmdet3d/models/detectors/imvotenet.py
  function sample_valid_seeds (line 14) | def sample_valid_seeds(mask, num_sampled_seed=1024):
  class ImVoteNet (line 57) | class ImVoteNet(Base3DDetector):
    method __init__ (line 60) | def __init__(self,
    method freeze_img_branch_params (line 171) | def freeze_img_branch_params(self):
    method _load_from_state_dict (line 189) | def _load_from_state_dict(self, state_dict, prefix, local_metadata, st...
    method train (line 203) | def train(self, mode=True):
    method with_img_bbox (line 219) | def with_img_bbox(self):
    method with_img_bbox_head (line 226) | def with_img_bbox_head(self):
    method with_img_backbone (line 232) | def with_img_backbone(self):
    method with_img_neck (line 237) | def with_img_neck(self):
    method with_img_rpn (line 242) | def with_img_rpn(self):
    method with_img_roi_head (line 247) | def with_img_roi_head(self):
    method with_pts_bbox (line 252) | def with_pts_bbox(self):
    method with_pts_backbone (line 258) | def with_pts_backbone(self):
    method with_pts_neck (line 263) | def with_pts_neck(self):
    method extract_feat (line 267) | def extract_feat(self, imgs):
    method extract_img_feat (line 271) | def extract_img_feat(self, img):
    method extract_img_feats (line 278) | def extract_img_feats(self, imgs):
    method extract_pts_feat (line 292) | def extract_pts_feat(self, pts):
    method extract_pts_feats (line 304) | def extract_pts_feats(self, pts):
    method extract_bboxes_2d (line 310) | def extract_bboxes_2d(self,
    method forward_train (line 368) | def forward_train(self,
    method forward_test (line 520) | def forward_test(self,
    method simple_test_img_only (line 600) | def simple_test_img_only(self,
    method simple_test (line 637) | def simple_test(self,
    method aug_test_img_only (line 703) | def aug_test_img_only(self, img, img_metas, rescale=False):
    method aug_test (line 735) | def aug_test(self,

FILE: mmdet3d/models/detectors/imvoxelnet.py
  class ImVoxelNet (line 11) | class ImVoxelNet(BaseDetector):
    method __init__ (line 14) | def __init__(self,
    method extract_feat (line 37) | def extract_feat(self, img, img_metas):
    method forward_train (line 78) | def forward_train(self, img, img_metas, gt_bboxes_3d, gt_labels_3d,
    method forward_test (line 96) | def forward_test(self, img, img_metas, **kwargs):
    method simple_test (line 109) | def simple_test(self, img, img_metas):
    method aug_test (line 128) | def aug_test(self, imgs, img_metas, **kwargs):

FILE: mmdet3d/models/detectors/mink_single_stage.py
  class MinkSingleStage3DDetector (line 15) | class MinkSingleStage3DDetector(Base3DDetector):
    method __init__ (line 31) | def __init__(self,
    method extract_feat (line 47) | def extract_feat(self, points):
    method forward_train (line 63) | def forward_train(self, points, gt_bboxes_3d, gt_labels_3d, img_metas):
    method simple_test (line 81) | def simple_test(self, points, img_metas, *args, **kwargs):
    method aug_test (line 99) | def aug_test(self, points, img_metas, **kwargs):

FILE: mmdet3d/models/detectors/mvx_faster_rcnn.py
  class MVXFasterRCNN (line 11) | class MVXFasterRCNN(MVXTwoStageDetector):
    method __init__ (line 14) | def __init__(self, **kwargs):
  class DynamicMVXFasterRCNN (line 19) | class DynamicMVXFasterRCNN(MVXTwoStageDetector):
    method __init__ (line 22) | def __init__(self, **kwargs):
    method voxelize (line 27) | def voxelize(self, points):
    method extract_pts_feat (line 49) | def extract_pts_feat(self, points, img_feats, img_metas):

FILE: mmdet3d/models/detectors/mvx_two_stage.py
  class MVXTwoStageDetector (line 21) | class MVXTwoStageDetector(Base3DDetector):
    method __init__ (line 24) | def __init__(self,
    method with_img_shared_head (line 106) | def with_img_shared_head(self):
    method with_pts_bbox (line 112) | def with_pts_bbox(self):
    method with_img_bbox (line 118) | def with_img_bbox(self):
    method with_img_backbone (line 124) | def with_img_backbone(self):
    method with_pts_backbone (line 129) | def with_pts_backbone(self):
    method with_fusion (line 134) | def with_fusion(self):
    method with_img_neck (line 140) | def with_img_neck(self):
    method with_pts_neck (line 145) | def with_pts_neck(self):
    method with_img_rpn (line 150) | def with_img_rpn(self):
    method with_img_roi_head (line 155) | def with_img_roi_head(self):
    method with_voxel_encoder (line 160) | def with_voxel_encoder(self):
    method with_middle_encoder (line 166) | def with_middle_encoder(self):
    method extract_img_feat (line 171) | def extract_img_feat(self, img, img_metas):
    method extract_pts_feat (line 191) | def extract_pts_feat(self, pts, img_feats, img_metas):
    method extract_feat (line 205) | def extract_feat(self, points, img, img_metas):
    method voxelize (line 213) | def voxelize(self, points):
    method forward_train (line 238) | def forward_train(self,
    method forward_pts_train (line 292) | def forward_pts_train(self,
    method forward_img_train (line 319) | def forward_img_train(self,
    method simple_test_img (line 373) | def simple_test_img(self, x, img_metas, proposals=None, rescale=False):
    method simple_test_rpn (line 384) | def simple_test_rpn(self, x, img_metas, rpn_test_cfg):
    method simple_test_pts (line 391) | def simple_test_pts(self, x, img_metas, rescale=False):
    method simple_test (line 402) | def simple_test(self, points, img_metas, img=None, rescale=False):
    method aug_test (line 420) | def aug_test(self, points, img_metas, imgs=None, rescale=False):
    method extract_feats (line 430) | def extract_feats(self, points, img_metas, imgs=None):
    method aug_test_pts (line 438) | def aug_test_pts(self, feats, img_metas, rescale=False):
    method show_results (line 457) | def show_results(self, data, result, out_dir):

FILE: mmdet3d/models/detectors/parta2.py
  class PartA2 (line 12) | class PartA2(TwoStage3DDetector):
    method __init__ (line 18) | def __init__(self,
    method extract_feat (line 43) | def extract_feat(self, points, img_metas):
    method voxelize (line 59) | def voxelize(self, points):
    method forward_train (line 89) | def forward_train(self,
    method simple_test (line 138) | def simple_test(self, points, img_metas, proposals=None, rescale=False):

FILE: mmdet3d/models/detectors/point_rcnn.py
  class PointRCNN (line 9) | class PointRCNN(TwoStage3DDetector):
    method __init__ (line 25) | def __init__(self,
    method extract_feat (line 44) | def extract_feat(self, points):
    method forward_train (line 59) | def forward_train(self, points, img_metas, gt_bboxes_3d, gt_labels_3d):
    method simple_test (line 110) | def simple_test(self, points, img_metas, imgs=None, rescale=False):

FILE: mmdet3d/models/detectors/sassd.py
  class SASSD (line 14) | class SASSD(SingleStage3DDetector):
    method __init__ (line 17) | def __init__(self,
    method extract_feat (line 41) | def extract_feat(self, points, img_metas=None, test_mode=False):
    method voxelize (line 55) | def voxelize(self, points):
    method forward_train (line 72) | def forward_train(self,
    method simple_test (line 104) | def simple_test(self, points, img_metas, imgs=None, rescale=False):
    method aug_test (line 116) | def aug_test(self, points, img_metas, imgs=None, rescale=False):

FILE: mmdet3d/models/detectors/single_stage.py
  class SingleStage3DDetector (line 7) | class SingleStage3DDetector(Base3DDetector):
    method __init__ (line 24) | def __init__(self,
    method forward_dummy (line 42) | def forward_dummy(self, points):
    method extract_feat (line 55) | def extract_feat(self, points, img_metas=None):
    method extract_feats (line 66) | def extract_feats(self, points, img_metas):

FILE: mmdet3d/models/detectors/single_stage_mono3d.py
  class SingleStageMono3DDetector (line 17) | class SingleStageMono3DDetector(SingleStageDetector):
    method __init__ (line 24) | def __init__(self,
    method extract_feats (line 46) | def extract_feats(self, imgs):
    method forward_train (line 51) | def forward_train(self,
    method simple_test (line 96) | def simple_test(self, img, img_metas, rescale=False):
    method aug_test (line 136) | def aug_test(self, imgs, img_metas, rescale=False):
    method show_results (line 206) | def show_results(self, data, result, out_dir, show=False, score_thr=No...

FILE: mmdet3d/models/detectors/smoke_mono3d.py
  class SMOKEMono3D (line 7) | class SMOKEMono3D(SingleStageMono3DDetector):
    method __init__ (line 13) | def __init__(self,

FILE: mmdet3d/models/detectors/ssd3dnet.py
  class SSD3DNet (line 7) | class SSD3DNet(VoteNet):
    method __init__ (line 13) | def __init__(self,

FILE: mmdet3d/models/detectors/two_stage.py
  class TwoStage3DDetector (line 10) | class TwoStage3DDetector(Base3DDetector, TwoStageDetector):
    method __init__ (line 18) | def __init__(self,

FILE: mmdet3d/models/detectors/votenet.py
  class VoteNet (line 10) | class VoteNet(SingleStage3DDetector):
    method __init__ (line 13) | def __init__(self,
    method forward_train (line 28) | def forward_train(self,
    method simple_test (line 63) | def simple_test(self, points, img_metas, imgs=None, rescale=False):
    method aug_test (line 86) | def aug_test(self, points, img_metas, imgs=None, rescale=False):

FILE: mmdet3d/models/detectors/voxelnet.py
  class VoxelNet (line 14) | class VoxelNet(SingleStage3DDetector):
    method __init__ (line 17) | def __init__(self,
    method extract_feat (line 40) | def extract_feat(self, points, img_metas=None):
    method voxelize (line 54) | def voxelize(self, points):
    method forward_train (line 71) | def forward_train(self,
    method simple_test (line 99) | def simple_test(self, points, img_metas, imgs=None, rescale=False):
    method aug_test (line 111) | def aug_test(self, points, img_metas, imgs=None, rescale=False):

FILE: mmdet3d/models/fbbev/detectors/bev_planner.py
  function generate_forward_transformation_matrix (line 25) | def generate_forward_transformation_matrix(bda, img_meta_dict=None):
  class BEVPlanner (line 34) | class BEVPlanner(CenterPoint):
    method __init__ (line 36) | def __init__(self,
    method _init_fuse_layers (line 172) | def _init_fuse_layers(self):
    method with_specific_component (line 199) | def with_specific_component(self, component_name):
    method image_encoder (line 203) | def image_encoder(self, img):
    method bev_encoder (line 231) | def bev_encoder(self, x):
    method fuse_history (line 244) | def fuse_history(self, curr_bev, img_metas, bda): # align features wit...
    method extract_img_bev_feat (line 409) | def extract_img_bev_feat(self, img, img_metas, **kwargs):
    method extract_lidar_bev_feat (line 472) | def extract_lidar_bev_feat(self, pts, img_feats, img_metas):
    method extract_feat (line 486) | def extract_feat(self, points, img, img_metas, **kwargs):
    method forward_train (line 496) | def forward_train(self,
    method forward_test (line 622) | def forward_test(self,
    method aug_test (line 666) | def aug_test(self,points,
    method simple_test (line 675) | def simple_test(self,
    method forward_dummy (line 836) | def forward_dummy(self,
    method world2bev_vis (line 847) | def world2bev_vis(self, x, y):
    method visual_sample (line 850) | def visual_sample(self, results, **kwargs):
    method _render_traj (line 985) | def _render_traj(self, future_traj, traj_score=1, colormap='winter', p...

FILE: mmdet3d/models/fbbev/heads/occupancy_head.py
  class OccHead (line 24) | class OccHead(BaseModule):
    method __init__ (line 25) | def __init__(
    method forward_coarse_voxel (line 146) | def forward_coarse_voxel(self, voxel_feats):
    method forward (line 185) | def forward(self, voxel_feats, img_feats=None, pts_feats=None, transfo...
    method forward_train (line 203) | def forward_train(self, voxel_feats, img_feats=None, pts_feats=None, t...
    method loss_voxel (line 214) | def loss_voxel(self, output_voxels, target_voxels, tag):
    method loss (line 260) | def loss(self, output_voxels=None,

FILE: mmdet3d/models/fbbev/heads/yolox.py
  class YOLOXHeadCustom (line 21) | class YOLOXHeadCustom(BaseDenseHead, BBoxTestMixin):
    method __init__ (line 51) | def __init__(self,
    method _init_layers (line 132) | def _init_layers(self):
    method _build_stacked_convs (line 148) | def _build_stacked_convs(self):
    method _build_predictor (line 172) | def _build_predictor(self):
    method init_weights (line 180) | def init_weights(self):
    method forward_single (line 190) | def forward_single(self, x, cls_convs, reg_convs, conv_cls, conv_reg,
    method forward (line 209) | def forward(self, feats):
    method _bbox_decode (line 236) | def _bbox_decode(self, priors, bbox_preds):
    method _centers2d_decode (line 248) | def _centers2d_decode(self, priors, centers2d):
    method _bboxes_nms (line 252) | def _bboxes_nms(self, cls_scores, bboxes, score_factor, cfg):
    method loss (line 267) | def loss(self,
    method _get_target_single (line 382) | def _get_target_single(self, cls_preds, objectness, priors, decoded_bb...
    method _get_l1_target (line 452) | def _get_l1_target(self, l1_target, gt_bboxes, priors, eps=1e-8):
    method _get_centers2d_target (line 459) | def _get_centers2d_target(self, centers2d_target, centers2d_labels, pr...

FILE: mmdet3d/models/fbbev/modules/depth_net.py
  function convert_color (line 25) | def convert_color(img_path):
  function save_tensor (line 32) | def save_tensor(tensor, path, pad_value=254.0,normalize=False):
  class NaiveDepthNet (line 48) | class NaiveDepthNet(BaseModule):
    method __init__ (line 58) | def __init__(
    method forward (line 78) | def forward(self, x, mlp_input=None):
    method get_mlp_input (line 100) | def get_mlp_input(self, rot, tran, intrin, post_rot, post_tran, bda):
  class _ASPPModule (line 105) | class _ASPPModule(nn.Module):
    method __init__ (line 107) | def __init__(self, inplanes, planes, kernel_size, padding, dilation,
    method forward (line 124) | def forward(self, x):
    method _init_weight (line 130) | def _init_weight(self):
  class ASPP (line 139) | class ASPP(nn.Module):
    method __init__ (line 141) | def __init__(self, inplanes, mid_channels=256, BatchNorm=nn.BatchNorm2d):
    method forward (line 189) | def forward(self, x):
    method _init_weight (line 205) | def _init_weight(self):
  class Mlp (line 214) | class Mlp(nn.Module):
    method __init__ (line 216) | def __init__(self,
    method forward (line 232) | def forward(self, x):
  class SELayer (line 241) | class SELayer(nn.Module):
    method __init__ (line 243) | def __init__(self, channels, act_layer=nn.ReLU, gate_layer=nn.Sigmoid):
    method forward (line 251) | def forward(self, x, x_se):
  class CM_DepthNet (line 259) | class CM_DepthNet(BaseModule):
    method __init__ (line 263) | def __init__(self,
    method forward (line 339) | def forward(self, x, mlp_input):
    method get_mlp_input (line 372) | def get_mlp_input(self, rot, tran, intrin, post_rot, post_tran, bda):
    method get_downsampled_gt_depth (line 399) | def get_downsampled_gt_depth(self, gt_depths):
    method get_depth_loss (line 440) | def get_depth_loss(self, depth_labels, depth_preds):
  class CM_ContextNet (line 459) | class CM_ContextNet(nn.Module):
    method __init__ (line 463) | def __init__(self,
    method forward (line 486) | def forward(self, x, mlp_input):

FILE: mmdet3d/models/fbbev/modules/fpn3d.py
  class FPN3D (line 15) | class FPN3D(BaseModule):
    method __init__ (line 28) | def __init__(self,
    method forward (line 71) | def forward(self, inputs):

FILE: mmdet3d/models/fbbev/modules/frpn.py
  class FRPN (line 16) | class FRPN(BaseModule):
    method __init__ (line 23) | def __init__(
    method forward (line 41) | def forward(self, input):
    method get_bev_mask_loss (line 49) | def get_bev_mask_loss(self, gt_bev_mask, pred_bev_mask):

FILE: mmdet3d/models/fbbev/modules/occ_loss_utils/focal_loss.py
  function py_sigmoid_focal_loss (line 12) | def py_sigmoid_focal_loss(pred,
  function py_focal_loss_with_prob (line 62) | def py_focal_loss_with_prob(pred,
  function sigmoid_focal_loss (line 115) | def sigmoid_focal_loss(pred,
  class CustomFocalLoss (line 163) | class CustomFocalLoss(nn.Module):
    method __init__ (line 165) | def __init__(self,
    method forward (line 206) | def forward(self,
  class CustomMSELoss (line 271) | class CustomMSELoss(nn.Module):
    method __init__ (line 279) | def __init__(self, reduction='mean', loss_weight=1.0):
    method forward (line 284) | def forward(self,

FILE: mmdet3d/models/fbbev/modules/occ_loss_utils/lovasz_softmax.py
  function lovasz_grad (line 22) | def lovasz_grad(gt_sorted):
  function iou_binary (line 37) | def iou_binary(preds, labels, EMPTY=1., ignore=None, per_image=True):
  function iou (line 57) | def iou(preds, labels, C, EMPTY=1., ignore=None, per_image=False):
  function lovasz_hinge (line 82) | def lovasz_hinge(logits, labels, per_image=True, ignore=None):
  function lovasz_hinge_flat (line 98) | def lovasz_hinge_flat(logits, labels):
  function flatten_binary_scores (line 118) | def flatten_binary_scores(scores, labels, ignore=None):
  class StableBCELoss (line 133) | class StableBCELoss(torch.nn.modules.Module):
    method __init__ (line 134) | def __init__(self):
    method forward (line 136) | def forward(self, input, target):
  function binary_xloss (line 142) | def binary_xloss(logits, labels, ignore=None):
  function lovasz_softmax (line 157) | def lovasz_softmax(probas, labels, classes='present', per_image=False, i...
  function lovasz_softmax_flat (line 176) | def lovasz_softmax_flat(probas, labels, classes='present'):
  function flatten_probas (line 207) | def flatten_probas(probas, labels, ignore=None):
  function xloss (line 236) | def xloss(logits, labels, ignore=None):
  function jaccard_loss (line 242) | def jaccard_loss(probas, labels,ignore=None, smooth = 100, bk_class = No...
  function hinge_jaccard_loss (line 270) | def hinge_jaccard_loss(probas, labels,ignore=None, classes = 'present', ...
  function isnan (line 307) | def isnan(x):
  function mean (line 311) | def mean(l, ignore_nan=False, empty=0):

FILE: mmdet3d/models/fbbev/modules/occ_loss_utils/nusc_param.py
  function KL_sep (line 116) | def KL_sep(p, target):
  function geo_scal_loss (line 126) | def geo_scal_loss(pred, ssc_target):
  function sem_scal_loss (line 153) | def sem_scal_loss(pred, ssc_target):
  function CE_ssc_loss (line 200) | def CE_ssc_loss(pred, target, class_weights):

FILE: mmdet3d/models/fbbev/modules/occ_loss_utils/semkitti.py
  function inverse_sigmoid (line 58) | def inverse_sigmoid(x, sign='A'):
  function KL_sep (line 68) | def KL_sep(p, target):
  function geo_scal_loss (line 78) | def geo_scal_loss(pred, ssc_target, ignore_index=255, non_empty_idx=0):
  function sem_scal_loss (line 108) | def sem_scal_loss(pred_, ssc_target, ignore_index=255):
  function CE_ssc_loss (line 166) | def CE_ssc_loss(pred, target, class_weights=None, ignore_index=255):
  function vel_loss (line 182) | def vel_loss(pred, gt):

FILE: mmdet3d/models/fbbev/modules/resnet3d.py
  function get_inplanes (line 16) | def get_inplanes():
  function conv3x3x3 (line 20) | def conv3x3x3(in_planes, out_planes, stride=1, use_spase_3dtensor=False):
  function conv1x1x1 (line 34) | def conv1x1x1(in_planes, out_planes, stride=1, use_spase_3dtensor=False):
  class BasicBlock (line 47) | class BasicBlock(BaseModule):
    method __init__ (line 50) | def __init__(self, in_planes, planes, stride=1, downsample=None, norm_...
    method forward (line 78) | def forward(self, x, debug=False):
  class Bottleneck (line 105) | class Bottleneck(BaseModule):
    method __init__ (line 108) | def __init__(self, in_planes, planes, stride=1, downsample=None, norm_...
    method forward (line 121) | def forward(self, x):
  class CustomResNet3D (line 144) | class CustomResNet3D(BaseModule):
    method __init__ (line 145) | def __init__(self,
    method _downsample_basic_block (line 211) | def _downsample_basic_block(self, x, planes, stride):
    method _make_layer (line 222) | def _make_layer(self, block, planes, blocks, shortcut_type, stride=1, ...
    method forward (line 251) | def forward(self, x):
  function generate_model (line 277) | def generate_model(model_depth, **kwargs):

FILE: mmdet3d/models/fbbev/motion_head/motion_head.py
  function get_ego_pos (line 21) | def get_ego_pos(points, pc_range):
  function get_rel_pos (line 28) | def get_rel_pos(points, pc_range):
  class MotionHead (line 36) | class MotionHead(BaseModule):
    method __init__ (line 67) | def __init__(self,
    method _init_layers (line 136) | def _init_layers(self):
    method forward (line 172) | def forward(self, agent_instances, preds_map_dicts, img_metas=None):
    method loss (line 247) | def loss(self,
    method get_bboxes (line 302) | def get_bboxes(self, preds_dicts, img_metas,  rescale=False):
  class MLN (line 336) | class MLN(nn.Module):
    method __init__ (line 343) | def __init__(self, c_dim, f_dim=256, use_ln=True):
    method init_weight (line 359) | def init_weight(self):
    method forward (line 365) | def forward(self, x, c):

FILE: mmdet3d/models/fbbev/motion_head/motion_planner_head.py
  function get_ego_pos (line 23) | def get_ego_pos(points, pc_range):
  function get_rel_pos (line 30) | def get_rel_pos(points, pc_range):
  class MotionPlannerHead (line 38) | class MotionPlannerHead(BaseModule):
    method __init__ (line 69) | def __init__(self,
    method _init_planer_layers (line 219) | def _init_planer_layers(self):
    method _init_layers (line 252) | def _init_layers(self):
    method pre_update_memory (line 287) | def pre_update_memory(self, data, fut_traj_from_velo):
    method post_update_memory (line 304) | def post_update_memory(self, data, ego_fut_trajs, ego_embeds):
    method forward (line 311) | def forward(self,
    method loss (line 479) | def loss(self,
    method get_motion (line 591) | def get_motion(self, preds_dicts, img_metas,  rescale=False):
    method get_traj (line 632) | def get_traj(self, preds_dicts, img_metas,  rescale=False, gt_ego_fut_...
  class MLN (line 707) | class MLN(nn.Module):
    method __init__ (line 714) | def __init__(self, c_dim, f_dim=256, use_ln=True):
    method init_weight (line 730) | def init_weight(self):
    method forward (line 736) | def forward(self, x, c):

FILE: mmdet3d/models/fbbev/motion_head/traj_loss.py
  class TrajLoss (line 16) | class TrajLoss(nn.Module):
    method __init__ (line 23) | def __init__(self, use_variance=False, cls_loss_weight=1., nll_loss_we...
    method forward (line 41) | def forward(self,
  function min_ade (line 97) | def min_ade(traj: torch.Tensor, traj_gt: torch.Tensor,
  function traj_nll (line 123) | def traj_nll(
  function min_fde (line 167) | def min_fde(traj: torch.Tensor, traj_gt: torch.Tensor,
  function miss_rate (line 203) | def miss_rate(

FILE: mmdet3d/models/fbbev/planner_head/AD_mlp.py
  function get_ego_pos (line 79) | def get_ego_pos(points, pc_range):
  function get_rel_pos (line 86) | def get_rel_pos(points, pc_range):
  class AD_MLP (line 94) | class AD_MLP(Base3DDetector):
    method __init__ (line 125) | def __init__(self,
    method _init_layers (line 182) | def _init_layers(self):
    method forward_train (line 198) | def forward_train(self,  img_metas=None, **kwargs):
    method inner_forward (line 212) | def inner_forward(self,  img_metas=None, **kwargs):
    method forward_test (line 269) | def forward_test(self, **kwargs):
    method loss (line 278) | def loss(self,
    method aug_test (line 308) | def aug_test(self): pass
    method point_sampling (line 311) | def point_sampling(self, reference_points, cam_params=None):
    method simple_test (line 341) | def simple_test(self, **kwargs):
    method extract_feat (line 386) | def extract_feat(self): pass
    method get_bboxes (line 388) | def get_bboxes(self, preds_dicts, img_metas=None,  rescale=False, gt_e...
    method world2bev_vis (line 489) | def world2bev_vis(self, x, y):
    method visual_sample (line 492) | def visual_sample(self, results, gt_bboxes_3d_=None, ego_info=None, ca...
    method _render_traj (line 670) | def _render_traj(self, future_traj, traj_score=1, colormap='winter', p...
    method _render_traj_v2 (line 685) | def _render_traj_v2(self, future_traj, traj_score=1, colormap='winter'...

FILE: mmdet3d/models/fbbev/planner_head/metric_stp3.py
  class PlanningMetric (line 15) | class PlanningMetric():
    method __init__ (line 16) | def __init__(self):
    method gen_dx_bx (line 39) | def gen_dx_bx(self, xbound, ybound, zbound):
    method calculate_birds_eye_view_parameters (line 46) | def calculate_birds_eye_view_parameters(self, x_bounds, y_bounds, z_bo...
    method evaluate_single_coll (line 69) | def evaluate_single_coll(self, traj, segmentation, input_gt, gt_traj=N...
    method evaluate_coll (line 165) | def evaluate_coll(
    method compute_L2 (line 223) | def compute_L2(self, trajs, gt_trajs):

FILE: mmdet3d/models/fbbev/planner_head/naive_planner.py
  function get_ego_pos (line 30) | def get_ego_pos(points, pc_range):
  function get_rel_pos (line 37) | def get_rel_pos(points, pc_range):
  class NaivePlannerHead (line 45) | class NaivePlannerHead(BaseModule):
    method __init__ (line 49) | def __init__(self,
    method reset_memory (line 157) | def reset_memory(self):
    method pre_update_memory (line 161) | def pre_update_memory(self, data, fut_traj_from_velo):
    method post_update_memory (line 177) | def post_update_memory(self, data, ego_fut_trajs, ego_embeds):
    method _init_layers (line 183) | def _init_layers(self):
    method calc_MDE (line 202) | def calc_MDE(self, reference_points_q, reference_points_v, pc_range, m...
    method forward (line 231) | def forward(self, results, gt_ego_lcf_feat, gt_ego_fut_cmd, gt_ego_his...
    method loss (line 323) | def loss(self,
    method get_bboxes (line 357) | def get_bboxes(self, preds_dicts, img_metas,  rescale=False, gt_ego_fu...
  class MLN (line 460) | class MLN(nn.Module):
    method __init__ (line 467) | def __init__(self, c_dim, f_dim=256, use_ln=True):
    method init_weight (line 483) | def init_weight(self):
    method forward (line 489) | def forward(self, x, c):

FILE: mmdet3d/models/fbbev/planner_head/plan_loss.py
  class PlanMapBoundLoss (line 12) | class PlanMapBoundLoss(nn.Module):
    method __init__ (line 25) | def __init__(
    method forward (line 44) | def forward(self,
  function plan_map_bound_loss (line 92) | def plan_map_bound_loss(pred, target, dis_thresh=1.0):
  function segments_intersect (line 149) | def segments_intersect(line1_start, line1_end, line2_start, line2_end):
  class PlanCollisionLoss (line 182) | class PlanCollisionLoss(nn.Module):
    method __init__ (line 195) | def __init__(
    method forward (line 212) | def forward(self,
  function plan_col_loss (line 264) | def plan_col_loss(
  class PlanMapDirectionLoss (line 322) | class PlanMapDirectionLoss(nn.Module):
    method __init__ (line 333) | def __init__(
    method forward (line 350) | def forward(self,
  function plan_map_dir_loss (line 394) | def plan_map_dir_loss(pred, target, dis_thresh=2.0):
  class PlanMapDirectionLoss2 (line 454) | class PlanMapDirectionLoss2(nn.Module):
    method __init__ (line 465) | def __init__(
    method forward (line 482) | def forward(self,
  function plan_map_dir_loss2 (line 526) | def plan_map_dir_loss2(pred, target, dis_thresh=2.0):

FILE: mmdet3d/models/fbbev/planner_head/plan_loss_gt.py
  class PlanMapBoundLoss_gt (line 16) | class PlanMapBoundLoss_gt(nn.Module):
    method __init__ (line 29) | def __init__(
    method forward (line 48) | def forward(self,
  function plan_map_bound_loss (line 93) | def plan_map_bound_loss(pred, target, dis_thresh=1.0):
  function segments_intersect (line 150) | def segments_intersect(line1_start, line1_end, line2_start, line2_end):
  class PlanCollisionLoss_gt (line 183) | class PlanCollisionLoss_gt(nn.Module):
    method __init__ (line 196) | def __init__(
    method forward (line 213) | def forward(self,
  function plan_col_loss (line 265) | def plan_col_loss(
  class PlanMapDirectionLoss_gt (line 322) | class PlanMapDirectionLoss_gt(nn.Module):
    method __init__ (line 333) | def __init__(
    method forward (line 350) | def forward(self,
  function plan_map_dir_loss (line 394) | def plan_map_dir_loss(pred, target, dis_thresh=2.0):

FILE: mmdet3d/models/fbbev/streammapnet/CustomMSDeformableAttention.py
  class CustomMSDeformableAttention (line 32) | class CustomMSDeformableAttention(BaseModule):
    method __init__ (line 59) | def __init__(self,
    method init_weights (line 111) | def init_weights(self):
    method forward (line 134) | def forward(self,

FILE: mmdet3d/models/fbbev/streammapnet/cost.py
  function chamfer_distance (line 8) | def chamfer_distance(line1, line2) -> float:
  class ClsSigmoidCost (line 28) | class ClsSigmoidCost:
    method __init__ (line 34) | def __init__(self, weight=1.):
    method __call__ (line 37) | def __call__(self, cls_pred, gt_labels):
  class LinesFixNumChamferCost (line 56) | class LinesFixNumChamferCost(object):
    method __init__ (line 62) | def __init__(self, weight=1.0, permute=False):
    method __call__ (line 66) | def __call__(self, lines_pred, gt_lines):
  class LinesL1Cost (line 110) | class LinesL1Cost(object):
    method __init__ (line 116) | def __init__(self, weight=1.0, beta=0.0, permute=False):
    method __call__ (line 121) | def __call__(self, lines_pred, gt_lines, **kwargs):
  class BBoxCostC (line 168) | class BBoxCostC:
    method __init__ (line 184) | def __init__(self, weight=1., box_format='xyxy'):
    method __call__ (line 189) | def __call__(self, bbox_pred, gt_bboxes):
  class IoUCostC (line 209) | class IoUCostC:
    method __init__ (line 225) | def __init__(self, iou_mode='giou', weight=1., box_format='xywh'):
    method __call__ (line 231) | def __call__(self, bboxes, gt_bboxes):
  class DynamicLinesCost (line 253) | class DynamicLinesCost(object):
    method __init__ (line 259) | def __init__(self, weight=1.):
    method __call__ (line 262) | def __call__(self, lines_pred, lines_gt, masks_pred, masks_gt):
    method cal_dist (line 284) | def cal_dist(self, x1, x2):
    method get_dynamic_line (line 301) | def get_dynamic_line(self, mat, m1, m2):
  class BBoxLogitsCost (line 327) | class BBoxLogitsCost(object):
    method __init__ (line 333) | def __init__(self, weight=1.):
    method calNLL (line 336) | def calNLL(self, logits, value):
    method __call__ (line 353) | def __call__(self, bbox_pred, bbox_gt, **kwargs):
  class MapQueriesCost (line 368) | class MapQueriesCost(object):
    method __init__ (line 370) | def __init__(self, cls_cost, reg_cost, iou_cost=None):
    method __call__ (line 379) | def __call__(self, preds: dict, gts: dict):

FILE: mmdet3d/models/fbbev/streammapnet/fp16_dattn.py
  class MultiScaleDeformableAttentionFp16 (line 36) | class MultiScaleDeformableAttentionFp16(BaseModule):
    method __init__ (line 38) | def __init__(self, attn_cfg=None,init_cfg=None,**kwarg):
    method forward (line 48) | def forward(self, query,
  class MultiScaleDeformableAttnFunctionFp32 (line 71) | class MultiScaleDeformableAttnFunctionFp32(Function):
    method forward (line 75) | def forward(ctx, value, value_spatial_shapes, value_level_start_index,
    method backward (line 112) | def backward(ctx, grad_output):
  function multi_scale_deformable_attn_pytorch (line 143) | def multi_scale_deformable_attn_pytorch(value, value_spatial_shapes,
  class MultiScaleDeformableAttentionFP32 (line 202) | class MultiScaleDeformableAttentionFP32(BaseModule):
    method __init__ (line 227) | def __init__(self,
    method init_weights (line 275) | def init_weights(self):
    method forward (line 297) | def forward(self,

FILE: mmdet3d/models/fbbev/streammapnet/hungarian_lines_assigner.py
  class HungarianLinesAssigner (line 10) | class HungarianLinesAssigner(BaseAssigner):
    method __init__ (line 29) | def __init__(self,
    method assign (line 39) | def assign(self,

FILE: mmdet3d/models/fbbev/streammapnet/loss.py
  class LinesL1Loss (line 12) | class LinesL1Loss(nn.Module):
    method __init__ (line 14) | def __init__(self, reduction='mean', loss_weight=1.0, beta=0.5):
    method forward (line 28) | def forward(self,
  function bce (line 69) | def bce(pred, label, class_weight=None):
  class MasksLoss (line 86) | class MasksLoss(nn.Module):
    method __init__ (line 88) | def __init__(self, reduction='mean', loss_weight=1.0):
    method forward (line 93) | def forward(self,
  function ce (line 114) | def ce(pred, label, class_weight=None):
  class LenLoss (line 130) | class LenLoss(nn.Module):
    method __init__ (line 132) | def __init__(self, reduction='mean', loss_weight=1.0):
    method forward (line 137) | def forward(self,

FILE: mmdet3d/models/fbbev/streammapnet/map_utils.py
  function normalize_2d_bbox (line 3) | def normalize_2d_bbox(bboxes, pc_range):
  function normalize_2d_pts (line 15) | def normalize_2d_pts(pts, pc_range):
  function denormalize_2d_bbox (line 25) | def denormalize_2d_bbox(bboxes, pc_range):
  function denormalize_2d_pts (line 35) | def denormalize_2d_pts(pts, pc_range):

FILE: mmdet3d/models/fbbev/streammapnet/streammapnet_head.py
  class MapDetectorHead (line 20) | class MapDetectorHead(nn.Module):
    method __init__ (line 22) | def __init__(self,
    method init_weights (line 100) | def init_weights(self):
    method _init_embedding (line 137) | def _init_embedding(self):
    method _init_branch (line 151) | def _init_branch(self,):
    method _prepare_context (line 184) | def _prepare_context(self, bev_features):
    method propagate (line 198) | def propagate(self, query_embedding, img_metas, start_of_sequence, ego...
    method forward_train (line 280) | def forward_train(self, input_dict, img_metas, map_gt_bboxes_3d, map_g...
    method forward_test (line 426) | def forward_test(self, input_dict, img_metas, map_gt_bboxes_3d=None, m...
    method world2bev_vis (line 562) | def world2bev_vis(self, x, y):
    method visual_sample (line 565) | def visual_sample(self, lines, index, pre=False, **kwargs):
    method _get_target_single (line 592) | def _get_target_single(self,
    method get_targets (line 675) | def get_targets(self, preds, map_gt_bboxes_3d,
    method loss_single (line 730) | def loss_single(self,
    method loss (line 806) | def loss(self,
    method get_bboxes (line 866) | def get_bboxes(self, preds_dict, img_metas, thr=0.0):
    method train (line 920) | def train(self, *args, **kwargs):
    method eval (line 926) | def eval(self):
    method forward (line 932) | def forward(self, *args, return_loss=True, **kwargs):

FILE: mmdet3d/models/fbbev/streammapnet/transformer.py
  class MapTransformerDecoder_new (line 24) | class MapTransformerDecoder_new(BaseModule):
    method __init__ (line 32) | def __init__(self,
    method forward (line 59) | def forward(self,
  class MapTransformerLayer (line 160) | class MapTransformerLayer(BaseTransformerLayer):
    method __init__ (line 196) | def __init__(self,
    method forward (line 222) | def forward(self,
  class MapTransformer (line 328) | class MapTransformer(Transformer):
    method __init__ (line 339) | def __init__(self,
    method init_layers (line 351) | def init_layers(self):
    method init_weights (line 356) | def init_weights(self):
    method forward (line 367) | def forward(self,
  class PlaceHolderEncoder (line 475) | class PlaceHolderEncoder(nn.Module):
    method __init__ (line 477) | def __init__(self, *args, embed_dims=None, **kwargs):
    method forward (line 481) | def forward(self, *args, query=None, **kwargs):

FILE: mmdet3d/models/fbbev/streammapnet/utils.py
  class StreamTensorMemory (line 10) | class StreamTensorMemory(object):
    method __init__ (line 11) | def __init__(self, batch_size):
    method memory_list (line 23) | def memory_list(self):
    method img_metas_memory (line 30) | def img_metas_memory(self):
    method update (line 36) | def update(self, memory, img_metas):
    method reset_single (line 41) | def reset_single(self, idx):
    method get (line 45) | def get(self, img_metas):
    method train (line 75) | def train(self, mode=True):
    method eval (line 82) | def eval(self):
  class MotionMLP (line 87) | class MotionMLP(nn.Module):
    method __init__ (line 94) | def __init__(self, c_dim, f_dim=512, identity=True):
    method init_weights (line 108) | def init_weights(self):
    method forward (line 117) | def forward(self, x, c):

FILE: mmdet3d/models/fbbev/streampetr/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: mmdet3d/models/fbbev/streampetr/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,

FILE: mmdet3d/models/fbbev/streampetr/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: mmdet3d/models/fbbev/streampetr/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 90) | def decode(self, preds_dicts, layer_index=-1):

FILE: mmdet3d/models/fbbev/streampetr/petr_transformer.py
  function get_ego_pos (line 29) | def get_ego_pos(points, pc_range):
  function get_rel_pos (line 36) | def get_rel_pos(points, pc_range):
  class Detr3DTransformer (line 44) | class Detr3DTransformer(BaseModule):
    method __init__ (line 55) | def __init__(self,
    method init_weights (line 61) | def init_weights(self):
    method forward (line 70) | def forward(self,
  class Detr3DTransformerDecoder (line 153) | class Detr3DTransformerDecoder(TransformerLayerSequence):
    method __init__ (line 161) | def __init__(self, embed_dims, *args,  predict_refine=True, **kwargs):
    method forward (line 165) | def forward(self,
  class Detr3DTemporalDecoderLayer (line 254) | class Detr3DTemporalDecoderLayer(BaseModule):
    method __init__ (line 290) | def __init__(self,
    method _forward (line 372) | def _forward(self,
    method forward (line 495) | def forward(self,
  class DeformableFeatureAggregationCuda (line 564) | class DeformableFeatureAggregationCuda(BaseModule):
    method __init__ (line 565) | def __init__(
    method init_weight (line 602) | def init_weight(self):
    method forward (line 608) | def forward(self, instance_feature, query_pos, feat_flatten, reference...
    method _get_weights (line 626) | def _get_weights(self, instance_feature, anchor_embed, lidar2img_mat):
    method feature_sampling (line 645) | def feature_sampling(self, feat_flatten, spatial_flatten, level_start_...
  class DeformableFeatureAggregationCuda_v2 (line 675) | class DeformableFeatureAggregationCuda_v2(BaseModule):
    method __init__ (line 676) | def __init__(
    method init_weight (line 712) | def init_weight(self):
    method forward (line 718) | def forward(self, instance_feature, query_pos, feat_flatten, reference...
    method _get_weights (line 735) | def _get_weights(self, instance_feature, anchor_embed, lidar2img_mat):
    method feature_sampling (line 750) | def feature_sampling(self, feat_flatten, spatial_flatten, level_start_...
  class MVDeformableFeatureAggregationCuda (line 782) | class MVDeformableFeatureAggregationCuda(BaseModule):
    method __init__ (line 783) | def __init__(
    method init_weight (line 815) | def init_weight(self):
    method forward (line 820) | def forward(self, instance_feature, query_pos, feat_flatten, reference...
    method _get_weights (line 833) | def _get_weights(self, instance_feature, anchor_embed, lidar2img_mat, ...
    method feature_sampling (line 848) | def feature_sampling(self, feat_flatten, spatial_flatten, level_start_...
  class SparseBEVSelfAttention (line 909) | class SparseBEVSelfAttention(BaseModule):
    method __init__ (line 910) | def __init__(self, embed_dims=256, num_heads=8, dropout=0.1, pc_range=...
    method init_weights (line 918) | def init_weights(self):
    method forward (line 922) | def forward(self,
    method calc_points_dists (line 952) | def calc_points_dists(self, reference_points, temp_reference_points, p...
  class MotionSelfAttention (line 969) | class MotionSelfAttention(BaseModule):
    method __init__ (line 970) | def __init__(self, embed_dims=256, num_heads=8, dropout=0.1, pc_range=...
    method init_weights (line 984) | def init_weights(self):
    method forward (line 988) | def forward(self,
    method calc_ADE (line 1026) | def calc_ADE(self, reference_points_q, reference_points_v, pc_range, *...
    method calc_MDE (line 1040) | def calc_MDE(self, reference_points_q, reference_points_v, pc_range, m...
    method calc_MDE_v2 (line 1069) | def calc_MDE_v2(self, reference_points_q, reference_points_v, pc_range...
  class CustomTransformerDecoder (line 1097) | class CustomTransformerDecoder(TransformerLayerSequence):
    method __init__ (line 1104) | def __init__(self, *args, return_intermediate=False, **kwargs):
    method forward (line 1109) | def forward(self,

FILE: mmdet3d/models/fbbev/streampetr/streampetr_utils.py
  function normalize_bbox (line 3) | def normalize_bbox(bboxes, pc_range):
  function denormalize_bbox (line 37) | def denormalize_bbox(normalized_bboxes, pc_range):
  function pos2posemb3d (line 66) | def pos2posemb3d(pos, num_pos_feats=128, temperature=10000):
  function bevpos2posemb (line 80) | def bevpos2posemb(pos, num_pos_feats=128, temperature=10000):
  function pos2posemb1d (line 92) | def pos2posemb1d(pos, num_pos_feats=256, temperature=10000):
  function nerf_positional_encoding (line 103) | def nerf_positional_encoding(
  function memory_refresh (line 154) | def memory_refresh(memory, prev_exist):
  function topk_gather (line 160) | def topk_gather(feat, topk_indexes):
  function apply_ltrb (line 173) | def apply_ltrb(locations, pred_ltrb):
  function apply_center_offset (line 192) | def apply_center_offset(locations, center_offset):
  function locations (line 206) | def locations(features, stride, pad_h, pad_w):
  function gaussian_2d (line 236) | def gaussian_2d(shape, sigma=1.0):
  function draw_heatmap_gaussian (line 255) | def draw_heatmap_gaussian(heatmap, center, radius, k=1):
  class SELayer_Linear (line 286) | class SELayer_Linear(nn.Module):
    method __init__ (line 287) | def __init__(self, channels, act_layer=nn.ReLU, gate_layer=nn.Sigmoid):
    method forward (line 294) | def forward(self, x, x_se):
  class MLN (line 301) | class MLN(nn.Module):
    method __init__ (line 308) | def __init__(self, c_dim, f_dim=256):
    method reset_parameters (line 322) | def reset_parameters(self):
    method forward (line 328) | def forward(self, x, c):
  function transform_reference_points (line 338) | def transform_reference_points(reference_points, egopose, reverse=False,...

FILE: mmdet3d/models/fbbev/streampetr/streampetr_v2.py
  class SparseHead4BEV (line 27) | class SparseHead4BEV(AnchorFreeHead):
    method __init__ (line 58) | def __init__(self,
    method _init_layers (line 209) | def _init_layers(self):
    method temporal_alignment (line 264) | def temporal_alignment(self, query_pos, tgt, reference_points):
    method prepare_for_dn (line 300) | def prepare_for_dn(self, batch_size, reference_points, img_metas, gt_b...
    method init_weights (line 385) | def init_weights(self):
    method reset_memory (line 398) | def reset_memory(self):
    method pre_update_memory (line 405) | def pre_update_memory(self, data):
    method post_update_memory (line 433) | def post_update_memory(self, data, rec_ego_pose, all_cls_scores, all_b...
    method forward (line 471) | def forward(self, input_dict, img_metas,  gt_bboxes_3d=None, gt_labels...
    method prepare_for_loss (line 608) | def prepare_for_loss(self, mask_dict):
    method _get_target_single (line 627) | def _get_target_single(self,
    method get_targets (line 687) | def get_targets(self,
    method loss_single (line 738) | def loss_single(self,
    method dn_loss_single (line 807) | def dn_loss_single(self,
    method loss (line 865) | def loss(self,
    method get_bboxes (line 975) | def get_bboxes(self, preds_dicts, img_metas, rescale=False):
  class MLN (line 998) | class MLN(nn.Module):
    method __init__ (line 1005) | def __init__(self, c_dim, f_dim=256, use_ln=True):
    method init_weight (line 1021) | def init_weight(self):
    method forward (line 1027) | def forward(self, x, c):

FILE: mmdet3d/models/fbbev/track_head/instances.py
  function topk_gather (line 16) | def topk_gather(feat, topk_indexes):
  class Instances (line 28) | class Instances:
    method __init__ (line 51) | def __init__(self, image_size: Tuple[int, int], **kwargs: Any):
    method image_size (line 63) | def image_size(self) -> Tuple[int, int]:
    method __setattr__ (line 70) | def __setattr__(self, name: str, val: Any) -> None:
    method __getattr__ (line 76) | def __getattr__(self, name: str) -> Any:
    method set (line 81) | def set(self, name: str, value: Any) -> None:
    method has (line 94) | def has(self, name: str) -> bool:
    method remove (line 101) | def remove(self, name: str) -> None:
    method get (line 107) | def get(self, name: str) -> Any:
    method get_fields (line 113) | def get_fields(self) -> Dict[str, Any]:
    method to (line 122) | def to(self, *args: Any, **kwargs: Any) -> "Instances":
    method numpy (line 134) | def numpy(self):
    method instances_topk_gather (line 143) | def instances_topk_gather(self, topk_indexes, valid_key_set=None):
    method __getitem__ (line 153) | def __getitem__(self, item: Union[int, slice, torch.BoolTensor]) -> "I...
    method __len__ (line 183) | def __len__(self) -> int:
    method __iter__ (line 189) | def __iter__(self):
    method cat (line 193) | def cat(instance_lists: List["Instances"], dim=0) -> "Instances":
    method clone (line 229) | def clone(self):
    method detach (line 237) | def detach(self):
    method __str__ (line 246) | def __str__(self) -> str:

FILE: mmdet3d/models/fbbev/track_head/losses/tracking_loss.py
  class TrackingLoss (line 19) | class TrackingLoss(TrackingLossBase):
    method __init__ (line 20) | def __init__(self,
    method loss_single_frame (line 26) | def loss_single_frame(self,

FILE: mmdet3d/models/fbbev/track_head/losses/tracking_loss_base.py
  class TrackingLossBase (line 24) | class TrackingLossBase(nn.Module):
    method __init__ (line 27) | def __init__(self,
    method _get_target_single (line 102) | def _get_target_single(self,
    method get_targets (line 166) | def get_targets(self,
    method loss_single_decoder (line 218) | def loss_single_decoder(self,
    method loss_single_frame (line 299) | def loss_single_frame(self,
    method forward (line 383) | def forward(self,
  function nan_to_num (line 400) | def nan_to_num(x, nan=0.0, posinf=None, neginf=None):

FILE: mmdet3d/models/fbbev/track_head/losses/tracking_loss_combo.py
  class TrackingLossCombo (line 26) | class TrackingLossCombo(TrackingLoss):
    method __init__ (line 29) | def __init__(self,
    method loss_prediction (line 58) | def loss_prediction(self,
    method loss_mem_bank (line 71) | def loss_mem_bank(self,
    method forward (line 140) | def forward(self,
  function nan_to_num (line 158) | def nan_to_num(x, nan=0.0, posinf=None, neginf=None):

FILE: mmdet3d/models/fbbev/track_head/losses/tracking_loss_mem_bank.py
  class TrackingLossMemBank (line 16) | class TrackingLossMemBank(TrackingLoss):
    method __init__ (line 17) | def __init__(self,
    method loss_mem_bank (line 43) | def loss_mem_bank(self,
    method forward (line 112) | def forward(self,

FILE: mmdet3d/models/fbbev/track_head/losses/tracking_loss_prediction.py
  class TrackingLossPrediction (line 20) | class TrackingLossPrediction(TrackingLoss):
    method __init__ (line 23) | def __init__(self,
    method loss_prediction (line 49) | def loss_prediction(self,
    method forward (line 63) | def forward(self,
  function nan_to_num (line 81) | def nan_to_num(x, nan=0.0, posinf=None, neginf=None):

FILE: mmdet3d/models/fbbev/track_head/runtime_tracker.py
  class RunTimeTracker (line 9) | class RunTimeTracker:
    method __init__ (line 10) | def __init__(self, output_threshold=0.2, score_threshold=0.4,
    method update_active_tracks (line 28) | def update_active_tracks(self, track_instances, active_mask):
    method get_active_mask (line 46) | def get_active_mask(self, track_instances, training=True):
    method empty (line 51) | def empty(self):

FILE: mmdet3d/models/fbbev/track_head/streampetr_utils.py
  function normalize_bbox (line 3) | def normalize_bbox(bboxes, pc_range):
  function denormalize_bbox (line 37) | def denormalize_bbox(normalized_bboxes, pc_range):
  function pos2posemb3d (line 66) | def pos2posemb3d(pos, num_pos_feats=128, temperature=10000):
  function bevpos2posemb (line 80) | def bevpos2posemb(pos, num_pos_feats=128, temperature=10000):
  function pos2posemb1d (line 92) | def pos2posemb1d(pos, num_pos_feats=256, temperature=10000):
  function nerf_positional_encoding (line 103) | def nerf_positional_encoding(
  function memory_refresh (line 154) | def memory_refresh(memory, prev_exist):
  function topk_gather (line 160) | def topk_gather(feat, topk_indexes):
  function apply_ltrb (line 173) | def apply_ltrb(locations, pred_ltrb):
  function apply_center_offset (line 192) | def apply_center_offset(locations, center_offset):
  function locations (line 206) | def locations(features, stride, pad_h, pad_w):
  function gaussian_2d (line 236) | def gaussian_2d(shape, sigma=1.0):
  function draw_heatmap_gaussian (line 255) | def draw_heatmap_gaussian(heatmap, center, radius, k=1):
  class SELayer_Linear (line 286) | class SELayer_Linear(nn.Module):
    method __init__ (line 287) | def __init__(self, channels, act_layer=nn.ReLU, gate_layer=nn.Sigmoid):
    method forward (line 294) | def forward(self, x, x_se):
  class MLN (line 301) | class MLN(nn.Module):
    method __init__ (line 308) | def __init__(self, c_dim, f_dim=256):
    method reset_parameters (line 322) | def reset_parameters(self):
    method forward (line 328) | def forward(self, x, c):
  function transform_reference_points (line 338) | def transform_reference_points(reference_points, egopose, reverse=False,...
  function transform_velo (line 355) | def transform_velo(velo, egopose, reverse=False, translation=False):

FILE: mmdet3d/models/fbbev/track_head/track_nms_free_coder.py
  class TrackNMSFreeCoder (line 18) | class TrackNMSFreeCoder(BaseBBoxCoder):
    method __init__ (line 30) | def __init__(self,
    method encode (line 47) | def encode(self):
    method decode_single (line 50) | def decode_single(self, cls_scores, bbox_preds, obj_idxes=None, track_...
    method decode (line 155) | def decode(self, preds_dicts, layer_index=-1):

FILE: mmdet3d/models/fbbev/track_head/trackpetr.py
  class TackerHead (line 29) | class TackerHead(AnchorFreeHead):
    method __init__ (line 60) | def __init__(self,
    method _init_layers (line 182) | def _init_layers(self):
    method init_weights (line 237) | def init_weights(self):
    method reset_history_track_instances (line 251) | def reset_history_track_instances(self):
    method generate_empty_instance (line 254) | def generate_empty_instance(self, B, init_memory_instances=False):
    method instance_temporal_alignment (line 375) | def instance_temporal_alignment(self):
    method pre_update_instances (line 409) | def pre_update_instances(self, data):
    method post_update_instances (line 452) | def post_update_instances(self, data, rec_ego_pose, all_cls_scores, al...
    method post_merge_instances (line 485) | def post_merge_instances(self, data, kept_indicator=0):
    method forward (line 518) | def forward(self, input_dict, img_metas,  gt_bboxes_3d=None, gt_labels...
    method loss (line 646) | def loss(self,
    method get_targets (line 690) | def get_targets(self):
    method forward_tracking (line 693) | def forward_tracking(self, input_dict, img_metas):
    method get_bboxes (line 723) | def get_bboxes(self, preds_dicts, img_metas, rescale=False):
  class MLN (line 749) | class MLN(nn.Module):
    method __init__ (line 756) | def __init__(self, c_dim, f_dim=256, use_ln=True):
    method init_weight (line 772) | def init_weight(self):
    method forward (line 778) | def forward(self, x, c):

FILE: mmdet3d/models/fbbev/track_head/utils.py
  class StreamTensorMemory (line 10) | class StreamTensorMemory(object):
    method __init__ (line 11) | def __init__(self, batch_size):
    method memory_list (line 23) | def memory_list(self):
    method img_metas_memory (line 30) | def img_metas_memory(self):
    method update (line 36) | def update(self, memory, img_metas):
    method reset_single (line 41) | def reset_single(self, idx):
    method get (line 45) | def get(self, img_metas):
    method train (line 75) | def train(self, mode=True):
    method eval (line 82) | def eval(self):
  class MotionMLP (line 87) | class MotionMLP(nn.Module):
    method __init__ (line 94) | def __init__(self, c_dim, f_dim=512, identity=True):
    method init_weights (line 108) | def init_weights(self):
    method forward (line 117) | def forward(self, x, c):

FILE: mmdet3d/models/fbbev/utils/bricks.py
  function compute_allocation (line 12) | def compute_allocation(obj) -> int:
  function convert_color (line 44) | def convert_color(img_path):
  function save_tensor (line 51) | def save_tensor(tensor, path, pad_value=254.0,normalize=False):
  function run_time (line 71) | def run_time(name):

FILE: mmdet3d/models/fbbev/utils/draw_bbox.py
  function plot_rect3d_on_img (line 22) | def plot_rect3d_on_img(img,
  function draw_lidar_bbox3d_on_img (line 95) | def draw_lidar_bbox3d_on_img(bboxes3d,
  function show_multi_modality_result (line 171) | def show_multi_modality_result(img,

FILE: mmdet3d/models/fbbev/utils/eval_hook.py
  function _calc_dynamic_intervals (line 20) | def _calc_dynamic_intervals(start_interval, dynamic_interval_list):
  class CustomDistEvalHook (line 32) | class CustomDistEvalHook(BaseDistEvalHook):
    method __init__ (line 34) | def __init__(self, *args, dynamic_intervals=None,  work_dir='test', **...
    method _decide_interval (line 41) | def _decide_interval(self, runner):
    method before_train_epoch (line 48) | def before_train_epoch(self, runner):
    method before_train_iter (line 53) | def before_train_iter(self, runner):
    method _do_evaluate (line 57) | def _do_evaluate(self, runner):
    method evaluate (line 122) | def evaluate(self, runner, results):

FILE: mmdet3d/models/fbbev/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):

FILE: mmdet3d/models/fbbev/utils/timer_cp.py
  class TimerCP (line 7) | class TimerCP(CheckpointHook):
    method __init__ (line 11) | def __init__(self, period=14400):
    method before_run (line 16) | def before_run(self, runner):
    method after_train_epoch (line 20) | def after_train_epoch(self, runner):
    method before_train_iter (line 23) | def before_train_iter(self, runner):
    method _save_checkpoint (line 35) | def _save_checkpoint(self, runner):

FILE: mmdet3d/models/fbbev/utils/wechat_logger.py
  class GradChecker (line 22) | class GradChecker(Hook):
    method after_train_iter (line 23) | def after_train_iter(self, runner):
  class MyWechatLoggerHook (line 45) | class MyWechatLoggerHook(LoggerHook):
    method __init__ (line 62) | def __init__(self,
    method before_run (line 77) | def before_run(self, runner) -> None:
    method get_table_text (line 81) | def get_table_text(self, runner, tags):
    method log (line 97) | def log(self, runner) -> None:
    method _send (line 117) | def _send(self, runner, text) -> None:
    method after_run (line 140) | def after_run(self, runner) -> None:

FILE: mmdet3d/models/fbbev/view_transformation/backward_projection/backward_projection.py
  class BackwardProjection (line 35) | class BackwardProjection(BaseModule):
    method __init__ (line 47) | def __init__(self,
    method _init_layers (line 76) | def _init_layers(self):
    method init_weights (line 80) | def init_weights(self):
    method forward (line 85) | def forward(self, mlvl_feats, img_metas, lss_bev=None, gt_bboxes_3d=No...

FILE: mmdet3d/models/fbbev/view_transformation/backward_projection/bevformer_utils/bevformer.py
  class BEVFormer (line 23) | class BEVFormer(BaseModule):
    method __init__ (line 34) | def __init__(self,
    method init_layers (line 52) | def init_layers(self):
    method init_weights (line 58) | def init_weights(self):
    method forward (line 72) | def forward(

FILE: mmdet3d/models/fbbev/view_transformation/backward_projection/bevformer_utils/bevformer_encoder.py
  class bevformer_encoder (line 28) | class bevformer_encoder(TransformerLayerSequence):
    method __init__ (line 39) | def __init__(self, *args, pc_range=None, grid_config=None, data_config...
    method get_reference_points (line 52) | def get_reference_points(self,H, W, Z=8, dim='3d', bs=1, device='cuda'...
    method point_sampling (line 92) | def point_sampling(self, reference_points, pc_range,  img_metas, cam_p...
    method forward (line 124) | def forward(self,
  class BEVFormerEncoderLayer (line 207) | class BEVFormerEncoderLayer(MyCustomBaseTransformerLayer):
    method __init__ (line 228) | def __init__(self,
    method forward (line 251) | def forward(self,

FILE: mmdet3d/models/fbbev/view_transformation/backward_projection/bevformer_utils/custom_base_transformer_layer.py
  class MyCustomBaseTransformerLayer (line 38) | class MyCustomBaseTransformerLayer(BaseModule):
    method __init__ (line 72) | def __init__(self,
    method forward (line 167) | def forward(self,

FILE: mmdet3d/models/fbbev/view_transformation/backward_projection/bevformer_utils/multi_scale_deformable_attn_function.py
  class MultiScaleDeformableAttnFunction_fp16 (line 15) | class MultiScaleDeformableAttnFunction_fp16(Function):
    method forward (line 19) | def forward(ctx, value, value_spatial_shapes, value_level_start_index,
    method backward (line 57) | def backward(ctx, grad_output):
  class MultiScaleDeformableAttnFunction_fp32 (line 90) | class MultiScaleDeformableAttnFunction_fp32(Function):
    method forward (line 94) | def forward(ctx, value, value_spatial_shapes, value_level_start_index,
    method backward (line 133) | def backward(ctx, grad_output):

FILE: mmdet3d/models/fbbev/view_transformation/backward_projection/bevformer_utils/positional_encoding.py
  class CustormLearnedPositionalEncoding (line 12) | class CustormLearnedPositionalEncoding(BaseModule):
    method __init__ (line 26) | def __init__(self,
    method forward (line 38) | def forward(self, bs, h, w, device):
    method __repr__ (line 62) | def __repr__(self):

FILE: mmdet3d/models/fbbev/view_transformation/backward_projection/bevformer_utils/spatial_cross_attention_depth.py
  class DA_SpatialCrossAttention (line 31) | class DA_SpatialCrossAttention(BaseModule):
    method __init__ (line 44) | def __init__(self,
    method init_weight (line 80) | def init_weight(self):
    method forward (line 85) | def forward(self,
  class DA_MSDeformableAttention (line 228) | class DA_MSDeformableAttention(BaseModule):
    method __init__ (line 253) | def __init__(self,
    method init_weights (line 306) | def init_weights(self):
    method forward (line 331) | def forward(self,

FILE: mmdet3d/models/fbbev/view_transformation/forward_projection/view_transformer.py
  function gen_dx_bx (line 16) | def gen_dx_bx(xbound, ybound, zbound):
  class LSSViewTransformerFunction (line 24) | class LSSViewTransformerFunction(BaseModule):
    method __init__ (line 43) | def __init__(
    method create_grid_infos (line 69) | def create_grid_infos(self, x, y, z, **kwargs):
    method create_frustum (line 87) | def create_frustum(self, depth_cfg, input_size, downsample):
    method get_lidar_coor (line 111) | def get_lidar_coor(self, rots, trans, cam2imgs, post_rots, post_trans,
    method init_acceleration_v2 (line 153) | def init_acceleration_v2(self, coor):
    method voxel_pooling_v2 (line 174) | def voxel_pooling_v2(self, coor, depth, feat):
    method voxel_pooling_prepare_v2 (line 200) | def voxel_pooling_prepare_v2(self, coor):
    method pre_compute (line 260) | def pre_compute(self, cam_params):
    method view_transform_core (line 266) | def view_transform_core(self, cam_params, depth, tran_feat):
    method view_transform (line 290) | def view_transform(self, cam_params, depth, tran_feat):
    method forward (line 296) | def forward(self, cam_params, context, depth,  **kwargs):
    method get_mlp_input (line 309) | def get_mlp_input(self, rot, tran, intrin, post_rot, post_tran, bda):
  class LSSViewTransformerFunction3D (line 315) | class LSSViewTransformerFunction3D(BaseModule):
    method __init__ (line 334) | def __init__(
    method create_grid_infos (line 370) | def create_grid_infos(self, x, y, z, **kwargs):
    method create_frustum (line 388) | def create_frustum(self, depth_cfg, input_size, downsample):
    method get_cam2ego_coor (line 412) | def get_cam2ego_coor(self, input, downsample=1):
    method get_lidar_coor (line 457) | def get_lidar_coor(self, rots, trans, cam2imgs, post_rots, post_trans,
    method init_acceleration_v2 (line 499) | def init_acceleration_v2(self, coor):
    method voxel_pooling_v2 (line 520) | def voxel_pooling_v2(self, coor, depth, feat):
    method voxel_pooling_prepare_v2 (line 546) | def voxel_pooling_prepare_v2(self, coor):
    method pre_compute (line 606) | def pre_compute(self, cam_params):
    method view_transform_core (line 612) | def view_transform_core(self, cam_params, depth, tran_feat):
    method view_transform (line 638) | def view_transform(self, cam_params, depth, tran_feat):
    method forward (line 645) | def forward(self, cam_params, context, depth, **kwargs):
    method get_mlp_input (line 661) | def get_mlp_input(self, rot, tran, intrin, post_rot, post_tran, bda):

FILE: mmdet3d/models/fusion_layers/coord_transform.py
  function apply_3d_transformation (line 9) | def apply_3d_transformation(pcd, coord_type, img_meta, reverse=False):
  function extract_2d_info (line 95) | def extract_2d_info(img_meta, tensor):
  function bbox_2d_transform (line 123) | def bbox_2d_transform(img_meta, bbox_2d, ori2new):
  function coord_2d_transform (line 177) | def coord_2d_transform(img_meta, coord_2d, ori2new):

FILE: mmdet3d/models/fusion_layers/point_fusion.py
  function point_sample (line 14) | def point_sample(img_meta,
  class PointFusion (line 95) | class PointFusion(BaseModule):
    method __init__ (line 130) | def __init__(self,
    method forward (line 210) | def forward(self, img_feats, pts, pts_feats, img_metas):
    method obtain_mlvl_feats (line 237) | def obtain_mlvl_feats(self, img_feats, pts, img_metas):
    method sample_single (line 270) | def sample_single(self, img_feats, pts, img_meta):

FILE: mmdet3d/models/fusion_layers/vote_fusion.py
  class VoteFusion (line 13) | class VoteFusion(nn.Module):
    method __init__ (line 21) | def __init__(self, num_classes=10, max_imvote_per_pixel=3):
    method forward (line 26) | def forward(self, imgs, bboxes_2d_rescaled, seeds_3d_depth, img_metas):

FILE: mmdet3d/models/losses/axis_aligned_iou_loss.py
  function axis_aligned_iou_loss (line 11) | def axis_aligned_iou_loss(pred, target):
  class AxisAlignedIoULoss (line 31) | class AxisAlignedIoULoss(nn.Module):
    method __init__ (line 40) | def __init__(self, reduction='mean', loss_weight=1.0):
    method forward (line 46) | def forward(self,

FILE: mmdet3d/models/losses/chamfer_distance.py
  function chamfer_distance (line 9) | def chamfer_distance(src,
  class ChamferDistance (line 76) | class ChamferDistance(nn.Module):
    method __init__ (line 88) | def __init__(self,
    method forward (line 102) | def forward(self,

FILE: mmdet3d/models/losses/multibin_loss.py
  function multibin_loss (line 11) | def multibin_loss(pred_orientations, gt_orientations, num_dir_bins=4):
  class MultiBinLoss (line 61) | class MultiBinLoss(nn.Module):
    method __init__ (line 71) | def __init__(self, reduction='none', loss_weight=1.0):
    method forward (line 77) | def forward(self, pred, target, num_dir_bins, reduction_override=None):

FILE: mmdet3d/models/losses/paconv_regularization_loss.py
  function weight_correlation (line 10) | def weight_correlation(conv):
  function paconv_regularization_loss (line 47) | def paconv_regularization_loss(modules, reduction):
  class PAConvRegularizationLoss (line 72) | class PAConvRegularizationLoss(nn.Module):
    method __init__ (line 84) | def __init__(self, reduction='mean', loss_weight=1.0):
    method forward (line 90) | def forward(self, modules, reduction_override=None, **kwargs):

FILE: mmdet3d/models/losses/rotated_iou_loss.py
  function rotated_iou_3d_loss (line 11) | def rotated_iou_3d_loss(pred, target):
  class RotatedIoU3DLoss (line 30) | class RotatedIoU3DLoss(nn.Module):
    method __init__ (line 39) | def __init__(self, reduction='mean', loss_weight=1.0):
    method forward (line 44) | def forward(self,

FILE: mmdet3d/models/losses/uncertain_smooth_l1_loss.py
  function uncertain_smooth_l1_loss (line 10) | def uncertain_smooth_l1_loss(pred, target, sigma, alpha=1.0, beta=1.0):
  function uncertain_l1_loss (line 39) | def uncertain_l1_loss(pred, target, sigma, alpha=1.0):
  class UncertainSmoothL1Loss (line 62) | class UncertainSmoothL1Loss(nn.Module):
    method __init__ (line 79) | def __init__(self, alpha=1.0, beta=1.0, reduction='mean', loss_weight=...
    method forward (line 87) | def forward(self,
  class UncertainL1Loss (line 126) | class UncertainL1Loss(nn.Module):
    method __init__ (line 137) | def __init__(self, alpha=1.0, reduction='mean', loss_weight=1.0):
    method forward (line 144) | def forward(self,

FILE: mmdet3d/models/middle_encoders/pillar_scatter.py
  class PointPillarsScatter (line 10) | class PointPillarsScatter(nn.Module):
    method __init__ (line 20) | def __init__(self, in_channels, output_shape):
    method forward (line 29) | def forward(self, voxel_features, coors, batch_size=None):
    method forward_single (line 38) | def forward_single(self, voxel_features, coors):
    method forward_batch (line 62) | def forward_batch(self, voxel_features, coors, batch_size):

FILE: mmdet3d/models/middle_encoders/sparse_encoder.py
  class SparseEncoder (line 18) | class SparseEncoder(nn.Module):
    method __init__ (line 42) | def __init__(self,
    method forward (line 107) | def forward(self, voxel_features, coors, batch_size):
    method make_encoder_layers (line 139) | def make_encoder_layers(self,
  class MySparseEncoder (line 217) | class MySparseEncoder(nn.Module):
    method __init__ (line 241) | def __init__(self,
    method forward (line 306) | def forward(self, voxel_features, coors, batch_size):
    method make_encoder_layers (line 339) | def make_encoder_layers(self,
  class SparseEncoderSASSD (line 417) | class SparseEncoderSASSD(SparseEncoder):
    method __init__ (line 441) | def __init__(self,
    method forward (line 469) | def forward(self, voxel_features, coors, batch_size, test_mode=False):
    method get_auxiliary_targets (line 538) | def get_auxiliary_targets(self, nxyz, gt_boxes3d, enlarge=1.0):
    method calculate_pts_offsets (line 571) | def calculate_pts_offsets(self, points, boxes):
    method aux_loss (line 609) | def aux_loss(self, points, point_cls, point_reg, gt_bboxes):
    method make_auxiliary_points (line 654) | def make_auxiliary_points(self,

FILE: mmdet3d/models/middle_encoders/sparse_unet.py
  class SparseUNet (line 19) | class SparseUNet(BaseModule):
    method __init__ (line 38) | def __init__(self,
    method forward (line 106) | def forward(self, voxel_features, coors, batch_size):
    method decoder_layer_forward (line 157) | def decoder_layer_forward(self, x_lateral, x_bottom, lateral_layer,
    method reduce_channel (line 181) | def reduce_channel(x, out_channels):
    method make_encoder_layers (line 199) | def make_encoder_layers(self, make_block, norm_cfg, in_channels):
    method make_decoder_layers (line 245) | def make_decoder_layers(self, make_block, norm_cfg, in_channels):

FILE: mmdet3d/models/model_utils/edge_fusion_module.py
  class EdgeFusionModule (line 8) | class EdgeFusionModule(BaseModule):
    method __init__ (line 23) | def __init__(self,
    method forward (line 42) | def forward(self, features, fused_features, edge_indices, edge_lens,

FILE: mmdet3d/models/model_utils/transformer.py
  class GroupFree3DMHA (line 8) | class GroupFree3DMHA(MultiheadAttention):
    method __init__ (line 30) | def __init__(self,
    method forward (line 42) | def forward(self,
  class ConvBNPositionalEncoding (line 112) | class ConvBNPositionalEncoding(nn.Module):
    method __init__ (line 121) | def __init__(self, input_channel, num_pos_feats=288):
    method forward (line 128) | def forward(self, xyz):

FILE: mmdet3d/models/model_utils/vote_module.py
  class VoteModule (line 10) | class VoteModule(nn.Module):
    method __init__ (line 38) | def __init__(self,
    method forward (line 89) | def forward(self, seed_points, seed_feats):
    method get_loss (line 153) | def get_loss(self, seed_points, vote_points, seed_indices,

FILE: mmdet3d/models/necks/dla_neck.py
  function fill_up_weights (line 12) | def fill_up_weights(up):
  class IDAUpsample (line 29) | class IDAUpsample(BaseModule):
    method __init__ (line 44) | def __init__(
    method forward (line 93) | def forward(self, mlvl_features, start_level, end_level):
  class DLAUpsample (line 110) | class DLAUpsample(BaseModule):
    method __init__ (line 128) | def __init__(self,
    method forward (line 152) | def forward(self, mlvl_features):
  class DLANeck (line 171) | class DLANeck(BaseModule):
    method __init__ (line 187) | def __init__(self,
    method forward (line 210) | def forward(self, x):
    method init_weights (line 219) | def init_weights(self):

FILE: mmdet3d/models/necks/fpn.py
  class CustomFPN (line 12) | class CustomFPN(BaseModule):
    method __init__ (line 63) | def __init__(self,
    method forward (line 157) | def forward(self, inputs):

FILE: mmdet3d/models/necks/imvoxel_neck.py
  class OutdoorImVoxelNeck (line 9) | class OutdoorImVoxelNeck(nn.Module):
    method __init__ (line 17) | def __init__(self, in_channels, out_channels):
    method forward (line 50) | def forward(self, x):
    method init_weights (line 64) | def init_weights(self):
  class ResModule (line 69) | class ResModule(nn.Module):
    method __init__ (line 76) | def __init__(self, n_channels):
    method forward (line 96) | def forward(self, x):

FILE: mmdet3d/models/necks/lss_fpn.py
  class FPN_LSS (line 11) | class FPN_LSS(nn.Module):
    method __init__ (line 13) | def __init__(self,
    method forward (line 90) | def forward(self, feats):

FILE: mmdet3d/models/necks/pointnet2_fp_neck.py
  class PointNetFPNeck (line 10) | class PointNetFPNeck(BaseModule):
    method __init__ (line 34) | def __init__(self, fp_channels, init_cfg=None):
    method _extract_input (line 42) | def _extract_input(self, feat_dict):
    method forward (line 64) | def forward(self, feat_dict):

FILE: mmdet3d/models/necks/second_fpn.py
  class SECONDFPN (line 12) | class SECONDFPN(BaseModule):
    method __init__ (line 26) | def __init__(self,
    method forward (line 75) | def forward(self, x):

FILE: mmdet3d/models/necks/view_transformer.py
  class LSSViewTransformer (line 16) | class LSSViewTransformer(BaseModule):
    method __init__ (line 35) | def __init__(
    method create_grid_infos (line 60) | def create_grid_infos(self, x, y, z, **kwargs):
    method create_frustum (line 78) | def create_frustum(self, depth_cfg, input_size, downsample):
    method get_lidar_coor (line 102) | def get_lidar_coor(self, rots, trans, cam2imgs, post_rots, post_trans,
    method init_acceleration_v2 (line 144) | def init_acceleration_v2(self, coor):
    method voxel_pooling_v2 (line 165) | def voxel_pooling_v2(self, coor, depth, feat):
    method voxel_pooling_prepare_v2 (line 194) | def voxel_pooling_prepare_v2(self, coor):
    method pre_compute (line 260) | def pre_compute(self, input):
    method view_transform_core (line 266) | def view_transform_core(self, input, depth, tran_feat):
    method view_transform (line 291) | def view_transform(self, input, depth, tran_feat):
    method forward (line 296) | def forward(self, input, return_depth_digit=False):
    method get_mlp_input (line 325) | def get_mlp_input(self, rot, tran, intrin, post_rot, post_tran, bda):
  class LSSViewTransformer2 (line 332) | class LSSViewTransformer2(BaseModule):
    method __init__ (line 351) | def __init__(
    method create_grid_infos (line 376) | def create_grid_infos(self, x, y, z, **kwargs):
    method create_frustum (line 394) | def create_frustum(self, depth_cfg, input_size, downsample):
    method get_lidar_coor (line 418) | def get_lidar_coor(self, rots, trans, cam2imgs, post_rots, post_trans,
    method init_acceleration_v2 (line 460) | def init_acceleration_v2(self, coor):
    method voxel_pooling_v2 (line 483) | def voxel_pooling_v2(self, coor, depth, feat):
    method voxel_pooling_prepare_v2 (line 520) | def voxel_pooling_prepare_v2(self, depth, coor):
    method voxel_pooling_prepare_v2_inf (line 580) | def voxel_pooling_prepare_v2_inf(self, coor):
    method pre_compute (line 639) | def pre_compute(self, input):
    method view_transform_core (line 645) | def view_transform_core(self, input, depth, tran_feat):
    method view_transform (line 688) | def view_transform(self, input, depth, tran_feat):
    method forward (line 693) | def forward(self, input, return_depth_digit=False):
    method get_mlp_input (line 722) | def get_mlp_input(self, rot, tran, intrin, post_rot, post_tran, bda):
  class _ASPPModule (line 727) | class _ASPPModule(nn.Module):
    method __init__ (line 729) | def __init__(self, inplanes, planes, kernel_size, padding, dilation,
    method forward (line 745) | def forward(self, x):
    method _init_weight (line 751) | def _init_weight(self):
  class ASPP (line 760) | class ASPP(nn.Module):
    method __init__ (line 762) | def __init__(self, inplanes, mid_channels=256, BatchNorm=nn.BatchNorm2d):
    method forward (line 809) | def forward(self, x):
    method _init_weight (line 825) | def _init_weight(self):
  class Mlp (line 834) | class Mlp(nn.Module):
    method __init__ (line 836) | def __init__(self,
    method forward (line 851) | def forward(self, x):
  class SELayer (line 860) | class SELayer(nn.Module):
    method __init__ (line 862) | def __init__(self, channels, act_layer=nn.ReLU, gate_layer=nn.Sigmoid):
    method forward (line 869) | def forward(self, x, x_se):
  class DepthNet (line 876) | class DepthNet(nn.Module):
    method __init__ (line 878) | def __init__(self,
    method forward (line 931) | def forward(self, x, mlp_input):
  class DepthAggregation (line 943) | class DepthAggregation(nn.Module):
    method __init__ (line 946) | def __init__(self, in_channels, mid_channels, out_channels):
    method forward (line 995) | def forward(self, x):
  class LSSViewTransformerBEVDepth (line 1005) | class LSSViewTransformerBEVDepth(LSSViewTransformer2):
    method __init__ (line 1007) | def __init__(self, loss_depth_weight=3.0, depthnet_cfg=dict(), with_cp...
    method get_mlp_input (line 1014) | def get_mlp_input(self, rot, tran, intrin, post_rot, post_tran, bda):
    method get_downsampled_gt_depth (line 1040) | def get_downsampled_gt_depth(self, gt_depths):
    method get_depth_loss (line 1072) | def get_depth_loss(self, depth_labels, depth_preds):
    method forward (line 1087) | def forward(self, input, return_depth_digit=False):

FILE: mmdet3d/models/roi_heads/base_3droi_head.py
  class Base3DRoIHead (line 7) | class Base3DRoIHead(BaseModule, metaclass=ABCMeta):
    method __init__ (line 10) | def __init__(self,
    method with_bbox (line 31) | def with_bbox(self):
    method with_mask (line 36) | def with_mask(self):
    method init_bbox_head (line 41) | def init_bbox_head(self):
    method init_mask_head (line 46) | def init_mask_head(self):
    method init_assigner_sampler (line 51) | def init_assigner_sampler(self):
    method forward_train (line 56) | def forward_train(self,
    method simple_test (line 82) | def simple_test(self,
    method aug_test (line 92) | def aug_test(self, x, proposal_list, img_metas, rescale=False, **kwargs):

FILE: mmdet3d/models/roi_heads/bbox_heads/h3d_bbox_head.py
  class H3DBboxHead (line 17) | class H3DBboxHead(BaseModule):
    method __init__ (line 57) | def __init__(self,
    method forward (line 203) | def forward(self, feats_dict, sample_mod):
    method loss (line 311) | def loss(self,
    method get_bboxes (line 439) | def get_bboxes(self,
    method multiclass_nms_single (line 485) | def multiclass_nms_single(self, obj_scores, sem_scores, bbox, points,
    method get_proposal_stage_loss (line 545) | def get_proposal_stage_loss(self,
    method get_targets (line 654) | def get_targets(self,
    method get_targets_single (line 754) | def get_targets_single(self,

FILE: mmdet3d/models/roi_heads/bbox_heads/parta2_bbox_head.py
  class PartA2BboxHead (line 26) | class PartA2BboxHead(BaseModule):
    method __init__ (line 56) | def __init__(self,
    method init_weights (line 237) | def init_weights(self):
    method forward (line 241) | def forward(self, seg_feats, part_feats):
    method loss (line 291) | def loss(self, cls_score, bbox_pred, rois, labels, bbox_targets,
    method get_targets (line 364) | def get_targets(self, sampling_results, rcnn_train_cfg, concat=True):
    method _get_target_single (line 404) | def _get_target_single(self, pos_bboxes, pos_gt_bboxes, ious, cfg):
    method get_corner_loss_lidar (line 469) | def get_corner_loss_lidar(self, pred_bbox3d, gt_bbox3d, delta=1.0):
    method get_bboxes (line 505) | def get_bboxes(self,
    method multi_class_nms (line 563) | def multi_class_nms(self,

FILE: mmdet3d/models/roi_heads/bbox_heads/point_rcnn_bbox_head.py
  class PointRCNNBboxHead (line 18) | class PointRCNNBboxHead(BaseModule):
    method __init__ (line 64) | def __init__(
    method _add_conv_branch (line 172) | def _add_conv_branch(self, in_channels, conv_channels):
    method init_weights (line 197) | def init_weights(self):
    method forward (line 206) | def forward(self, feats):
    method loss (line 242) | def loss(self, cls_score, bbox_pred, rois, labels, bbox_targets,
    method get_corner_loss_lidar (line 313) | def get_corner_loss_lidar(self, pred_bbox3d, gt_bbox3d, delta=1.0):
    method get_targets (line 348) | def get_targets(self, sampling_results, rcnn_train_cfg, concat=True):
    method _get_target_single (line 388) | def _get_target_single(self, pos_bboxes, pos_gt_bboxes, ious, cfg):
    method get_bboxes (line 452) | def get_bboxes(self,
    method multi_class_nms (line 509) | def multi_class_nms(self,

FILE: mmdet3d/models/roi_heads/h3d_roi_head.py
  class H3DRoIHead (line 8) | class H3DRoIHead(Base3DRoIHead):
    method __init__ (line 18) | def __init__(self,
    method init_mask_head (line 37) | def init_mask_head(self):
    method init_bbox_head (line 42) | def init_bbox_head(self, bbox_head):
    method init_assigner_sampler (line 48) | def init_assigner_sampler(self):
    method forward_train (line 52) | def forward_train(self,
    method simple_test (line 120) | def simple_test(self, feats_dict, img_metas, points, rescale=False):

FILE: mmdet3d/models/roi_heads/mask_heads/pointwise_semantic_head.py
  class PointwiseSemanticHead (line 13) | class PointwiseSemanticHead(BaseModule):
    method __init__ (line 27) | def __init__(self,
    method forward (line 54) | def forward(self, x):
    method get_targets_single (line 80) | def get_targets_single(self, voxel_centers, gt_bboxes_3d, gt_labels_3d):
    method get_targets (line 129) | def get_targets(self, voxels_dict, gt_bboxes_3d, gt_labels_3d):
    method loss (line 161) | def loss(self, semantic_results, semantic_targets):

FILE: mmdet3d/models/roi_heads/mask_heads/primitive_head.py
  class PrimitiveHead (line 16) | class PrimitiveHead(BaseModule):
    method __init__ (line 41) | def __init__(self,
    method forward (line 116) | def forward(self, feats_dict, sample_mod):
    method loss (line 187) | def loss(self,
    method get_targets (line 258) | def get_targets(self,
    method get_targets_single (line 326) | def get_targets_single(self,
    method primitive_decode_scores (line 602) | def primitive_decode_scores(self, predictions, aggregated_points):
    method check_horizon (line 630) | def check_horizon(self, points):
    method check_dist (line 643) | def check_dist(self, plane_equ, points):
    method point2line_dist (line 656) | def point2line_dist(self, points, pts_a, pts_b):
    method match_point2line (line 675) | def match_point2line(self, points, corners, with_yaw, mode='bottom'):
    method match_point2plane (line 716) | def match_point2plane(self, plane, points):
    method compute_primitive_loss (line 734) | def compute_primitive_loss(self, primitive_center, primitive_semantic,
    method get_primitive_center (line 783) | def get_primitive_center(self, pred_flag, center):
    method _assign_primitive_line_targets (line 802) | def _assign_primitive_line_targets(self,
    method _assign_primitive_surface_targets (line 867) | def _assign_primitive_surface_targets(self,
    method _get_plane_fomulation (line 951) | def _get_plane_fomulation(self, vector1, vector2, point):

FILE: mmdet3d/models/roi_heads/part_aggregation_roi_head.py
  class PartAggregationROIHead (line 14) | class PartAggregationROIHead(Base3DRoIHead):
    method __init__ (line 27) | def __init__(self,
    method init_mask_head (line 60) | def init_mask_head(self):
    method init_bbox_head (line 65) | def init_bbox_head(self, bbox_head):
    method init_assigner_sampler (line 69) | def init_assigner_sampler(self):
    method with_semantic (line 83) | def with_semantic(self):
    method forward_train (line 88) | def forward_train(self, feats_dict, voxels_dict, img_metas, proposal_l...
    method simple_test (line 130) | def simple_test(self, feats_dict, voxels_dict, img_metas, proposal_list,
    method _bbox_forward_train (line 173) | def _bbox_forward_train(self, seg_feats, part_feats, voxels_dict,
    method _bbox_forward (line 200) | def _bbox_forward(self, seg_feats, part_feats, voxels_dict, rois):
    method _assign_and_sample (line 231) | def _assign_and_sample(self, proposal_list, gt_bboxes_3d, gt_labels_3d):
    method _semantic_forward_train (line 305) | def _semantic_forward_train(self, x, voxels_dict, gt_bboxes_3d,

FILE: mmdet3d/models/roi_heads/point_rcnn_roi_head.py
  class PointRCNNRoIHead (line 13) | class PointRCNNRoIHead(Base3DRoIHead):
    method __init__ (line 26) | def __init__(self,
    method init_bbox_head (line 47) | def init_bbox_head(self, bbox_head):
    method init_mask_head (line 55) | def init_mask_head(self):
    method init_assigner_sampler (line 59) | def init_assigner_sampler(self):
    method forward_train (line 72) | def forward_train(self, feats_dict, input_metas, proposal_list,
    method simple_test (line 118) | def simple_test(self, feats_dict, img_metas, proposal_list, **kwargs):
    method _bbox_forward_train (line 166) | def _bbox_forward_train(self, features, points, sampling_results):
    method _bbox_forward (line 192) | def _bbox_forward(self, features, points, batch_size, rois):
    method _assign_and_sample (line 214) | def _assign_and_sample(self, proposal_list, gt_bboxes_3d, gt_labels_3d):

FILE: mmdet3d/models/roi_heads/roi_extractors/single_roiaware_extractor.py
  class Single3DRoIAwareExtractor (line 10) | class Single3DRoIAwareExtractor(BaseModule):
    method __init__ (line 19) | def __init__(self, roi_layer=None, init_cfg=None):
    method build_roi_layers (line 23) | def build_roi_layers(self, layer_cfg):
    method forward (line 32) | def forward(self, feats, coordinate, batch_inds, rois):

FILE: mmdet3d/models/roi_heads/roi_extractors/single_roipoint_extractor.py
  class Single3DRoIPointExtractor (line 11) | class Single3DRoIPointExtractor(nn.Module):
    method __init__ (line 20) | def __init__(self, roi_layer=None):
    method build_roi_layers (line 24) | def build_roi_layers(self, layer_cfg):
    method forward (line 33) | def forward(self, feats, coordinate, batch_inds, rois):

FILE: mmdet3d/models/segmentors/base.py
  class Base3DSegmentor (line 14) | class Base3DSegmentor(BaseSegmentor):
    method with_regularization_loss (line 22) | def with_regularization_loss(self):
    method forward_test (line 27) | def forward_test(self, points, img_metas, **kwargs):
    method forward (line 55) | def forward(self, return_loss=True, **kwargs):
    method show_results (line 71) | def show_results(self,

FILE: mmdet3d/models/segmentors/encoder_decoder.py
  class EncoderDecoder3D (line 14) | class EncoderDecoder3D(Base3DSegmentor):
    method __init__ (line 22) | def __init__(self,
    method _init_decode_head (line 45) | def _init_decode_head(self, decode_head):
    method _init_auxiliary_head (line 50) | def _init_auxiliary_head(self, auxiliary_head):
    method _init_loss_regularization (line 60) | def _init_loss_regularization(self, loss_regularization):
    method extract_feat (line 70) | def extract_feat(self, points):
    method encode_decode (line 77) | def encode_decode(self, points, img_metas):
    method _decode_head_forward_train (line 92) | def _decode_head_forward_train(self, x, img_metas, pts_semantic_mask):
    method _decode_head_forward_test (line 103) | def _decode_head_forward_test(self, x, img_metas):
    method _auxiliary_head_forward_train (line 109) | def _auxiliary_head_forward_train(self, x, img_metas, pts_semantic_mask):
    method _loss_regularization_forward_train (line 126) | def _loss_regularization_forward_train(self):
    method forward_dummy (line 141) | def forward_dummy(self, points):
    method forward_train (line 147) | def forward_train(self, points, img_metas, pts_semantic_mask):
    method _input_generation (line 183) | def _input_generation(coords,
    method _sliding_patch_generation (line 218) | def _sliding_patch_generation(self,
    method slide_inference (line 322) | def slide_inference(self, point, img_meta, rescale):
    method whole_inference (line 366) | def whole_inference(self, points, img_metas, rescale):
    method inference (line 372) | def inference(self, points, img_metas, rescale):
    method simple_test (line 395) | def simple_test(self, points, img_metas, rescale=True):
    method aug_test (line 425) | def aug_test(self, points, img_metas, rescale=True):

FILE: mmdet3d/models/utils/clip_sigmoid.py
  function clip_sigmoid (line 5) | def clip_sigmoid(x, eps=1e-4):

FILE: mmdet3d/models/utils/edge_indices.py
  function get_edge_indices (line 6) | def get_edge_indices(img_metas,

FILE: mmdet3d/models/utils/gen_keypoints.py
  function get_keypoints (line 7) | def get_keypoints(gt_bboxes_3d_list,

FILE: mmdet3d/models/utils/handle_objs.py
  function filter_outside_objs (line 5) | def filter_outside_objs(gt_bboxes_list, gt_labels_list, gt_bboxes_3d_list,
  function get_centers2d_target (line 39) | def get_centers2d_target(centers2d, centers, img_shape):
  function handle_proj_objs (line 83) | def handle_proj_objs(centers2d_list, gt_bboxes_list, img_metas):

FILE: mmdet3d/models/utils/mlp.py
  class MLP (line 7) | class MLP(BaseModule):
    method __init__ (line 25) | def __init__(self,
    method forward (line 50) | def forward(self, img_features):

FILE: mmdet3d/models/voxel_encoders/pillar_encoder.py
  class PillarFeatureNet (line 13) | class PillarFeatureNet(nn.Module):
    method __init__ (line 40) | def __init__(self,
    method forward (line 94) | def forward(self, features, num_points, coors):
  class DynamicPillarFeatureNet (line 163) | class DynamicPillarFeatureNet(PillarFeatureNet):
    method __init__ (line 192) | def __init__(self,
    method map_voxel_center_to_point (line 236) | def map_voxel_center_to_point(self, pts_coors, voxel_mean, voxel_coors):
    method forward (line 276) | def forward(self, features, coors):

FILE: mmdet3d/models/voxel_encoders/utils.py
  function get_paddings_indicator (line 9) | def get_paddings_indicator(actual_num, max_num, axis=0):
  class VFELayer (line 32) | class VFELayer(nn.Module):
    method __init__ (line 49) | def __init__(self,
    method forward (line 65) | def forward(self, inputs):
  class PFNLayer (line 107) | class PFNLayer(nn.Module):
    method __init__ (line 124) | def __init__(self,
    method forward (line 146) | def forward(self, inputs, num_voxels=None, aligned_distance=None):

FILE: mmdet3d/models/voxel_encoders/voxel_encoder.py
  class HardSimpleVFE (line 14) | class HardSimpleVFE(nn.Module):
    method __init__ (line 23) | def __init__(self, num_features=4):
    method forward (line 29) | def forward(self, features, num_points, coors):
  class DynamicSimpleVFE (line 49) | class DynamicSimpleVFE(nn.Module):
    method __init__ (line 60) | def __init__(self,
    method forward (line 69) | def forward(self, features, coors):
  class DynamicVFE (line 88) | class DynamicVFE(nn.Module):
    method __init__ (line 119) | def __init__(self,
    method map_voxel_center_to_point (line 179) | def map_voxel_center_to_point(self, pts_coors, voxel_mean, voxel_coors):
    method forward (line 222) | def forward(self,
  class HardVFE (line 290) | class HardVFE(nn.Module):
    method __init__ (line 319) | def __init__(self,
    method forward (line 388) | def forward(self,
    method fusion_with_mask (line 456) | def fusion_with_mask(self, features, mask, voxel_feats, coors, img_feats,

FILE: mmdet3d/ops/bev_pool_v2/bev_pool.py
  class QuickCumsumCuda (line 11) | class QuickCumsumCuda(torch.autograd.Function):
    method forward (line 14) | def forward(ctx, depth, feat, ranks_depth, ranks_feat, ranks_bev,
    method backward (line 41) | def backward(ctx, out_grad):
  function bev_pool_v2 (line 83) | def bev_pool_v2(depth, feat, ranks_depth, ranks_feat, ranks_bev,
  class TRTBEVPoolv2 (line 92) | class TRTBEVPoolv2(torch.autograd.Function):
    method symbolic (line 95) | def symbolic(g,
    method forward (line 119) | def forward(g,
  function test_bev_pool_v2 (line 144) | def test_bev_pool_v2():

FILE: mmdet3d/ops/bev_pool_v2/src/bev_pool.cpp
  function bev_pool_v2_forward (line 28) | void bev_pool_v2_forward(
  function bev_pool_v2_backward (line 72) | void bev_pool_v2_backward(
  function PYBIND11_MODULE (line 104) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: mmdet3d/ops/dgcnn_modules/dgcnn_fa_module.py
  class DGCNNFAModule (line 8) | class DGCNNFAModule(BaseModule):
    method __init__ (line 22) | def __init__(self,
    method forward (line 43) | def forward(self, points):

FILE: mmdet3d/ops/dgcnn_modules/dgcnn_fp_module.py
  class DGCNNFPModule (line 7) | class DGCNNFPModule(BaseModule):
    method __init__ (line 21) | def __init__(self,
    method forward (line 42) | def forward(self, points):

FILE: mmdet3d/ops/dgcnn_modules/dgcnn_gf_module.py
  class BaseDGCNNGFModule (line 9) | class BaseDGCNNGFModule(nn.Module):
    method __init__ (line 32) | def __init__(self,
    method _pool_features (line 85) | def _pool_features(self, features):
    method forward (line 109) | def forward(self, points):
  class DGCNNGFModule (line 154) | class DGCNNGFModule(BaseDGCNNGFModule):
    method __init__ (line 183) | def __init__(self,

FILE: mmdet3d/ops/norm.py
  class AllReduce (line 10) | class AllReduce(Function):
    method forward (line 13) | def forward(ctx, input):
    method backward (line 23) | def backward(ctx, grad_output):
  class NaiveSyncBatchNorm1d (line 29) | class NaiveSyncBatchNorm1d(nn.BatchNorm1d):
    method __init__ (line 47) | def __init__(self, *args, **kwargs):
    method forward (line 55) | def forward(self, input):
  class NaiveSyncBatchNorm2d (line 102) | class NaiveSyncBatchNorm2d(nn.BatchNorm2d):
    method __init__ (line 120) | def __init__(self, *args, **kwargs):
    method forward (line 128) | def forward(self, input):

FILE: mmdet3d/ops/ops_dcnv3/functions/dcnv3_func.py
  class DCNv3Function (line 19) | class DCNv3Function(Function):
    method forward (line 22) | def forward(
    method backward (line 51) | def backward(ctx, grad_output):
    method symbolic (line 64) | def symbolic(g, input, offset, mask, kernel_h, kernel_w, stride_h,
  function _get_reference_points (line 91) | def _get_reference_points(spatial_shapes, device, kernel_h, kernel_w, di...
  function _generate_dilation_grids (line 122) | def _generate_dilation_grids(spatial_shapes, kernel_h, kernel_w, dilatio...
  function dcnv3_core_pytorch (line 147) | def dcnv3_core_pytorch(

FILE: mmdet3d/ops/ops_dcnv3/modules/dcnv3.py
  class to_channels_first (line 19) | class to_channels_first(nn.Module):
    method __init__ (line 21) | def __init__(self):
    method forward (line 24) | def forward(self, x):
  class to_channels_last (line 28) | class to_channels_last(nn.Module):
    method __init__ (line 30) | def __init__(self):
    method forward (line 33) | def forward(self, x):
  function build_norm_layer (line 37) | def build_norm_layer(dim,
  function build_act_layer (line 61) | def build_act_layer(act_layer):
  function _is_power_of_2 (line 72) | def _is_power_of_2(n):
  class CenterFeatureScaleModule (line 80) | class CenterFeatureScaleModule(nn.Module):
    method forward (line 81) | def forward(self,
  class DCNv3_pytorch (line 91) | class DCNv3_pytorch(nn.Module):
    method __init__ (line 92) | def __init__(
    method _reset_parameters (line 172) | def _reset_parameters(self):
    method forward (line 182) | def forward(self, input):
  class DCNv3 (line 218) | class DCNv3(nn.Module):
    method __init__ (line 219) | def __init__(
    method _reset_parameters (line 299) | def _reset_parameters(self):
    method forward (line 309) | def forward(self, input):

FILE: mmdet3d/ops/ops_dcnv3/setup.py
  function get_extensions (line 22) | def get_extensions():

FILE: mmdet3d/ops/ops_dcnv3/src/cpu/dcnv3_cpu.cpp
  function dcnv3_cpu_forward (line 17) | at::Tensor dcnv3_cpu_forward(const at::Tensor &input, const at::Tensor &...
  function dcnv3_cpu_backward (line 28) | std::vector<at::Tensor>

FILE: mmdet3d/ops/ops_dcnv3/src/vision.cpp
  function PYBIND11_MODULE (line 14) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: mmdet3d/ops/ops_dcnv3/test.py
  function check_forward_equal_with_pytorch_double (line 34) | def check_forward_equal_with_pytorch_double():
  function check_forward_equal_with_pytorch_float (line 64) | def check_forward_equal_with_pytorch_float():
  function check_backward_equal_with_pytorch_double (line 93) | def check_backward_equal_with_pytorch_double(channels=4, grad_input=True...
  function check_backward_equal_with_pytorch_float (line 156) | def check_backward_equal_with_pytorch_float(channels=4, grad_input=True,...
  function check_time_cost (line 220) | def check_time_cost(im2col_step=128):

FILE: mmdet3d/ops/paconv/paconv.py
  class ScoreNet (line 14) | class ScoreNet(nn.Module):
    method __init__ (line 40) | def __init__(self,
    method forward (line 82) | def forward(self, xyz_features):
  class PAConv (line 108) | class PAConv(nn.Module):
    method __init__ (line 142) | def __init__(self,
    method init_weights (line 221) | def init_weights(self):
    method _prepare_scorenet_input (line 226) | def _prepare_scorenet_input(self, points_xyz):
    method forward (line 253) | def forward(self, inputs):
  class PAConvCUDA (line 307) | class PAConvCUDA(PAConv):
    method __init__ (line 318) | def __init__(self,
    method forward (line 346) | def forward(self, inputs):

FILE: mmdet3d/ops/paconv/utils.py
  function calc_euclidian_dist (line 5) | def calc_euclidian_dist(xyz1, xyz2):
  function assign_score (line 21) | def assign_score(scores, point_features):
  function assign_kernel_withoutk (line 47) | def assign_kernel_withoutk(features, kernels, M):

FILE: mmdet3d/ops/pointnet_modules/builder.py
  function build_sa_module (line 7) | def build_sa_module(cfg, *args, **kwargs):

FILE: mmdet3d/ops/pointnet_modules/paconv_sa_module.py
  class PAConvSAModuleMSG (line 11) | class PAConvSAModuleMSG(BasePointSAModule):
    method __init__ (line 44) | def __init__(self,
  class PAConvSAModule (line 109) | class PAConvSAModule(PAConvSAModuleMSG):
    method __init__ (line 117) | def __init__(self,
  class PAConvCUDASAModuleMSG (line 154) | class PAConvCUDASAModuleMSG(BasePointSAModule):
    method __init__ (line 163) | def __init__(self,
    method forward (line 231) | def forward(
  class PAConvCUDASAModule (line 300) | class PAConvCUDASAModule(PAConvCUDASAModuleMSG):
    method __init__ (line 309) | def __init__(self,

FILE: mmdet3d/ops/pointnet_modules/point_fp_module.py
  class PointFPModule (line 11) | class PointFPModule(BaseModule):
    method __init__ (line 22) | def __init__(self,
    method forward (line 41) | def forward(self, target: torch.Tensor, source: torch.Tensor,

FILE: mmdet3d/ops/pointnet_modules/point_sa_module.py
  class BasePointSAModule (line 14) | class BasePointSAModule(nn.Module):
    method __init__ (line 44) | def __init__(self,
    method _sample_points (line 112) | def _sample_points(self, points_xyz, features, indices, target_xyz):
    method _pool_features (line 146) | def _pool_features(self, features):
    method forward (line 170) | def forward(
  class PointSAModuleMSG (line 226) | class PointSAModuleMSG(BasePointSAModule):
    method __init__ (line 258) | def __init__(self,
  class PointSAModule (line 304) | class PointSAModule(PointSAModuleMSG):
    method __init__ (line 331) | def __init__(self,

FILE: mmdet3d/ops/sparse_block.py
  function replace_feature (line 14) | def replace_feature(out, new_features):
  class SparseBottleneck (line 23) | class SparseBottleneck(Bottleneck, SparseModule):
    method __init__ (line 41) | def __init__(self,
    method forward (line 59) | def forward(self, x):
  class SparseBasicBlock (line 82) | class SparseBasicBlock(BasicBlock, SparseModule):
    method __init__ (line 100) | def __init__(self,
    method forward (line 117) | def forward(self, x):
  function make_sparse_convmodule (line 137) | def make_sparse_convmodule(in_channels,

FILE: mmdet3d/ops/spconv/overwrite_spconv/write_spconv2.py
  function register_spconv2 (line 8) | def register_spconv2():
  function _load_from_state_dict (line 42) | def _load_from_state_dict(self, state_dict, prefix, local_metadata, strict,

FILE: mmdet3d/utils/collect_env.py
  function collect_env (line 11) | def collect_env():

FILE: mmdet3d/utils/compat_cfg.py
  function compat_cfg (line 8) | def compat_cfg(cfg):
  function compat_runner_args (line 22) | def compat_runner_args(cfg):
  function compat_imgs_per_gpu (line 37) | def compat_imgs_per_gpu(cfg):
  function compat_loader_args (line 54) | def compat_loader_args(cfg):

FILE: mmdet3d/utils/logger.py
  function get_root_logger (line 7) | def get_root_logger(log_file=None, log_level=logging.INFO, name='mmdet3d'):

FILE: mmdet3d/utils/misc.py
  function find_latest_checkpoint (line 7) | def find_latest_checkpoint(path, suffix='pth'):

FILE: mmdet3d/utils/setup_env.py
  function setup_multi_processes (line 10) | def setup_multi_processes(cfg):

FILE: mmdet3d/version.py
  function parse_version_info (line 7) | def parse_version_info(version_str):

FILE: tools/analysis_tools/analyze_logs.py
  function cal_train_time (line 11) | def cal_train_time(log_dicts, args):
  function plot_curve (line 34) | def plot_curve(log_dicts, args):
  function add_plot_parser (line 113) | def add_plot_parser(subparsers):
  function add_time_parser (line 143) | def add_time_parser(subparsers):
  function parse_args (line 159) | def parse_args():
  function load_json_logs (line 169) | def load_json_logs(json_logs):
  function main (line 189) | def main():

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

FILE: tools/analysis_tools/benchmark_sequential.py
  function parse_args (line 15) | def parse_args():
  function main (line 35) | def main():

FILE: tools/analysis_tools/benchmark_trt.py
  function parse_args (line 25) | def parse_args():
  function torch_dtype_from_trt (line 35) | def torch_dtype_from_trt(dtype: trt.DataType) -> torch.dtype:
  class TRTWrapper (line 59) | class TRTWrapper(torch.nn.Module):
    method __init__ (line 61) | def __init__(self,
    method forward (line 81) | def forward(self, inputs: Dict[str, torch.Tensor]):
  function get_plugin_names (line 104) | def get_plugin_names():
  function main (line 108) | def main():

FILE: tools/analysis_tools/benchmark_view_transformer.py
  function parse_args (line 15) | def parse_args():
  function main (line 34) | def main():

FILE: tools/analysis_tools/create_video.py
  class RecordMovie (line 7) | class RecordMovie(object):
    method __init__ (line 9) | def __init__(self, img_width, img_height):
    method start (line 16) | def start(self, file_name, freq):
    method record (line 26) | def record(self, img):
    method end (line 31) | def end(self):
  function main_waymo (line 37) | def main_waymo():

FILE: tools/analysis_tools/generate_mask_based_on_lidar_points.py
  function convert_color (line 21) | def convert_color(img_path):
  function save_tensor (line 28) | def save_tensor(tensor, path, pad_value=254.0,normalize=False):
  function generate_forward_transformation_matrix (line 43) | def generate_forward_transformation_matrix(bda, img_meta_dict=None):
  function show_mask (line 56) | def show_mask(mask, ax, random_color=False, cls_=None):
  function show_points (line 90) | def show_points(coords, labels, ax, marker_size=375):
  function show_box (line 96) | def show_box(box, ax):
  function f (line 181) | def f(gap=0):

FILE: tools/analysis_tools/get_flops.py
  function parse_args (line 15) | def parse_args():
  function construct_input (line 44) | def construct_input(input_shape):
  function main (line 57) | def main():

FILE: tools/analysis_tools/vis.py
  function check_point_in_img (line 14) | def check_point_in_img(points, height, width):
  function depth2color (line 21) | def depth2color(depth):
  function lidar2img (line 37) | def lidar2img(points_lidar, camrera_info):
  function get_lidar2global (line 57) | def get_lidar2global(infos):
  function parse_args (line 68) | def parse_args():
  function main (line 129) | def main():

FILE: tools/analysis_tools/vis_occupancy.py
  function remove_far (line 49) | def remove_far(points, point_cloud_range):
  function voxelize (line 54) | def voxelize(voxel: np.array, label_count: np.array):
  function points2voxel (line 76) | def points2voxel(points, voxel_shape, voxel_size, max_points=5, specific...
  function get_grid_coords (line 110) | def get_grid_coords(dims, resolution):
  function draw (line 135) | def draw(
  function voxel_exist (line 298) | def voxel_exist(voxels, x,y,z):
  function max_connected (line 304) | def max_connected(voxels, distance=3):
  function main (line 347) | def main(filepath='*.npz'):

FILE: tools/create_data.py
  function kitti_data_prep (line 13) | def kitti_data_prep(root_path,
  function nuscenes_data_prep (line 54) | def nuscenes_data_prep(root_path,
  function lyft_data_prep (line 93) | def lyft_data_prep(root_path, info_prefix, version, max_sweeps=10):
  function scannet_data_prep (line 111) | def scannet_data_prep(root_path, info_prefix, out_dir, workers):
  function s3dis_data_prep (line 124) | def s3dis_data_prep(root_path, info_prefix, out_dir, workers):
  function sunrgbd_data_prep (line 137) | def sunrgbd_data_prep(root_path, info_prefix, out_dir, workers, num_poin...
  function waymo_data_prep (line 154) | def waymo_data_prep(root_path,

FILE: tools/create_data_bev_planner.py
  function get_gt (line 51) | def get_gt(info, traj_in_lidar_coor=None, traj_mask_in_lidar_coor=None):
  function nuscenes_data_prep (line 115) | def nuscenes_data_prep(root_path, info_prefix, version, max_sweeps=10):
  function add_ann_adj_info (line 133) | def add_ann_adj_info(extra_tag, with_lidar_seg=False):

FILE: tools/data_converter/create_gt_database.py
  function _poly2mask (line 17) | def _poly2mask(mask_ann, img_h, img_w):
  function _parse_coco_ann_info (line 33) | def _parse_coco_ann_info(ann_info):
  function crop_image_patch_v2 (line 70) | def crop_image_patch_v2(pos_proposals, pos_assigned_gt_inds, gt_masks):
  function crop_image_patch (line 90) | def crop_image_patch(pos_proposals, gt_masks, pos_assigned_gt_inds, org_...
  function create_groundtruth_database (line 110) | def create_groundtruth_database(dataset_class_name,
  class GTDatabaseCreater (line 399) | class GTDatabaseCreater:
    method __init__ (line 426) | def __init__(self,
    method create_single (line 459) | def create_single(self, input_dict):
    method create (line 560) | def create(self):

FILE: tools/data_converter/imgaug_demo.py
  function get_available_scenes (line 24) | def get_available_scenes(nusc):

FILE: tools/data_converter/indoor_converter.py
  function create_indoor_info_file (line 12) | def create_indoor_info_file(data_path,

FILE: tools/data_converter/kitti_converter.py
  function convert_to_kitti_info_version2 (line 16) | def convert_to_kitti_info_version2(info):
  function _read_imageset_file (line 41) | def _read_imageset_file(path):
  class _NumPointsInGTCalculater (line 47) | class _NumPointsInGTCalculater:
    method __init__ (line 63) | def __init__(self,
    method calculate_single (line 75) | def calculate_single(self, info):
    method calculate (line 109) | def calculate(self, infos):
  function _calculate_num_points_in_gt (line 116) | def _calculate_num_points_in_gt(data_path,
  function create_kitti_info_file (line 157) | def create_kitti_info_file(data_path,
  function create_waymo_info_file (line 229) | def create_waymo_info_file(data_path,
  function _create_reduced_point_cloud (line 305) | def _create_reduced_point_cloud(data_path,
  function create_reduced_point_cloud (line 366) | def create_reduced_point_cloud(data_path,
  function export_2d_annotation (line 411) | def export_2d_annotation(root_path, info_path, mono3d=True):
  function get_2d_boxes (line 459) | def get_2d_boxes(info, occluded, mono3d=True):
  function generate_record (line 570) | def generate_record(ann_rec, x1, y1, x2, y2, sample_data_token, filename):

FILE: tools/data_converter/kitti_data_utils.py
  function get_image_index_str (line 13) | def get_image_index_str(img_idx, use_prefix_id=False):
  function get_kitti_info_path (line 20) | def get_kitti_info_path(idx,
  function get_image_path (line 43) | def get_image_path(idx,
  function get_label_path (line 54) | def get_label_path(idx,
  function get_plane_path (line 65) | def get_plane_path(idx,
  function get_velodyne_path (line 76) | def get_velodyne_path(idx,
  function get_calib_path (line 86) | def get_calib_path(idx,
  function get_pose_path (line 96) | def get_pose_path(idx,
  function get_timestamp_path (line 106) | def get_timestamp_path(idx,
  function get_label_anno (line 116) | def get_label_anno(label_path):
  function _extend_matrix (line 160) | def _extend_matrix(mat):
  function get_kitti_image_info (line 165) | def get_kitti_image_info(path,
  class WaymoInfoGatherer (line 297) | class WaymoInfoGatherer:
    method __init__ (line 328) | def __init__(self,
    method gather_single (line 352) | def gather_single(self, idx):
    method gather (line 501) | def gather(self, image_ids):
  function kitti_anno_to_label_file (line 509) | def kitti_anno_to_label_file(annos, folder):
  function add_difficulty_to_annos (line 532) | def add_difficulty_to_annos(info):
  function kitti_result_line (line 577) | def kitti_result_line(result_dict, precision=4):

FILE: tools/data_converter/lyft_converter.py
  function create_lyft_infos (line 19) | def create_lyft_infos(root_path,
  function _fill_trainval_infos (line 94) | def _fill_trainval_infos(lyft,
  function export_2d_annotation (line 217) | def export_2d_annotation(root_path, info_path, version):

FILE: tools/data_converter/lyft_data_fixer.py
  function fix_lyft (line 8) | def fix_lyft(root_folder='./data/lyft', version='v1.01'):

FILE: tools/data_converter/nuimage_converter.py
  function parse_args (line 33) | def parse_args():
  function get_img_annos (line 64) | def get_img_annos(nuim, img_info, cat2id, out_dir, data_root, seg_root):
  function export_nuim_to_coco (line 151) | def export_nuim_to_coco(nuim, data_root, out_dir, extra_tag, version, np...
  function main (line 216) | def main():

FILE: tools/data_converter/nuscenes_converter.py
  function quart_to_rpy (line 34) | def quart_to_rpy(qua):
  function locate_message (line 41) | def locate_message(utimes, utime):
  function create_nuscenes_infos (line 47) | def create_nuscenes_infos(root_path,
  function get_available_scenes (line 126) | def get_available_scenes(nusc):
  function _get_future_traj_info (line 166) | def _get_future_traj_info(nusc, sample, predict_steps=8, in_agent_frame=...
  function _get_can_bus_info (line 202) | def _get_can_bus_info(nusc, nusc_can_bus, sample):
  function _fill_trainval_infos (line 226) | def _fill_trainval_infos(nusc,
  function get_global_sensor_pose (line 679) | def get_global_sensor_pose(rec, nusc, inverse=False):
  function obtain_sensor2top (line 700) | def obtain_sensor2top(nusc,
  function export_2d_annotation (line 761) | def export_2d_annotation(root_path, info_path, version, mono3d=True):
  function get_2d_boxes (line 826) | def get_2d_boxes(nusc,
  function post_process_coords (line 959) | def post_process_coords(
  function generate_record (line 992) | def generate_record(ann_rec: dict, x1: float, y1: float, x2: float, y2: ...

FILE: tools/data_converter/nuscenes_prediction_tools.py
  function get_forecasting_annotations (line 14) | def get_forecasting_annotations(nusc: NuScenes, annotations, length):
  function window (line 75) | def window(iterable, size):
  function get_time (line 83) | def get_time(nusc, src_token, dst_token):
  function center_distance (line 91) | def center_distance(gt_box, pred_box) -> float:
  function trajectory_type (line 101) | def trajectory_type(nusc, boxes, time, timesteps=7, past=False):

FILE: tools/data_converter/nuscenes_track_converter.py
  function create_nuscenes_infos (line 37) | def create_nuscenes_infos(root_path,
  function get_available_scenes (line 120) | def get_available_scenes(nusc):
  function _fill_trainval_infos (line 158) | def _fill_trainval_infos(nusc,
  function obtain_sensor2top (line 316) | def obtain_sensor2top(nusc,
  function export_2d_annotation (line 375) | def export_2d_annotation(root_path, info_path, version, mono3d=True):
  function get_2d_boxes (line 438) | def get_2d_boxes(nusc,
  function post_process_coords (line 563) | def post_process_coords(
  function generate_record (line 594) | def generate_record(ann_rec: dict, x1: float, y1: float, x2: float, y2: ...

FILE: tools/data_converter/s3dis_data_utils.py
  class S3DISData (line 10) | class S3DISData(object):
    method __init__ (line 20) | def __init__(self, root_path, split='Area_1'):
    method __len__ (line 43) | def __len__(self):
    method get_infos (line 46) | def get_infos(self, num_workers=4, has_label=True, sample_id_list=None):
    method get_bboxes (line 115) | def get_bboxes(self, points, pts_instance_mask, pts_semantic_mask):
  class S3DISSegData (line 154) | class S3DISSegData(object):
    method __init__ (line 167) | def __init__(self,
    method get_seg_infos (line 193) | def get_seg_infos(self):
    method _convert_to_label (line 205) | def _convert_to_label(self, mask):
    method get_scene_idxs_and_label_weight (line 215) | def get_scene_idxs_and_label_weight(self):

FILE: tools/data_converter/scannet_data_utils.py
  class ScanNetData (line 10) | class ScanNetData(object):
    method __init__ (line 20) | def __init__(self, root_path, split='train'):
    method __len__ (line 45) | def __len__(self):
    method get_aligned_box_label (line 48) | def get_aligned_box_label(self, idx):
    method get_unaligned_box_label (line 54) | def get_unaligned_box_label(self, idx):
    method get_axis_align_matrix (line 60) | def get_axis_align_matrix(self, idx):
    method get_images (line 66) | def get_images(self, idx):
    method get_extrinsics (line 74) | def get_extrinsics(self, idx):
    method get_intrinsics (line 82) | def get_intrinsics(self, idx):
    method get_infos (line 88) | def get_infos(self, num_workers=4, has_label=True, sample_id_list=None):
  class ScanNetSegData (line 201) | class ScanNetSegData(object):
    method __init__ (line 214) | def __init__(self,
    method get_seg_infos (line 243) | def get_seg_infos(self):
    method _convert_to_label (line 257) | def _convert_to_label(self, mask):
    method get_scene_idxs_and_label_weight (line 267) | def get_scene_idxs_and_label_weight(self):

FILE: tools/data_converter/sunrgbd_data_utils.py
  function random_sampling (line 10) | def random_sampling(points, num_points, replace=None):
  class SUNRGBDInstance (line 31) | class SUNRGBDInstance(object):
    method __init__ (line 33) | def __init__(self, line):
  class SUNRGBDData (line 59) | class SUNRGBDData(object):
    method __init__ (line 72) | def __init__(self, root_path, split='train', use_v1=False, num_points=...
    method __len__ (line 98) | def __len__(self):
    method get_image (line 101) | def get_image(self, idx):
    method get_image_shape (line 105) | def get_image_shape(self, idx):
    method get_depth (line 109) | def get_depth(self, idx):
    method get_calibration (line 114) | def get_calibration(self, idx):
    method get_label_objects (line 123) | def get_label_objects(self, idx):
    method get_infos (line 129) | def get_infos(self, num_workers=4, has_label=True, sample_id_list=None):

FILE: tools/data_converter/waymo_converter.py
  class Waymo2KITTI (line 24) | class Waymo2KITTI(object):
    method __init__ (line 39) | def __init__(self,
    method convert (line 94) | def convert(self):
    method convert_one (line 101) | def convert_one(self, file_idx):
    method __len__ (line 128) | def __len__(self):
    method save_image (line 132) | def save_image(self, frame, file_idx, frame_idx):
    method save_calib (line 147) | def save_calib(self, frame, file_idx, frame_idx):
    method save_lidar (line 203) | def save_lidar(self, frame, file_idx, frame_idx):
    method save_label (line 257) | def save_label(self, frame, file_idx, frame_idx):
    method save_pose (line 357) | def save_pose(self, frame, file_idx, frame_idx):
    method save_timestamp (line 376) | def save_timestamp(self, frame, file_idx, frame_idx):
    method create_folder (line 396) | def create_folder(self):
    method convert_range_image_to_point_cloud (line 417) | def convert_range_image_to_point_cloud(self,
    method cart_to_homo (line 537) | def cart_to_homo(self, mat):

FILE: tools/deployment/mmdet3d2torchserve.py
  function mmdet3d2torchserve (line 15) | def mmdet3d2torchserve(
  function parse_args (line 71) | def parse_args():

FILE: tools/deployment/mmdet3d_handler.py
  class MMdet3dHandler (line 13) | class MMdet3dHandler(BaseHandler):
    method initialize (line 25) | def initialize(self, context):
    method preprocess (line 46) | def preprocess(self, data):
    method inference (line 74) | def inference(self, data):
    method postprocess (line 90) | def postprocess(self, data):

FILE: tools/deployment/test_torchserver.py
  function parse_args (line 9) | def parse_args():
  function parse_result (line 27) | def parse_result(input):
  function main (line 33) | def main(args):

FILE: tools/misc/browse_dataset.py
  function parse_args (line 18) | def parse_args():
  function build_data_cfg (line 60) | def build_data_cfg(config_path, skip_type, aug, cfg_options):
  function to_depth_mode (line 96) | def to_depth_mode(points, bboxes):
  function show_det_data (line 107) | def show_det_data(input, out_dir, show=False):
  function show_seg_data (line 125) | def show_seg_data(input, out_dir, show=False):
  function show_proj_bbox_img (line 143) | def show_proj_bbox_img(input, out_dir, show=False, is_nus_mono=False):
  function main (line 195) | def main():

FILE: tools/misc/fuse_conv_bn.py
  function fuse_conv_bn (line 11) | def fuse_conv_bn(conv, bn):
  function fuse_module (line 27) | def fuse_module(m):
  function parse_args (line 48) | def parse_args():
  function main (line 58) | def main():

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

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

FILE: tools/model_converters/convert_h3dnet_checkpoints.py
  function parse_args (line 12) | def parse_args():
  function parse_config (line 21) | def parse_config(config_strings):
  function main (line 95) | def main():

FILE: tools/model_converters/convert_votenet_checkpoints.py
  function parse_args (line 12) | def parse_args():
  function parse_config (line 21) | def parse_config(config_strings):
  function main (line 71) | def main():

FILE: tools/model_converters/publish_model.py
  function parse_args (line 8) | def parse_args():
  function process_checkpoint (line 17) | def process_checkpoint(in_file, out_file):
  function main (line 30) | def main():

FILE: tools/model_converters/regnet2mmdet.py
  function convert_stem (line 8) | def convert_stem(model_key, model_weight, state_dict, converted_names):
  function convert_head (line 16) | def convert_head(model_key, model_weight, state_dict, converted_names):
  function convert_reslayer (line 23) | def convert_reslayer(model_key, model_weight, state_dict, converted_names):
  function convert (line 55) | def convert(src, dst):
  function main (line 81) | def main():

FILE: tools/test.py
  function parse_args (line 38) | def parse_args():
  function main (line 144) | def main():

FILE: tools/train.py
  function parse_args (line 37) | def parse_args():
  function main (line 120) | def main():

FILE: tools/update_data_coords.py
  function update_sunrgbd_infos (line 11) | def update_sunrgbd_infos(root_dir, out_dir, pkl_files):
  function update_outdoor_dbinfos (line 34) | def update_outdoor_dbinfos(root_dir, out_dir, pkl_files):
  function update_nuscenes_or_lyft_infos (line 63) | def update_nuscenes_or_lyft_infos(root_dir, out_dir, pkl_files):