SYMBOL INDEX (552 symbols across 66 files)

FILE: plugin/core/apis/mmdet_train.py
  class MyRunnerWrapper (line 33) | class MyRunnerWrapper(IterBasedRunner):
    method train (line 34) | def train(self, data_loader, **kwargs):
  function custom_train_detector (line 54) | def custom_train_detector(model,

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

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

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

FILE: plugin/datasets/argo_dataset.py
  class AV2Dataset (line 15) | class AV2Dataset(BaseMapDataset):
    method __init__ (line 30) | def __init__(self, **kwargs,):
    method load_annotations (line 36) | def load_annotations(self, ann_file):
    method load_matching (line 96) | def load_matching(self, matching_file):
    method get_sample (line 107) | def get_sample(self, idx):

FILE: plugin/datasets/base_dataset.py
  class BaseMapDataset (line 18) | class BaseMapDataset(Dataset):
    method __init__ (line 32) | def __init__(self,
    method _set_sequence_info (line 96) | def _set_sequence_info(self):
    method _set_sequence_group_flag (line 117) | def _set_sequence_group_flag(self):
    method load_annotations (line 161) | def load_annotations(self, ann_file):
    method load_matching (line 164) | def load_matching(self, matching_file):
    method get_sample (line 167) | def get_sample(self, idx):
    method format_results (line 170) | def format_results(self, results, denormalize=True, prefix=None, save_...
    method evaluate (line 298) | def evaluate(self, results, logger=None, **kwargs):
    method _evaluate (line 319) | def _evaluate(self, result_path, logger=None, eval_semantic=False):
    method show_gt (line 327) | def show_gt(self, idx, out_dir='demo/'):
    method show_result (line 361) | def show_result(self, submission, idx, score_thr=0, draw_score=False, ...
    method show_track (line 406) | def show_track(self, submission, idx, out_dir='demo/'):
    method __len__ (line 429) | def __len__(self):
    method _rand_another (line 437) | def _rand_another(self, idx):
    method __getitem__ (line 445) | def __getitem__(self, idx):

FILE: plugin/datasets/builder.py
  function build_dataloader (line 23) | def build_dataloader(dataset,
  function worker_init_fn (line 121) | def worker_init_fn(worker_id, num_workers, rank, seed):

FILE: plugin/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: plugin/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: plugin/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: plugin/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: plugin/datasets/map_utils/av2map_extractor.py
  class AV2MapExtractor (line 24) | class AV2MapExtractor(object):
    method __init__ (line 31) | def __init__(self, roi_size: Union[Tuple, List], id2map: Dict) -> None:
    method generate_nearby_dividers (line 38) | def generate_nearby_dividers(self,avm, e2g_translation, e2g_rotation,p...
    method proc_polygon (line 175) | def proc_polygon(self,polygon, ego_SE3_city):
    method proc_line (line 187) | def proc_line(self,line,ego_SE3_city):
    method extract_local_divider (line 193) | def extract_local_divider(self,nearby_dividers, ego_SE3_city, patch_bo...
    method extract_local_boundary (line 241) | def extract_local_boundary(self,avm, ego_SE3_city, patch_box, patch_an...
    method get_scene_dividers (line 326) | def get_scene_dividers(self,avm,patch_box,patch_angle):
    method get_scene_ped_crossings (line 354) | def get_scene_ped_crossings(self,avm,e2g_translation,e2g_rotation,poly...
    method get_map_geom (line 391) | def get_map_geom(self,

FILE: plugin/datasets/map_utils/nuscmap_extractor.py
  class NuscMapExtractor (line 17) | class NuscMapExtractor(object):
    method __init__ (line 24) | def __init__(self, data_root: str, roi_size: Union[List, Tuple]) -> None:
    method get_map_geom (line 36) | def get_map_geom(self,
  class VectorizedLocalMap (line 57) | class VectorizedLocalMap(object):
    method __init__ (line 65) | def __init__(self,
    method gen_vectorized_samples (line 91) | def gen_vectorized_samples(self, lidar2global_translation, lidar2globa...
    method get_centerline_geom (line 140) | def get_centerline_geom(self, patch_box, patch_angle, layer_names):
    method get_map_geom (line 153) | def get_map_geom(self, patch_box, patch_angle, layer_names):
    method get_divider_line (line 170) | def get_divider_line(self,patch_box,patch_angle,layer_name):
    method get_contour_line (line 198) | def get_contour_line(self,patch_box,patch_angle,layer_name):
    method get_ped_crossing_line (line 243) | def get_ped_crossing_line(self, patch_box, patch_angle):
    method _union_ped_stmmapnet (line 266) | def _union_ped_stmmapnet(self, ped_geoms: List[Polygon]) -> List[Polyg...
    method _handle_small_peds (line 329) | def _handle_small_peds(self, ped_geoms):
    method get_ped_crossing_line_stmmapnet (line 386) | def get_ped_crossing_line_stmmapnet(self, patch_box, patch_angle):
    method line_geoms_to_instances (line 428) | def line_geoms_to_instances(self, line_geom):
    method _one_type_line_geom_to_instances (line 436) | def _one_type_line_geom_to_instances(self, line_geom):
    method ped_poly_geoms_to_instances (line 450) | def ped_poly_geoms_to_instances(self, ped_geom):
    method poly_geoms_to_instances (line 487) | def poly_geoms_to_instances(self, polygon_geom):
    method centerline_geoms_to_instances (line 525) | def centerline_geoms_to_instances(self,geoms_dict):
    method centerline_geoms2vec (line 532) | def centerline_geoms2vec(self, centerline_geoms_list):
    method union_centerline (line 542) | def union_centerline(self, centerline_geoms):
  class CNuScenesMapExplorer (line 608) | class CNuScenesMapExplorer(NuScenesMapExplorer):
    method __ini__ (line 609) | def __ini__(self, *args, **kwargs):
    method _get_centerline (line 612) | def _get_centerline(self,
  function to_patch_coord (line 679) | def to_patch_coord(new_polygon, patch_angle, patch_x, patch_y):

FILE: plugin/datasets/map_utils/utils.py
  function split_collections (line 9) | def split_collections(geom: BaseGeometry) -> List[Optional[BaseGeometry]]:
  function get_drivable_area_contour (line 32) | def get_drivable_area_contour(drivable_areas: List[Polygon],
  function get_ped_crossing_contour (line 89) | def get_ped_crossing_contour(polygon: Polygon,
  function remove_repeated_lines (line 131) | def remove_repeated_lines(lines: List[LineString]) -> List[LineString]:
  function remove_repeated_lanesegment (line 167) | def remove_repeated_lanesegment(lane_dict):
  function reassign_graph_attribute (line 203) | def reassign_graph_attribute(lane_dict):
  function remove_boundary_dividers (line 220) | def remove_boundary_dividers(dividers: List[LineString],
  function connect_lines (line 248) | def connect_lines(lines: List[LineString]) -> List[LineString]:
  function transform_from (line 299) | def transform_from(xyz: NDArray,

FILE: plugin/datasets/nusc_dataset.py
  class NuscDataset (line 13) | class NuscDataset(BaseMapDataset):
    method __init__ (line 28) | def __init__(self, data_root, **kwargs):
    method load_annotations (line 33) | def load_annotations(self, ann_file):
    method load_matching (line 50) | def load_matching(self, matching_file):
    method get_sample (line 60) | def get_sample(self, idx):

FILE: plugin/datasets/pipelines/formating.py
  class FormatBundleMap (line 9) | class FormatBundleMap(object):
    method __init__ (line 20) | def __init__(self, process_img=True,
    method __call__ (line 28) | def __call__(self, results):
    method __repr__ (line 77) | def __repr__(self):

FILE: plugin/datasets/pipelines/loading.py
  class LoadMultiViewImagesFromFiles (line 6) | class LoadMultiViewImagesFromFiles(object):
    method __init__ (line 17) | def __init__(self, to_float32=False, color_type='unchanged'):
    method __call__ (line 21) | def __call__(self, results):
    method __repr__ (line 57) | def __repr__(self):

FILE: plugin/datasets/pipelines/rasterize.py
  class RasterizeMap (line 14) | class RasterizeMap(object):
    method __init__ (line 25) | def __init__(self,
    method line_ego_to_mask (line 41) | def line_ego_to_mask(self,
    method polygons_ego_to_mask (line 71) | def polygons_ego_to_mask(self,
    method get_semantic_mask (line 102) | def get_semantic_mask(self, map_geoms: Dict) -> NDArray:
    method __call__ (line 152) | def __call__(self, input_dict: Dict) -> Dict:
    method __repr__ (line 159) | def __repr__(self):
  class PV_Map (line 170) | class PV_Map(object):
    method __init__ (line 181) | def __init__(self,
    method perspective (line 203) | def perspective(self,cam_coords, proj_mat):
    method get_valid_pix_coords (line 212) | def get_valid_pix_coords(pix_coords):
    method line_ego_to_pvmask (line 218) | def line_ego_to_pvmask(self,
    method lines_ego_to_pv (line 236) | def lines_ego_to_pv(self, lines_ego, pv_mask, ego2imgs, color=1, thick...
    method get_pvmask_old (line 256) | def get_pvmask_old(self,map_geoms: Dict,ego2img: List, img_filenames: ...
    method get_pvmask (line 287) | def get_pvmask(self, map_geoms: Dict,ego2img: List, img_filenames: Lis...
    method __call__ (line 317) | def __call__(self, input_dict: Dict) -> Dict:
    method __repr__ (line 323) | def __repr__(self):
    method visualize_all_pv_masks (line 332) | def visualize_all_pv_masks(self, gt_pv_mask, img_filenames):
    method _visualize_pv_mask (line 349) | def _visualize_pv_mask(self, pv_mask):

FILE: plugin/datasets/pipelines/transform.py
  class Normalize3D (line 8) | class Normalize3D(object):
    method __init__ (line 18) | def __init__(self, mean, std, to_rgb=True):
    method __call__ (line 23) | def __call__(self, results):
    method __repr__ (line 38) | def __repr__(self):
  class PadMultiViewImages (line 45) | class PadMultiViewImages(object):
    method __init__ (line 59) | def __init__(self, size=None, size_divisor=None, pad_val=0, change_int...
    method _pad_img (line 69) | def _pad_img(self, results):
    method __call__ (line 109) | def __call__(self, results):
    method __repr__ (line 119) | def __repr__(self):
  class ResizeMultiViewImages (line 130) | class ResizeMultiViewImages(object):
    method __init__ (line 138) | def __init__(self, size=None, scale=None, change_intrinsics=True):
    method __call__ (line 144) | def __call__(self, results:dict):
    method __repr__ (line 184) | def __repr__(self):
  class PhotoMetricDistortionMultiViewImage (line 193) | class PhotoMetricDistortionMultiViewImage:
    method __init__ (line 212) | def __init__(self,
    method __call__ (line 222) | def __call__(self, results):
    method __repr__ (line 281) | def __repr__(self):

FILE: plugin/datasets/pipelines/vectorize.py
  class VectorizeMap (line 8) | class VectorizeMap(object):
    method __init__ (line 23) | def __init__(self,
    method interp_fixed_num (line 49) | def interp_fixed_num(self, line: LineString) -> NDArray:
    method interp_fixed_dist (line 65) | def interp_fixed_dist(self, line: LineString) -> NDArray:
    method get_vectorized_lines (line 84) | def get_vectorized_lines(self, map_geoms: Dict) -> Dict:
    method normalize_line (line 121) | def normalize_line(self, line: NDArray) -> NDArray:
    method permute_line (line 141) | def permute_line(self, line: np.ndarray, padding=1e5):
    method __call__ (line 176) | def __call__(self, input_dict):
    method __repr__ (line 182) | def __repr__(self):

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

FILE: plugin/datasets/samplers/group_sampler.py
  class GroupSampler (line 19) | class GroupSampler(Sampler):
    method __init__ (line 21) | def __init__(self, dataset, samples_per_gpu=1):
    method __iter__ (line 34) | def __iter__(self):
    method __len__ (line 42) | def __len__(self):
  class DistributedGroupSampler (line 46) | class DistributedGroupSampler(Sampler):
    method __init__ (line 64) | def __init__(self,
    method __iter__ (line 93) | def __iter__(self):
    method __len__ (line 136) | def __len__(self):
    method set_epoch (line 139) | def set_epoch(self, epoch):
  function sync_random_seed (line 143) | def sync_random_seed(seed=None, device='cuda'):
  class InfiniteGroupEachSampleInBatchSampler (line 178) | class InfiniteGroupEachSampleInBatchSampler(Sampler):
    method __init__ (line 199) | def __init__(self,
    method _infinite_group_indices (line 256) | def _infinite_group_indices(self):
    method _group_indices_per_global_sample_idx (line 262) | def _group_indices_per_global_sample_idx(self, global_sample_idx):
    method _sample_sub_sequence (line 268) | def _sample_sub_sequence(self, group_idx):
    method __iter__ (line 296) | def __iter__(self):
    method __len__ (line 319) | def __len__(self):
    method set_epoch (line 323) | def set_epoch(self, epoch):

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

FILE: plugin/datasets/visualize/renderer.py
  function remove_nan_values (line 13) | def remove_nan_values(uv):
  function points_ego2img (line 21) | def points_ego2img(pts_ego, extrinsics, intrinsics):
  function draw_polyline_ego_on_img (line 32) | def draw_polyline_ego_on_img(polyline_ego, img_bgr, extrinsics, intrinsi...
  function draw_visible_polyline_cv2 (line 61) | def draw_visible_polyline_cv2(line, valid_pts_bool, image, color, thickn...
  class Renderer (line 111) | class Renderer(object):
    method __init__ (line 120) | def __init__(self, cat2id, roi_size, dataset='av2'):
    method render_bev_from_vectors (line 129) | def render_bev_from_vectors(self, vectors, out_dir, draw_scores=False,...
    method render_camera_views_from_vectors (line 188) | def render_camera_views_from_vectors(self, vectors, imgs, extrinsics,
    method render_bev_from_mask (line 220) | def render_bev_from_mask(self, semantic_mask, out_dir, flip=False):

FILE: plugin/models/assigner/assigner.py
  class HungarianLinesAssigner (line 12) | class HungarianLinesAssigner(BaseAssigner):
    method __init__ (line 31) | def __init__(self,
    method assign (line 41) | def assign(self,

FILE: plugin/models/assigner/match_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 165) | class BBoxCostC:
    method __init__ (line 181) | def __init__(self, weight=1., box_format='xyxy'):
    method __call__ (line 186) | def __call__(self, bbox_pred, gt_bboxes):
  class IoUCostC (line 206) | class IoUCostC:
    method __init__ (line 222) | def __init__(self, iou_mode='giou', weight=1., box_format='xywh'):
    method __call__ (line 228) | def __call__(self, bboxes, gt_bboxes):
  class DynamicLinesCost (line 250) | class DynamicLinesCost(object):
    method __init__ (line 256) | def __init__(self, weight=1.):
    method __call__ (line 259) | def __call__(self, lines_pred, lines_gt, masks_pred, masks_gt):
    method cal_dist (line 281) | def cal_dist(self, x1, x2):
    method get_dynamic_line (line 298) | def get_dynamic_line(self, mat, m1, m2):
  class BBoxLogitsCost (line 324) | class BBoxLogitsCost(object):
    method __init__ (line 330) | def __init__(self, weight=1.):
    method calNLL (line 333) | def calNLL(self, logits, value):
    method __call__ (line 350) | def __call__(self, bbox_pred, bbox_gt, **kwargs):
  class MapQueriesCost (line 365) | class MapQueriesCost(object):
    method __init__ (line 367) | def __init__(self, cls_cost, reg_cost, iou_cost=None):
    method __call__ (line 376) | def __call__(self, preds: dict, gts: dict):

FILE: plugin/models/backbones/bevformer/custom_base_transformer_layer.py
  class MyCustomBaseTransformerLayer (line 38) | class MyCustomBaseTransformerLayer(BaseModule):
    method __init__ (line 72) | def __init__(self,
    method forward (line 165) | def forward(self,
  class MyCustomBaseTransformerLayerWithoutSelfAttn (line 265) | class MyCustomBaseTransformerLayerWithoutSelfAttn(BaseModule):
    method __init__ (line 299) | def __init__(self,
    method forward (line 392) | def forward(self,

FILE: plugin/models/backbones/bevformer/encoder.py
  class BEVFormerEncoder (line 27) | class BEVFormerEncoder(TransformerLayerSequence):
    method __init__ (line 38) | def __init__(self, *args, pc_range=None, num_points_in_pillar=4, retur...
    method get_reference_points (line 55) | def get_reference_points(H, W, Z=8, num_points_in_pillar=4, dim='3d', ...
    method point_sampling (line 102) | def point_sampling(self, reference_points, pc_range, img_metas):
    method forward (line 158) | def forward(self,
  class BEVFormerLayer (line 259) | class BEVFormerLayer(MyCustomBaseTransformerLayer):
    method __init__ (line 280) | def __init__(self,
    method forward (line 303) | def forward(self,

FILE: plugin/models/backbones/bevformer/grid_mask.py
  class Grid (line 7) | class Grid(object):
    method __init__ (line 8) | def __init__(self, use_h, use_w, rotate = 1, offset=False, ratio = 0.5...
    method set_prob (line 18) | def set_prob(self, epoch, max_epoch):
    method __call__ (line 21) | def __call__(self, img, label):
  class GridMask (line 70) | class GridMask(nn.Module):
    method __init__ (line 71) | def __init__(self, use_h, use_w, rotate = 1, offset=False, ratio = 0.5...
    method set_prob (line 83) | def set_prob(self, epoch, max_epoch):
    method set_ratio_and_prob (line 86) | def set_ratio_and_prob(self, ratio, prob):
    method forward (line 91) | def forward(self, x):

FILE: plugin/models/backbones/bevformer/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: plugin/models/backbones/bevformer/spatial_cross_attention.py
  class SpatialCrossAttention (line 31) | class SpatialCrossAttention(BaseModule):
    method __init__ (line 44) | def __init__(self,
    method init_weight (line 70) | def init_weight(self):
    method forward (line 75) | def forward(self,
  class MSDeformableAttention3D (line 178) | class MSDeformableAttention3D(BaseModule):
    method __init__ (line 203) | def __init__(self,
    method init_weights (line 252) | def init_weights(self):
    method forward (line 272) | def forward(self,
  class MSIPM3D (line 403) | class MSIPM3D(BaseModule):
    method __init__ (line 428) | def __init__(self,
    method init_weights (line 477) | def init_weights(self):
    method forward (line 498) | def forward(self,

FILE: plugin/models/backbones/bevformer/temporal_net.py
  class MyResBlock (line 12) | class MyResBlock(nn.Module):
    method __init__ (line 13) | def __init__(self,
    method forward (line 31) | def forward(self, x: Tensor) -> Tensor:
  class TemporalNet (line 48) | class TemporalNet(nn.Module):
    method __init__ (line 49) | def __init__(self, history_steps, hidden_dims, num_blocks):
    method init_weights (line 67) | def init_weights(self):
    method forward (line 73) | def forward(self, history_feats, curr_feat):

FILE: plugin/models/backbones/bevformer/temporal_self_attention.py
  class TemporalSelfAttention (line 25) | class TemporalSelfAttention(BaseModule):
    method __init__ (line 54) | def __init__(self,
    method init_weights (line 106) | def init_weights(self):
    method forward (line 127) | def forward(self,

FILE: plugin/models/backbones/bevformer/transformer.py
  class PerceptionTransformer (line 26) | class PerceptionTransformer(BaseModule):
    method __init__ (line 37) | def __init__(self,
    method init_layers (line 56) | def init_layers(self):
    method init_weights (line 64) | def init_weights(self):
    method get_bev_features (line 80) | def get_bev_features(
    method forward (line 149) | def forward(self,

FILE: plugin/models/backbones/bevformer_backbone.py
  class UpsampleBlock (line 15) | class UpsampleBlock(nn.Module):
    method __init__ (line 16) | def __init__(self, ins, outs):
    method init_weights (line 23) | def init_weights(self):
    method forward (line 28) | def forward(self, x):
    method upsample2x (line 36) | def upsample2x(self, x):
  class BEVFormerBackbone (line 43) | class BEVFormerBackbone(nn.Module):
    method __init__ (line 55) | def __init__(self,
    method _init_layers (line 107) | def _init_layers(self):
    method init_weights (line 113) | def init_weights(self):
    method extract_img_feat (line 123) | def extract_img_feat(self, img, img_metas, len_queue=None):
    method forward (line 159) | def forward(self, img, img_metas, timestep, history_bev_feats, history...

FILE: plugin/models/heads/MapDetectorHead.py
  class MapDetectorHead (line 18) | class MapDetectorHead(nn.Module):
    method __init__ (line 20) | def __init__(self,
    method init_weights (line 81) | def init_weights(self):
    method _init_embedding (line 115) | def _init_embedding(self):
    method _init_branch (line 129) | def _init_branch(self,):
    method _prepare_context (line 162) | def _prepare_context(self, bev_features):
    method forward_train (line 175) | def forward_train(self, bev_features, img_metas, gts, track_query_info...
    method forward_test (line 275) | def forward_test(self, bev_features, img_metas, track_query_info=None,...
    method _get_target_single (line 359) | def _get_target_single(self,
    method get_targets (line 443) | def get_targets(self, preds, gts, track_info=None, gt_bboxes_ignore_li...
    method loss_single (line 507) | def loss_single(self,
    method loss (line 593) | def loss(self,
    method post_process (line 644) | def post_process(self, preds_dict, tokens, track_dict=None, thr=0.0):
    method prepare_temporal_propagation (line 709) | def prepare_temporal_propagation(self, preds_dict, scene_name, local_i...
    method get_track_info (line 785) | def get_track_info(self, scene_name, local_idx):
    method get_self_iter_track_query (line 801) | def get_self_iter_track_query(self, preds_dict):
    method clear_temporal_cache (line 821) | def clear_temporal_cache(self):
    method train (line 824) | def train(self, *args, **kwargs):
    method eval (line 827) | def eval(self):
    method forward (line 830) | def forward(self, *args, return_loss=True, **kwargs):

FILE: plugin/models/heads/MapSegHead.py
  class MapSegHead (line 17) | class MapSegHead(nn.Module):
    method __init__ (line 19) | def __init__(self,
    method init_weights (line 65) | def init_weights(self):
    method forward_train (line 71) | def forward_train(self, bev_features, gts, history_coords):
    method forward_test (line 87) | def forward_test(self, bev_features):
    method train (line 97) | def train(self, *args, **kwargs):
    method eval (line 100) | def eval(self):
    method forward (line 103) | def forward(self, *args, return_loss=True, **kwargs):

FILE: plugin/models/heads/base_map_head.py
  class BaseMapHead (line 10) | class BaseMapHead(nn.Module, metaclass=ABCMeta):
    method __init__ (line 13) | def __init__(self):
    method init_weights (line 17) | def init_weights(self, pretrained=None):
    method forward (line 28) | def forward(self, *args, **kwargs):
    method loss (line 32) | def loss(self, pred, gt):
    method post_process (line 47) | def post_process(self, pred):

FILE: plugin/models/losses/detr_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: plugin/models/losses/seg_loss.py
  function py_sigmoid_focal_loss (line 12) | def py_sigmoid_focal_loss(pred,
  class MaskFocalLoss (line 61) | class MaskFocalLoss(FocalLoss):
    method __init__ (line 62) | def __init__(self,**kwargs):
    method forward (line 65) | def forward(self,
  class MaskDiceLoss (line 94) | class MaskDiceLoss(nn.Module):
    method __init__ (line 106) | def __init__(self, loss_weight):
    method forward (line 111) | def forward(self, pred, target):

FILE: plugin/models/mapers/MapTracker.py
  class MapTracker (line 22) | class MapTracker(BaseMapper):
    method __init__ (line 24) | def __init__(self,
    method init_weights (line 111) | def init_weights(self, pretrained=None):
    method temporal_propagate (line 124) | def temporal_propagate(self, curr_bev_feats, img_metas, all_history_cu...
    method add_noise_to_pose (line 280) | def add_noise_to_pose(self, rot, trans):
    method process_history_info (line 294) | def process_history_info(self, img_metas, history_img_metas):
    method forward_train (line 344) | def forward_train(self, img, vectors, semantic_mask, points=None, img_...
    method forward_test (line 587) | def forward_test(self, img, points=None, img_metas=None, seq_info=None...
    method batch_data (line 694) | def batch_data(self, vectors, imgs, img_metas, device, points=None):
    method get_two_frame_matching (line 741) | def get_two_frame_matching(self, local2global_prev, local2global_curr,...
    method _compute_cur2prev (line 766) | def _compute_cur2prev(self, gt2local_curr, gt2local_prev, local2gt_prev,
    method _reverse_id_mapping (line 786) | def _reverse_id_mapping(self, id_mapping):
    method prepare_track_queries_and_targets (line 793) | def prepare_track_queries_and_targets(self, gts, prev_inds_list, prev_...
    method _batchify_tracks (line 909) | def _batchify_tracks(self, targets):
    method train (line 926) | def train(self, *args, **kwargs):
    method eval (line 935) | def eval(self):
    method _freeze_bev (line 938) | def _freeze_bev(self,):
    method _unfreeze_bev (line 946) | def _unfreeze_bev(self,):
    method _denorm_lines (line 954) | def _denorm_lines(self, line_pts):
    method _norm_lines (line 962) | def _norm_lines(self, line_pts):
    method _process_track_query_info (line 970) | def _process_track_query_info(self, track_info):
    method select_memory_entries (line 987) | def select_memory_entries(self, history_metas, curr_meta):
    method _viz_temporal_supervision (line 1026) | def _viz_temporal_supervision(self, outputs_prev, all_track_info, gts,...

FILE: plugin/models/mapers/base_mapper.py
  class BaseMapper (line 12) | class BaseMapper(nn.Module, metaclass=ABCMeta):
    method __init__ (line 15) | def __init__(self):
    method with_neck (line 20) | def with_neck(self):
    method with_shared_head (line 27) | def with_shared_head(self):
    method with_bbox (line 32) | def with_bbox(self):
    method with_mask (line 38) | def with_mask(self):
    method extract_feat (line 44) | def extract_feat(self, imgs):
    method forward_train (line 48) | def forward_train(self, *args, **kwargs):
    method simple_test (line 52) | def simple_test(self, img, img_metas, **kwargs):
    method aug_test (line 56) | def aug_test(self, imgs, img_metas, **kwargs):
    method init_weights (line 60) | def init_weights(self, pretrained=None):
    method forward_test (line 71) | def forward_test(self, *args, **kwargs):
    method forward (line 81) | def forward(self, *args, return_loss=True, **kwargs):
    method train_step (line 97) | def train_step(self, data_dict, optimizer):
    method val_step (line 131) | def val_step(self, data, optimizer):
    method show_result (line 145) | def show_result(self,

FILE: plugin/models/mapers/vector_memory.py
  function get_emb (line 9) | def get_emb(sin_inp):
  class PositionalEncoding1D (line 17) | class PositionalEncoding1D(nn.Module):
    method __init__ (line 18) | def __init__(self, channels):
    method forward (line 30) | def forward(self, tensor):
  class VectorInstanceMemory (line 53) | class VectorInstanceMemory(nn.Module):
    method __init__ (line 55) | def __init__(self,
    method set_bank_size (line 73) | def set_bank_size(self, bank_size):
    method init_memory (line 76) | def init_memory(self, bs):
    method update_memory (line 97) | def update_memory(self, batch_i, is_first_frame, propagated_ids, prev_...
    method prepare_transformation_batch (line 158) | def prepare_transformation_batch(self,history_e2g_trans,history_e2g_ro...
    method clear_dict (line 180) | def clear_dict(self,):
    method trans_memory_bank (line 185) | def trans_memory_bank(self, query_prop, b_i, metas):
    method add_noise_to_pose (line 270) | def add_noise_to_pose(self, rot, trans):
    method select_memory_entries (line 284) | def select_memory_entries(self, mem_trans, curr_meta):

FILE: plugin/models/necks/gru.py
  class ConvGRU (line 8) | class ConvGRU(nn.Module):
    method __init__ (line 9) | def __init__(self, out_channels):
    method init_weights (line 23) | def init_weights(self):
    method forward (line 30) | def forward(self, h, x):

FILE: plugin/models/transformer_utils/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: plugin/models/transformer_utils/MapTransformer.py
  class MapTransformerDecoder_new (line 25) | class MapTransformerDecoder_new(BaseModule):
    method __init__ (line 33) | def __init__(self,
    method forward (line 58) | def forward(self,
  class MapTransformerLayer (line 137) | class MapTransformerLayer(BaseTransformerLayer):
    method __init__ (line 173) | def __init__(self,
    method forward (line 199) | def forward(self,
  class MapTransformer (line 346) | class MapTransformer(Transformer):
    method __init__ (line 357) | def __init__(self,
    method init_layers (line 369) | def init_layers(self):
    method init_weights (line 374) | def init_weights(self):
    method forward (line 383) | def forward(self,

FILE: plugin/models/transformer_utils/base_transformer.py
  class PlaceHolderEncoder (line 17) | class PlaceHolderEncoder(nn.Module):
    method __init__ (line 19) | def __init__(self, *args, embed_dims=None, **kwargs):
    method forward (line 23) | def forward(self, *args, query=None, **kwargs):

FILE: plugin/models/transformer_utils/deformable_transformer.py
  function inverse_sigmoid (line 30) | def inverse_sigmoid(x, eps=1e-5):
  class DeformableDetrTransformerDecoder_ (line 48) | class DeformableDetrTransformerDecoder_(TransformerLayerSequence):
    method __init__ (line 56) | def __init__(self, *args,
    method forward (line 64) | def forward(self,
  class DeformableDetrTransformer_ (line 130) | class DeformableDetrTransformer_(Transformer):
    method __init__ (line 141) | def __init__(self,
    method init_layers (line 155) | def init_layers(self):
    method init_weights (line 169) | def init_weights(self):
    method get_reference_points (line 184) | def get_reference_points(spatial_shapes, valid_ratios, device):
    method get_valid_ratio (line 216) | def get_valid_ratio(self, mask):
    method get_proposal_pos_embed (line 226) | def get_proposal_pos_embed(self,
    method forward (line 244) | def forward(self,

FILE: plugin/models/transformer_utils/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: plugin/models/utils/query_update.py
  class Embedder (line 8) | class Embedder:
    method __init__ (line 9) | def __init__(self, **kwargs):
    method create_embedding_fn (line 13) | def create_embedding_fn(self):
    method embed (line 37) | def embed(self, inputs):
  class MotionMLP (line 41) | class MotionMLP(nn.Module):
    method __init__ (line 48) | def __init__(self, c_dim, f_dim=512, identity=True):
    method init_weights (line 73) | def init_weights(self):
    method forward (line 80) | def forward(self, x, pose_info):

FILE: plugin/models/utils/renderer_track.py
  function remove_nan_values (line 10) | def remove_nan_values(uv):
  function points_ego2img (line 18) | def points_ego2img(pts_ego, extrinsics, intrinsics):
  function draw_polyline_ego_on_img (line 29) | def draw_polyline_ego_on_img(polyline_ego, img_bgr, extrinsics, intrinsi...
  function draw_visible_polyline_cv2 (line 58) | def draw_visible_polyline_cv2(line, valid_pts_bool, image, color, thickn...
  class Renderer (line 108) | class Renderer(object):
    method __init__ (line 117) | def __init__(self, cat2id, roi_size, dataset='av2'):
    method render_bev_from_vectors (line 126) | def render_bev_from_vectors(self, vectors, labels, out_path, id_info=N...
    method render_bev_from_mask (line 174) | def render_bev_from_mask(self, semantic_mask, out_path):

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

FILE: tools/data_converter/argoverse_converter.py
  function parse_args (line 29) | def parse_args():
  function create_av2_infos_mp (line 47) | def create_av2_infos_mp(root_path,
  function get_data_from_logid (line 135) | def get_data_from_logid(log_id, loaders, data_root):

FILE: tools/data_converter/nuscenes_converter.py
  function parse_args (line 19) | def parse_args():
  function create_nuscenes_infos_map (line 36) | def create_nuscenes_infos_map(root_path,

FILE: tools/mmdet_test.py
  function single_gpu_test (line 16) | def single_gpu_test(model,
  function multi_gpu_test (line 69) | def multi_gpu_test(model, data_loader, tmpdir=None, gpu_collect=False):
  function collect_results_cpu (line 117) | def collect_results_cpu(result_part, size, tmpdir=None):
  function collect_results_gpu (line 160) | def collect_results_gpu(result_part, size):

FILE: tools/mmdet_train.py
  function set_random_seed (line 18) | def set_random_seed(seed, deterministic=False):
  function train_detector (line 37) | def train_detector(model,

FILE: tools/test.py
  function parse_args (line 21) | def parse_args():
  function main (line 103) | def main():

FILE: tools/tracking/calculate_cmap.py
  function parse_args (line 53) | def parse_args():
  function instance_match (line 79) | def instance_match(pred_lines, scores, gt_lines, threshold, metric='cham...
  function _evaluate_single (line 119) | def _evaluate_single(pred_vectors, scores, gt_vectors, threshold, metric...
  function match_gt_w_pred (line 135) | def match_gt_w_pred(curr_data,curr_data_gt,thresh):
  function get_scene_matching_result (line 170) | def get_scene_matching_result(gts,pred_results,scene_name2token,scene_na...
  function pred2gt_global_matching (line 207) | def pred2gt_global_matching(ids_info,ids_info_gt,pred2gt_seq):
  function get_tpfp_from_scene_single (line 232) | def get_tpfp_from_scene_single(scene_name,args,scene_name2token,pred_res...
  function get_mAP (line 281) | def get_mAP(tpfp_score_record,num_gts,threshold):
  function main (line 306) | def main():

FILE: tools/tracking/cmap_utils/data_utils.py
  function get_gts (line 18) | def get_gts(dataset,new_split=False,N_WORKERS=16):
  function prepare_data_multi (line 55) | def prepare_data_multi(token,idx,pred,gts,origin,roi_size,interp_num,dat...
  function get_data (line 84) | def get_data(pred_matching_result_raw,gts,origin,roi_size,num_interp,res...

FILE: tools/tracking/cmap_utils/match_utils.py
  function get_prev2curr_matrix (line 13) | def get_prev2curr_matrix(prev_meta,curr_meta):
  function find_matchings_iou (line 32) | def find_matchings_iou(src_masks, tgt_masks, thresh=0.1):
  function find_matchings_chamfer (line 69) | def find_matchings_chamfer(pred_vectors, gt_vectors, score_dict,thresh=0...
  function get_consecutive_vectors (line 101) | def get_consecutive_vectors(prev_vectors,curr_vectors,prev2curr_matrix,o...
  function filter_vectors (line 136) | def filter_vectors(data_info, origin,roi_size,thr,num_interp=20):

FILE: tools/tracking/cmap_utils/utils.py
  function import_plugin (line 8) | def import_plugin(cfg):
  function draw_polylines (line 33) | def draw_polylines(vecs, roi_size, origin, cfg):
  function draw_polygons (line 47) | def draw_polygons(vecs, roi_size, origin, cfg):
  function draw_instance_masks (line 65) | def draw_instance_masks(vectors, roi_size, origin, cfg):
  function interp_fixed_num (line 78) | def interp_fixed_num(vector, num_pts):
  function chamfer_distance_batch (line 87) | def chamfer_distance_batch(pred_lines, gt_lines):
  function average_precision (line 110) | def average_precision(recalls, precisions, mode='area'):

FILE: tools/tracking/prepare_gt_tracks.py
  function parse_args (line 27) | def parse_args():
  function import_plugin (line 48) | def import_plugin(cfg):
  function draw_polylines (line 76) | def draw_polylines(vecs, roi_size, origin, cfg):
  function draw_polygons (line 91) | def draw_polygons(vecs, roi_size, origin, cfg):
  function draw_instance_masks (line 109) | def draw_instance_masks(vectors, roi_size, origin, cfg):
  function _mask_iou (line 119) | def _mask_iou(mask1, mask2):
  function find_matchings (line 127) | def find_matchings(src_masks, tgt_masks, thresh=0.1):
  function match_two_consecutive_frames (line 157) | def match_two_consecutive_frames(prev_data, curr_data, roi_size, origin,...
  function assign_global_ids (line 238) | def assign_global_ids(matchings_seq, vectors_seq):
  function _denorm (line 274) | def _denorm(vectors, roi_size, origin):
  function form_gt_track_single (line 281) | def form_gt_track_single(scene_name, scene_name2idx, dataset, out_dir, c...
  function main (line 341) | def main():

FILE: tools/tracking/prepare_pred_tracks.py
  function parse_args (line 32) | def parse_args():
  function match_two_consecutive_frames_pred (line 59) | def match_two_consecutive_frames_pred(args,prev_data,prev_meta,  curr_da...
  function collect_pred (line 76) | def collect_pred(data,thr):
  function get_scene_matching_result (line 86) | def get_scene_matching_result(args,cfg,pred_results,dataset,origin,roi_s...
  function generate_results (line 176) | def generate_results(ids_info,vectors_seq,scores_seq,meta_list,scene_name):
  function get_matching_single (line 204) | def get_matching_single(scene_name,args,scene_name2idx,dataset,cfg,pred_...
  function main (line 213) | def main():

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

FILE: tools/visualization/vis_global.py
  function parse_args (line 28) | def parse_args():
  function combine_images_with_labels (line 94) | def combine_images_with_labels(image_paths, labels, output_path, font_sc...
  function merge_corssing (line 126) | def merge_corssing(polylines):
  function find_largest_convex_hull (line 131) | def find_largest_convex_hull(polylines):
  function project_point_onto_line (line 152) | def project_point_onto_line(point, line):
  function find_nearest_projection_on_polyline (line 166) | def find_nearest_projection_on_polyline(point, polyline):
  function find_and_sort_intersections (line 180) | def find_and_sort_intersections(segmenet1, segment2):
  function get_intersection_point_on_line (line 213) | def get_intersection_point_on_line(line, intersection):
  function merge_l2_points_to_l1 (line 231) | def merge_l2_points_to_l1(line1, line2, line2_intersect_start, line2_int...
  function segment_line_based_on_merged_area (line 261) | def segment_line_based_on_merged_area(line, merged_points):
  function get_bbox_size_for_points (line 296) | def get_bbox_size_for_points(points):
  function get_longer_segmenent_to_merged_points (line 312) | def get_longer_segmenent_to_merged_points(l1_segment, l2_segment, merged...
  function get_line_lineList_max_intersection (line 385) | def get_line_lineList_max_intersection(merged_lines, line, thickness=4):
  function algin_l2_with_l1 (line 401) | def algin_l2_with_l1(line1, line2):
  function _is_u_shape (line 455) | def _is_u_shape(line, direction):
  function check_circle (line 473) | def check_circle(pre_line, vec):
  function connect_polygon (line 488) | def connect_polygon(merged_polyline, merged_lines):
  function iou_merge_boundry (line 506) | def iou_merge_boundry(merged_lines, vec, thickness=1):
  function iou_merge_divider (line 588) | def iou_merge_divider(merged_lines, vec, thickness=1):
  function merge_divider (line 628) | def merge_divider(vecs=None, thickness=1):
  function merge_boundary (line 657) | def merge_boundary(vecs=None, thickness=1, iou_threshold=0.95):
  function get_consecutive_vectors_with_opt (line 688) | def get_consecutive_vectors_with_opt(prev_vectors=None,prev2curr_matrix=...
  function get_prev2curr_vectors (line 720) | def get_prev2curr_vectors(vecs=None, prev2curr_matrix=None,origin=None,r...
  function plot_fig_merged_per_frame (line 744) | def plot_fig_merged_per_frame(num_frames, car_trajectory, x_min, x_max, ...
  function plot_fig_merged (line 885) | def plot_fig_merged(car_trajectory, x_min, x_max, y_min, y_max, pred_sav...
  function plot_fig_unmerged_per_frame (line 975) | def plot_fig_unmerged_per_frame(num_frames, car_trajectory, x_min, x_max...
  function plot_fig_unmerged (line 1046) | def plot_fig_unmerged(car_trajectory, x_min, x_max, y_min, y_max, pred_s...
  function save_t (line 1111) | def save_t(t_max, main_save_folder):
  function save_as_video (line 1132) | def save_as_video(image_list, mp4_output_path, scale=None):
  function vis_pred_data (line 1150) | def vis_pred_data(scene_name="", pred_results=None, origin=None, roi_siz...
  function vis_gt_data (line 1247) | def vis_gt_data(scene_name, args, dataset, gt_data, origin, roi_size):
  function main (line 1355) | def main():

FILE: tools/visualization/vis_per_frame.py
  function parse_args (line 23) | def parse_args():
  function save_as_video (line 70) | def save_as_video(image_list, mp4_output_path, scale=None):
  function plot_one_frame_results (line 85) | def plot_one_frame_results(vectors, id_info, roi_size, scene_dir, args):
  function vis_pred_data (line 142) | def vis_pred_data(scene_name, args, pred_results, origin,roi_size):
  function vis_gt_data (line 206) | def vis_gt_data(scene_name, args, dataset, scene_name2idx, gt_data, orig...
  function main (line 249) | def main():