SYMBOL INDEX (230 symbols across 20 files)

FILE: data/Load_Data.py
  class LaneDataset (line 27) | class LaneDataset(Dataset):
    method __init__ (line 39) | def __init__(self, dataset_base_dir, json_file_path, args, data_aug=Fa...
    method preprocess_data_from_json_once (line 125) | def preprocess_data_from_json_once(self, idx_json_file):
    method preprocess_data_from_json_openlane (line 226) | def preprocess_data_from_json_openlane(self, idx_json_file):
    method __len__ (line 322) | def __len__(self):
    method WIP__getitem__ (line 329) | def WIP__getitem__(self, idx):
    method __getitem__ (line 473) | def __getitem__(self, idx):
    method transform_mats_impl (line 479) | def transform_mats_impl(self, cam_extrinsics, cam_intrinsics, cam_pitc...
  function make_lane_y_mono_inc (line 491) | def make_lane_y_mono_inc(lane):
  function data_aug_rotate (line 509) | def data_aug_rotate(img):
  function seed_worker (line 521) | def seed_worker(worker_id):
  function get_loader (line 527) | def get_loader(transformed_dataset, args):
  function map_once_json2img (line 587) | def map_once_json2img(json_label_file):

FILE: data/apollo_dataset.py
  class ApolloLaneDataset (line 45) | class ApolloLaneDataset(Dataset):
    method __init__ (line 46) | def __init__(self, dataset_base_dir, json_file_path, args, data_aug=Fa...
    method gen_single_file_json (line 110) | def gen_single_file_json(self):
    method parse_processed_info_dict_apollo (line 144) | def parse_processed_info_dict_apollo(self, idx):
    method __len__ (line 153) | def __len__(self):
    method WIP__getitem__ (line 160) | def WIP__getitem__(self, idx):
    method __getitem__ (line 306) | def __getitem__(self, idx):
    method init_dataset_3D (line 312) | def init_dataset_3D(self, dataset_base_dir, json_file_path):
    method transform_mats_impl (line 430) | def transform_mats_impl(self, cam_pitch, cam_height):
  function data_aug_rotate (line 438) | def data_aug_rotate(img):
  function seed_worker (line 450) | def seed_worker(worker_id):
  function get_loader (line 456) | def get_loader(transformed_dataset, args):

FILE: data/transform.py
  function get_random_state (line 9) | def get_random_state() -> np.random.RandomState:
  function normal (line 13) | def normal(
  class PhotoMetricDistortionMultiViewImage (line 24) | class PhotoMetricDistortionMultiViewImage:
    method __init__ (line 43) | def __init__(self,
    method __call__ (line 53) | def __call__(self, results):

FILE: experiments/ddp.py
  function setup_dist_launch (line 25) | def setup_dist_launch(args):
  function setup_slurm (line 36) | def setup_slurm(args):
  function setup_distributed (line 60) | def setup_distributed(args):
  function ddp_init (line 68) | def ddp_init(args):
  function to_python_float (line 89) | def to_python_float(t):
  function reduce_tensor (line 95) | def reduce_tensor(tensor, world_size):
  function reduce_tensors (line 102) | def reduce_tensors(*tensors, world_size):

FILE: experiments/gpu_utils.py
  function get_rank (line 4) | def get_rank() -> int:
  function is_main_process (line 12) | def is_main_process() -> bool:
  function gpu_available (line 16) | def gpu_available() -> bool:

FILE: experiments/runner.py
  class Runner (line 30) | class Runner:
    method __init__ (line 31) | def __init__(self, args):
    method train (line 76) | def train(self):
    method _log_model_info (line 237) | def _log_model_info(self, model):
    method _log_training_loss (line 246) | def _log_training_loss(self, output, epoch, step, data_loader):
    method save_checkpoint (line 259) | def save_checkpoint(self, state, to_copy, epoch, save_path):
    method validate (line 272) | def validate(self, model, **kwargs):
    method _recal_gpus_val (line 419) | def _recal_gpus_val(self, gather_output, eval_stats):
    method _get_model_from_cfg (line 481) | def _get_model_from_cfg(self):
    method _load_ckpt_from_workdir (line 496) | def _load_ckpt_from_workdir(self, model):
    method eval (line 514) | def eval(self):
    method _get_train_dataset (line 530) | def _get_train_dataset(self):
    method _get_model_ddp (line 545) | def _get_model_ddp(self):
    method resume_model (line 587) | def resume_model(self, model, path=''):
    method _get_valid_dataset (line 621) | def _get_valid_dataset(self):
    method save_eval_result_once (line 638) | def save_eval_result_once(self, args, img_path, lanelines_pred, laneli...
    method log_eval_stats (line 685) | def log_eval_stats(self, eval_stats):
    method _log_genlane_eval_info (line 699) | def _log_genlane_eval_info(self, eval_stats):
  function set_work_dir (line 714) | def set_work_dir(cfg):

FILE: main.py
  function get_args (line 9) | def get_args():
  function main (line 29) | def main():

FILE: models/latr.py
  class LATR (line 15) | class LATR(nn.Module):
    method __init__ (line 16) | def __init__(self, args):
    method forward (line 60) | def forward(self, image, _M_inv=None, is_training=True, extra_dict=None):

FILE: models/latr_head.py
  class LATRHead (line 22) | class LATRHead(nn.Module):
    method __init__ (line 23) | def __init__(self, args,
    method _init_weights (line 177) | def _init_weights(self):
    method forward (line 186) | def forward(self, input_dict, is_training=True):
    method get_project_loss (line 308) | def get_project_loss(self, results, input_dict, h=20, w=30):
    method get_loss (line 347) | def get_loss(self, output_dict, input_dict):
    method get_reference_points (line 439) | def get_reference_points(H, W, bs=1, device='cuda', dtype=torch.float):
  function build_nn_loss (line 452) | def build_nn_loss(loss_cfg):

FILE: models/ms2one.py
  function build_ms2one (line 9) | def build_ms2one(config):
  class Naive (line 18) | class Naive(nn.Module):
    method __init__ (line 19) | def __init__(self, inc, outc, kernel_size=1):
    method forward (line 23) | def forward(self, ms_feats):
  class DilateNaive (line 30) | class DilateNaive(nn.Module):
    method __init__ (line 31) | def __init__(self, inc, outc, num_scales=4,
    method forward (line 74) | def forward(self, x):

FILE: models/scatter_utils.py
  function broadcast (line 8) | def broadcast(src: torch.Tensor, other: torch.Tensor, dim: int):
  function scatter_sum (line 20) | def scatter_sum(src: torch.Tensor, index: torch.Tensor, dim: int = -1,
  function scatter_add (line 38) | def scatter_add(src: torch.Tensor, index: torch.Tensor, dim: int = -1,
  function scatter_mul (line 44) | def scatter_mul(src: torch.Tensor, index: torch.Tensor, dim: int = -1,
  function scatter_mean (line 50) | def scatter_mean(src: torch.Tensor, index: torch.Tensor, dim: int = -1,
  function scatter_min (line 73) | def scatter_min(
  function scatter_max (line 80) | def scatter_max(
  function scatter (line 87) | def scatter(src: torch.Tensor, index: torch.Tensor, dim: int = -1,

FILE: models/sparse_ins.py
  function _make_stack_3x3_convs (line 10) | def _make_stack_3x3_convs(num_convs, in_channels, out_channels):
  class MaskBranch (line 20) | class MaskBranch(nn.Module):
    method __init__ (line 21) | def __init__(self, cfg, in_channels):
    method _init_weights (line 30) | def _init_weights(self):
    method forward (line 36) | def forward(self, features):
  class InstanceBranch (line 42) | class InstanceBranch(nn.Module):
    method __init__ (line 43) | def __init__(self, cfg, in_channels, **kwargs):
    method _init_weights (line 75) | def _init_weights(self):
    method forward (line 89) | def forward(self, seg_features, is_training=True):
  class SparseInsDecoder (line 154) | class SparseInsDecoder(nn.Module):
    method __init__ (line 155) | def __init__(self, cfg, **kargs) -> None:
    method _init_weights (line 170) | def _init_weights(self):
    method compute_coordinates (line 175) | def compute_coordinates(self, x):
    method forward (line 185) | def forward(self, features, is_training=True, **kwargs):
    method loss (line 219) | def loss(self, output, lane_idx_map, input_shape):
    method prepare_targets (line 261) | def prepare_targets(self, targets):

FILE: models/sparse_inst_loss.py
  function _max_by_axis (line 19) | def _max_by_axis(the_list):
  class NestedTensor (line 28) | class NestedTensor(object):
    method __init__ (line 29) | def __init__(self, tensors, mask: Optional[Tensor]):
    method to (line 33) | def to(self, device):
    method decompose (line 43) | def decompose(self):
    method __repr__ (line 46) | def __repr__(self):
  function _onnx_nested_tensor_from_tensor_list (line 54) | def _onnx_nested_tensor_from_tensor_list(tensor_list: List[Tensor]) -> N...
  function nested_tensor_from_tensor_list (line 83) | def nested_tensor_from_tensor_list(tensor_list: List[Tensor]):
  function nested_masks_from_list (line 108) | def nested_masks_from_list(tensor_list: List[Tensor], input_shape=None):
  function is_dist_avail_and_initialized (line 133) | def is_dist_avail_and_initialized():
  function get_world_size (line 141) | def get_world_size():
  function aligned_bilinear (line 147) | def aligned_bilinear(tensor, factor):
  function compute_mask_iou (line 174) | def compute_mask_iou(inputs, targets):
  function dice_score (line 185) | def dice_score(inputs, targets):
  function dice_loss (line 194) | def dice_loss(inputs, targets, reduction='sum'):
  class SparseInstCriterion (line 206) | class SparseInstCriterion(nn.Module):
    method __init__ (line 209) | def __init__(self, num_classes=4, cfg=None, matcher=None):
    method get_weight_dict (line 216) | def get_weight_dict(self, cfg):
    method _get_src_permutation_idx (line 228) | def _get_src_permutation_idx(self, indices):
    method _get_tgt_permutation_idx (line 235) | def _get_tgt_permutation_idx(self, indices):
    method loss_labels (line 242) | def loss_labels(self, outputs, targets, indices, num_instances, input_...
    method loss_masks_with_iou_objectness (line 271) | def loss_masks_with_iou_objectness(self, outputs, targets, indices, nu...
    method get_loss (line 322) | def get_loss(self, loss, outputs, targets, indices, num_instances, **k...
    method forward (line 333) | def forward(self, outputs, targets, input_shape):
  class SparseInstMatcherV1 (line 364) | class SparseInstMatcherV1(nn.Module):
    method __init__ (line 366) | def __init__(self, cfg=None):
    method forward (line 373) | def forward(self, outputs, targets, input_shape):
  class SparseInstMatcher (line 418) | class SparseInstMatcher(nn.Module):
    method __init__ (line 420) | def __init__(self, cfg=None):
    method forward (line 426) | def forward(self, outputs, targets, input_shape):

FILE: models/transformer_bricks.py
  function pos2posemb3d (line 28) | def pos2posemb3d(pos, num_pos_feats=128, temperature=10000):
  function generate_ref_pt (line 43) | def generate_ref_pt(minx, miny, maxx, maxy, z, nx, ny, device='cuda'):
  function ground2img (line 65) | def ground2img(coords3d, H, W, lidar2img, ori_shape, mask=None, return_i...
  class MSDeformableAttention3D (line 105) | class MSDeformableAttention3D(BaseModule):
    method __init__ (line 106) | def __init__(self,
    method init_weights (line 165) | def init_weights(self):
    method ref_to_lidar (line 185) | def ref_to_lidar(self, reference_points, pc_range, not_y=True):
    method point_sampling (line 196) | def point_sampling(self, reference_points, lidar2img, ori_shape):
    method forward (line 203) | def forward(self,
  class LATRDecoderLayer (line 311) | class LATRDecoderLayer(BaseTransformerLayer):
    method __init__ (line 312) | def __init__(self,
    method forward (line 331) | def forward(self,
  class LATRTransformerDecoder (line 350) | class LATRTransformerDecoder(TransformerLayerSequence):
    method __init__ (line 351) | def __init__(self,
    method init_weights (line 397) | def init_weights(self):
    method pred2M (line 402) | def pred2M(self, pitch_z):
    method forward (line 417) | def forward(self, query, key, value,
  class LATRTransformer (line 525) | class LATRTransformer(BaseModule):
    method __init__ (line 526) | def __init__(self, encoder=None, decoder=None, init_cfg=None):
    method init_weights (line 536) | def init_weights(self):
    method get_reference_points (line 544) | def get_reference_points(spatial_shapes, valid_ratios, device):
    method get_valid_ratio (line 574) | def get_valid_ratio(self, mask):
    method with_encoder (line 585) | def with_encoder(self):
    method forward (line 588) | def forward(self, x, mask, query,

FILE: models/utils.py
  function inverse_sigmoid (line 6) | def inverse_sigmoid(x, eps=1e-5):
  class deepFeatureExtractor_EfficientNet (line 25) | class deepFeatureExtractor_EfficientNet(nn.Module):
    method __init__ (line 26) | def __init__(self, architecture="EfficientNet-B5", lv6=False, lv5=Fals...
    method forward (line 100) | def forward(self, x):
    method freeze_bn (line 129) | def freeze_bn(self, enable=False):

FILE: utils/MinCostFlow.py
  function SolveMinCostFlow (line 32) | def SolveMinCostFlow(adj_mat, cost_mat):
  function main (line 99) | def main():

FILE: utils/eval_3D_lane.py
  class LaneEval (line 43) | class LaneEval(object):
    method __init__ (line 44) | def __init__(self, args, logger):
    method bench (line 61) | def bench(self, pred_lanes, pred_category, gt_lanes, gt_visibility, gt...
    method bench_one_submit (line 259) | def bench_one_submit(self, pred_dir, gt_dir, test_txt, prob_th=0.5, vi...
    method bench_one_submit_ddp (line 423) | def bench_one_submit_ddp(self, pred_lines_sub, gt_lines_sub, model_nam...

FILE: utils/eval_3D_lane_apollo.py
  class LaneEval (line 48) | class LaneEval(object):
    method __init__ (line 49) | def __init__(self, args, logger=None):
    method log_eval_info (line 68) | def log_eval_info(self):
    method bench (line 75) | def bench(self, pred_lanes, gt_lanes, gt_visibility, raw_file, gt_cam_...
    method bench_one_submit (line 280) | def bench_one_submit(self, pred_file, gt_file, prob_th=0.5, vis=False):
    method bench_one_submit_ddp (line 479) | def bench_one_submit_ddp(self, pred_lines_sub, gt_lines_sub, model_nam...
    method bench_PR (line 595) | def bench_PR(self, pred_lanes, gt_lanes, gt_visibility):
    method bench_one_submit_varying_probs (line 701) | def bench_one_submit_varying_probs(self, pred_file, gt_file, eval_out_...

FILE: utils/eval_3D_once.py
  class Bev_Projector (line 19) | class Bev_Projector:
    method __init__ (line 20) | def __init__(self, side_range, fwd_range, height_range, res, lane_widt...
    method proj_oneline_zx (line 32) | def proj_oneline_zx(self, one_lane):
  class LaneEval (line 56) | class LaneEval:
    method file_parser (line 58) | def file_parser(gt_root_path, pred_root_path):
    method summarize (line 79) | def summarize(res):
    method lane_evaluation (line 105) | def lane_evaluation(self, gt_root_path, pred_root_path, config_path, a...
  function evaluate_list (line 188) | def evaluate_list(gt_path_list, pred_path_list, config):
  class LaneEvalOneFile (line 222) | class LaneEvalOneFile:
    method __init__ (line 223) | def __init__(self, gt_path, pred_path, bev_projector, iou_thresh, dist...
    method preprocess (line 233) | def preprocess(self, store_spec):
    method calc_iou (line 246) | def calc_iou(self, lane1, lane2):
    method cal_mean_dist (line 262) | def cal_mean_dist(self, src_line, dst_line):
    method sort_lanes_z (line 278) | def sort_lanes_z(self, lanes):
    method eval (line 286) | def eval(self):
    method cal_tp (line 303) | def cal_tp(self, gt_num, pred_num, gt_lanes, pred_lanes):
  function parse_config (line 329) | def parse_config():

FILE: utils/utils.py
  function create_logger (line 41) | def create_logger(args):
  function define_args (line 66) | def define_args():
  function prune_3d_lane_by_visibility (line 107) | def prune_3d_lane_by_visibility(lane_3d, visibility):
  function prune_3d_lane_by_range (line 112) | def prune_3d_lane_by_range(lane_3d, x_min, x_max):
  function resample_laneline_in_y (line 125) | def resample_laneline_in_y(input_lane, y_steps, out_vis=False):
  function resample_laneline_in_y_with_vis (line 156) | def resample_laneline_in_y_with_vis(input_lane, y_steps, vis_vec):
  function homograpthy_g2im (line 186) | def homograpthy_g2im(cam_pitch, cam_height, K):
  function projection_g2im (line 195) | def projection_g2im(cam_pitch, cam_height, K):
  function homograpthy_g2im_extrinsic (line 203) | def homograpthy_g2im_extrinsic(E, K):
  function projection_g2im_extrinsic (line 211) | def projection_g2im_extrinsic(E, K):
  function homography_crop_resize (line 217) | def homography_crop_resize(org_img_size, crop_y, resize_img_size):
  function homographic_transformation (line 234) | def homographic_transformation(Matrix, x, y):
  function projective_transformation (line 252) | def projective_transformation(Matrix, x, y, z):
  function first_run (line 271) | def first_run(save_path):
  function mkdir_if_missing (line 284) | def mkdir_if_missing(directory):
  function str2bool (line 294) | def str2bool(argument):
  class Logger (line 303) | class Logger(object):
    method __init__ (line 307) | def __init__(self, fpath=None):
    method __del__ (line 315) | def __del__(self):
    method __enter__ (line 318) | def __enter__(self):
    method __exit__ (line 321) | def __exit__(self, *args):
    method write (line 324) | def write(self, msg):
    method flush (line 329) | def flush(self):
    method close (line 335) | def close(self):
  class AverageMeter (line 341) | class AverageMeter(object):
    method __init__ (line 343) | def __init__(self):
    method reset (line 346) | def reset(self):
    method update (line 352) | def update(self, val, n=1):
  function define_optim (line 359) | def define_optim(optim, params, lr, weight_decay):
  function cosine_schedule_with_warmup (line 373) | def cosine_schedule_with_warmup(k, args, dataset_size=None):
  function define_scheduler (line 392) | def define_scheduler(optimizer, args, dataset_size=None):
  function define_init_weights (line 436) | def define_init_weights(model, init_w='normal', activation='relu'):
  function weights_init_normal (line 450) | def weights_init_normal(m):
  function weights_init_xavier (line 474) | def weights_init_xavier(m):
  function weights_init_kaiming (line 490) | def weights_init_kaiming(m):
  function weights_init_orthogonal (line 506) | def weights_init_orthogonal(m):