Functions — github.com/JMoonr/LATR

hub / github.com/JMoonr/LATR / functions

Functions207 in github.com/JMoonr/LATR

⨍ Functions 207◇ Types & classes 27

↓ 1 callersMethodpoint_sampling

(self, reference_points, lidar2img, ori_shape)

models/transformer_bricks.py:196

↓ 1 callersMethodpred2M

(self, pitch_z)

models/transformer_bricks.py:402

↓ 1 callersMethodprepare_targets

(self, targets)

models/sparse_ins.py:261

↓ 1 callersMethodpreprocess

(self, store_spec)

utils/eval_3D_once.py:233

↓ 1 callersMethodpreprocess_data_from_json_once

(self, idx_json_file)

data/Load_Data.py:125

↓ 1 callersMethodpreprocess_data_from_json_openlane

(self, idx_json_file)

data/Load_Data.py:226

↓ 1 callersFunctionreduce_tensor

(tensor, world_size)

experiments/ddp.py:95

↓ 1 callersMethodref_to_lidar

(self, reference_points, pc_range, not_y=True)

models/transformer_bricks.py:185

↓ 1 callersMethodreset

(self)

utils/utils.py:346

↓ 1 callersMethodresume_model

(self, model, path='')

experiments/runner.py:587

↓ 1 callersMethodsave_eval_result_once

(self, args, img_path, lanelines_pred, lanelines_prob)

experiments/runner.py:638

↓ 1 callersFunctionscatter_max

( src: torch.Tensor, index: torch.Tensor, dim: int = -1, out: Optional[torch.Tensor] = None,

models/scatter_utils.py:80

↓ 1 callersFunctionscatter_min

( src: torch.Tensor, index: torch.Tensor, dim: int = -1, out: Optional[torch.Tensor] = None,

models/scatter_utils.py:73

↓ 1 callersFunctionscatter_mul

(src: torch.Tensor, index: torch.Tensor, dim: int = -1, out: Optional[torch.Tensor] = None,

models/scatter_utils.py:44

↓ 1 callersFunctionset_work_dir

(cfg)

experiments/runner.py:714

↓ 1 callersFunctionsetup_dist_launch

(args)

experiments/ddp.py:25

↓ 1 callersFunctionsetup_distributed

(args)

experiments/ddp.py:60

↓ 1 callersFunctionsetup_slurm

(args)

experiments/ddp.py:36

↓ 1 callersMethodtransform_mats_impl

(self, cam_pitch, cam_height)

data/apollo_dataset.py:430

↓ 1 callersMethodtransform_mats_impl

(self, cam_extrinsics, cam_intrinsics, cam_pitch, cam_height)

data/Load_Data.py:479

Method__call__

Call function to perform photometric distortion on images. Args: results (dict): Result dict from loading pipeline. Return

Args: idx (int): Index in list to load image

data/apollo_dataset.py:306

Method__getitem__

Args: idx (int): Index in list to load image

data/Load_Data.py:473

utils/eval_3D_lane.py:44

Method__init__

(self, side_range, fwd_range, height_range, res, lane_width_x, lane_width_y)

utils/eval_3D_once.py:20

Method__init__

(self, gt_path, pred_path, bev_projector, iou_thresh, distance_thresh, score_l, score_h, score_step)

utils/eval_3D_once.py:223

Method__init__

(self, args, logger=None)

utils/eval_3D_lane_apollo.py:49

Method__init__

(self, dataset_base_dir, json_file_path, args, data_aug=False, **kwargs)

data/apollo_dataset.py:46

Method__init__

(self, brightness_delta=32, contrast_range=(0.5, 1.5), satu

data/transform.py:43

Method__init__

:param dataset_info_file: json file list

data/Load_Data.py:39

Method__init__

(self, args)

experiments/runner.py:31

Method__init__

(self, inc, outc, num_scales=4, dilations=(1, 2, 5, 9), merge=True, fpn=Fals

models/ms2one.py:31

Method__init__

(self, num_classes=4, cfg=None, matcher=None)

models/sparse_inst_loss.py:209

Method__init__

(self, cfg=None)

models/sparse_inst_loss.py:366

Method__init__

(self, cfg=None)

models/sparse_inst_loss.py:420

Method__init__

(self, architecture="EfficientNet-B5", lv6=False, lv5=False, lv4=False, lv3=False)

(self, cfg, in_channels, **kwargs)

models/sparse_ins.py:43

Method__init__

(self, cfg, **kargs)

models/sparse_ins.py:155

Method__init__

(self, args, dim=128, num_group=1, num_convs=4,

models/latr_head.py:23

Method__init__

(self, embed_dims=256, num_heads=8, num_levels=4,

models/transformer_bricks.py:106

Method__init__

(self, attn_cfgs, feedforward_channels, ffn_dropout=0.0,

models/transformer_bricks.py:312

Method__init__

(self, *args, embed_dims=None, post_norm_cfg=dict(type='LN'),

models/transformer_bricks.py:351

Method__len__

Conventional len method

data/apollo_dataset.py:153

Method__len__

Conventional len method

data/Load_Data.py:322

Method__repr__

(self)

models/sparse_inst_loss.py:46

Method_is_power_of_2

(n)

models/transformer_bricks.py:137

Functionaligned_bilinear

(tensor, factor)

models/sparse_inst_loss.py:147

Methodbench_one_submit

(self, pred_dir, gt_dir, test_txt, prob_th=0.5, vis=False)

utils/eval_3D_lane.py:259

Methodbench_one_submit

(self, pred_file, gt_file, prob_th=0.5, vis=False)

utils/eval_3D_lane_apollo.py:280

Methodbench_one_submit_ddp

(self, pred_lines_sub, gt_lines_sub, model_name, prob_th=0.5, vis=False)

utils/eval_3D_lane_apollo.py:479

Methodbench_one_submit_varying_probs

(self, pred_file, gt_file, eval_out_file=None, eval_fig_file=None)

utils/eval_3D_lane_apollo.py:701

Functioncosine_schedule_with_warmup

(k, args, dataset_size=None)

Functiondefine_init_weights

(model, init_w='normal', activation='relu')

utils/utils.py:436

Functiondefine_optim

(optim, params, lr, weight_decay)

utils/utils.py:359

Functiondice_score

(inputs, targets)

models/sparse_inst_loss.py:185

Functionevaluate_list

(gt_path_list, pred_path_list, config)

utils/eval_3D_once.py:188

(self, outputs, targets, input_shape)

models/sparse_inst_loss.py:333

Methodforward

(self, outputs, targets, input_shape)

models/sparse_inst_loss.py:373

Methodforward

(self, outputs, targets, input_shape)

models/sparse_inst_loss.py:426

(self, image, _M_inv=None, is_training=True, extra_dict=None)

models/latr.py:60

Methodforward

(self, features)

models/sparse_ins.py:36

Methodforward

(self, seg_features, is_training=True)

models/sparse_ins.py:89

Methodforward

(self, features, is_training=True, **kwargs)

models/sparse_ins.py:185

Methodforward

(self, input_dict, is_training=True)

models/latr_head.py:186

Methodforward

(self, query, key=None, value=None, identity=N

models/transformer_bricks.py:203

Methodforward

(self, query, key=None, value=None, query_pos=

models/transformer_bricks.py:331

Methodforward

(self, query, key, value, top_view_region=None, z_region=None, bev_h=None, bev

models/transformer_bricks.py:417

Methodfreeze_bn

Adapted from https://discuss.pytorch.org/t/how-to-train-with-frozen-batchnorm/12106/8

models/utils.py:129

Functionget_loader

create dataset from ground-truth return a batch sampler based ont the dataset

data/apollo_dataset.py:456

Methodget_reference_points

(H, W, bs=1, device='cuda', dtype=torch.float)

models/latr_head.py:439

Functionhomographic_transformation

Helper function to transform coordinates defined by transformation matrix Args: Matrix (multi dim - array): 3x3 homography m

utils/utils.py:234

Methodinit_weights

(self)

models/transformer_bricks.py:397

(self, outputs, targets, indices, num_instances, input_shape=None)

models/sparse_inst_loss.py:242

Methodloss_masks_with_iou_objectness

(self, outputs, targets, indices, num_instances, input_shape)

models/sparse_inst_loss.py:271

Functionmake_lane_y_mono_inc

Due to lose of height dim, projected lanes to flat ground plane may not have monotonically increasing y. This function trace the y wi

data/Load_Data.py:491

Functionnested_tensor_from_tensor_list

(tensor_list: List[Tensor])

models/sparse_inst_loss.py:83

Functionnormal

( loc=0.0, scale=1.0, size=None, random_state=None, )

data/transform.py:13

Functionparse_config

()

utils/eval_3D_once.py:329

Functionpos2posemb3d

(pos, num_pos_feats=128, temperature=10000)

models/transformer_bricks.py:28

Functionreduce_tensors

(*tensors, world_size)

experiments/ddp.py:102

Functionresample_laneline_in_y_with_vis

Interpolate x, z values at each anchor grid, including those beyond the range of input lnae y range :param input_lane: N x 2 or N x 3 n

utils/utils.py:156

Methodsave_cur_ckpt

( loss, with_eval=True, eval_stats=None)

experiments/runner.py:91

Functionscatter

r""" | .. image:: https://raw.githubusercontent.com/rusty1s/pytorch_scatter/ master/docs/source/_figures/add.svg?sanitize=true

models/scatter_utils.py:87

Functionscatter_add

(src: torch.Tensor, index: torch.Tensor, dim: int = -1, out: Optional[torch.Tensor] = None,

models/scatter_utils.py:38

Functionseed_worker

(worker_id)

data/apollo_dataset.py:450

Functionseed_worker

(worker_id)

data/Load_Data.py:521

Methodsingle_layer_loss

(layer_idx)

models/latr_head.py:358

Functionstr2bool

(argument)

utils/utils.py:294

← previous next →101–200 of 207, ranked by callers