MCPcopy Create free account

hub / github.com/cxmomo/RaCFormer / functions

Functions355 in github.com/cxmomo/RaCFormer

Method__init__
(self, in_dim, in_points, n_groups=1, query_dim=None, out_dim=None, out_points=None)
models/racformer_transformer.py:547
Method__init__
(self, embed_dims=256, hidden_dims=64, num_frames=8,
models/racformer_transformer.py:616
Method__init__
(self, input_channels, hidden_channels, kernel_size)
models/racformer_transformer.py:692
Method__init__
(self, pc_range, voxel_size=None, post_center_range=None,
models/bbox/coders/nms_free_coder.py:20
Method__init__
(self, weight=1.0)
models/bbox/match_costs/match_cost.py:12
Method__init__
(self, weight=1.0)
models/bbox/match_costs/match_cost.py:36
Method__init__
(self, weight, pc_range)
models/bbox/match_costs/match_cost.py:68
Method__init__
(self, weight)
models/bbox/match_costs/match_cost.py:84
Method__init__
(self, cls_cost=dict(type='ClassificationCost', weight=1.0), reg_cost=dict(t
models/bbox/assigners/polar_hungarian_assigner_3d.py:17
Method__init__
(self, cls_cost=dict(type='ClassificationCost', weight=1.), reg_cost=dict(ty
models/bbox/assigners/hungarian_assigner_3d.py:17
Method__init__
(self, in_channels, out_channels, scale_factor=4,
models/necks/lss_fpn.py:13
Method__init__
( self, grid_config, input_size, downsample=16, in_channels=512,
models/necks/view_transformer_racformer.py:37
Method__init__
(self, inplanes, planes, kernel_size, padding, dilation, norm_cfg=dict(type='BN'))
models/necks/view_transformer_racformer.py:331
Method__init__
(self, inplanes, mid_channels=256, norm_cfg=dict(type='BN'))
models/necks/view_transformer_racformer.py:365
Method__init__
(self, channels, act_layer=nn.ReLU, gate_layer=nn.Sigmoid)
models/necks/view_transformer_racformer.py:467
Method__init__
(self, in_channels, mid_channels, context_channels,
models/necks/view_transformer_racformer.py:483
Method__init__
(self, loss_depth_weight=3.0, loss_reg_depth_weight=1.0, depthnet_cfg=dict(), **kwargs)
models/necks/view_transformer_racformer.py:574
Method__init__
(self, in_channels, out_channels, num_outs,
models/necks/fpn.py:63
Method__init__
(self, alpha: float, gamma: float = 2.0, reduction: str = 'none')
models/necks/focalloss.py:275
Method__init__
(self, inplace=True)
models/backbones/vovnet.py:158
Method__init__
(self, channel, reduction=4)
models/backbones/vovnet.py:167
Method__init__
(self, in_ch, stage_ch, concat_ch, layer_per_block, module_name, SE=False, identity=False, depthwise=False, wi
models/backbones/vovnet.py:182
Method__init__
(self, in_ch, stage_ch, concat_ch, block_per_stage, layer_per_block, stage_num, SE=False, depthwise=False, wit
models/backbones/vovnet.py:242
Method__init__
( self, numC_input, num_layer=[2, 2, 2], num_channels=None,
models/backbones/resnet.py:19
Method__init__
(self, softmax_scale=None, attention_dropout=0.0, device=None, dtype=None)
models/model_utils/attention.py:44
Method__init__
(self, save_path=None)
models/hook/is_save.py:26
Method__init__
(self, init_updates=0, decay=0.9990, resume=None, interval=1)
models/hook/ema.py:70
Method__init__
(self, interval: int = -1, by_epoch: bool = True, save_opti
models/hook/check_checkpoint.py:55
Method__init__
(self, start_epoch=1)
models/hook/sequentialsontrol.py:12
Method__repr__
(self)
loaders/pipelines/transforms.py:61
Method__repr__
(self)
loaders/pipelines/transforms.py:111
Method__repr__
(self)
loaders/pipelines/transforms.py:208
Method__repr__
(self)
loaders/pipelines/transforms.py:655
Method__repr__
str: Return a string that describes the module.
loaders/pipelines/loading.py:260
Method__repr__
str: Return a string that describes the module.
loaders/pipelines/loading.py:433
Method__repr__
str: Return a string that describes the module.
loaders/pipelines/loading.py:798
Method__repr__
str: Return a string that describes the module.
loaders/pipelines/formatng.py:128
Method_get_target_single
(self, cls_score, bbox_pred,
models/racformer_head.py:326
Method_init_layers
(self)
models/racformer_head.py:51
Method_initialize_weights
(self)
models/backbones/vovnet.py:344
Method_is_power_of_2
(n)
models/bev_self_attention.py:76
Method_parse_ann_info
Parse bbox and mask annotation. Args: ann_info (list[dict]): Annotation info of an image. with_mask (bool): Whether t
loaders/vod_mono_dataset.py:62
Methodafter_run
(self, runner)
utils.py:190
Methodafter_train_epoch
(self, runner)
utils.py:122
Methodafter_train_epoch
(self, runner)
models/hook/is_save.py:64
Methodafter_train_epoch
(self, runner)
models/hook/ema.py:103
Methodafter_train_epoch
(self, runner)
models/hook/check_checkpoint.py:109
Methodafter_train_iter
(self, runner)
models/hook/ema.py:100
Methodafter_train_iter
(self, runner)
models/hook/check_checkpoint.py:179
Methodassign
(self, bbox_pred, cls_pred, gt_bboxes, gt_labels,
models/bbox/assigners/polar_hungarian_assigner_3d.py:29
Functionbackup_code
(work_dir, verbose=False)
utils.py:28
Methodbackward
GPU version of backward function. Args: grad_output (Tensor): Gradient of output tensor of forward.
models/multi_scale_deformable_attn_function.py:57
Methodbackward
GPU version of backward function. Args: grad_output (Tensor): Gradient of output tensor of forward.
models/multi_scale_deformable_attn_function.py:133
Methodbackward
(ctx, *args)
models/checkpoint.py:113
Methodbackward
(ctx, grad_output)
models/csrc/wrapper.py:89
Methodbackward
(ctx, grad_output)
models/csrc/wrapper.py:111
Methodbackward
(ctx, grad_output)
models/csrc/wrapper.py:133
Methodbefore_run
(self, runner)
utils.py:143
Methodbefore_run
(self, runner)
models/hook/ema.py:77
Methodbefore_run
(self, runner)
models/hook/check_checkpoint.py:75
Methodbefore_run
(self, runner)
models/hook/sequentialsontrol.py:23
Methodbefore_run
(self, runner)
models/hook/sequentialsontrol.py:44
Methodbefore_train_epoch
(self, runner)
models/hook/sequentialsontrol.py:26
Methodbefore_train_epoch
(self, runner)
models/hook/sequentialsontrol.py:47
Functionbev_pool_v2_backward
Function: pillar pooling (backward, cuda) Args: out_grad : grad of output bev feature, FloatTensor[b, c, h_out, w_out] depth_grad
models/csrc/bev_pool_v2/src/bev_pool.cpp:74
Functionbev_pool_v2_forward
Function: pillar pooling (forward, cuda) Args: depth : input depth, FloatTensor[n, d, h, w] feat : input features, Fl
models/csrc/bev_pool_v2/src/bev_pool.cpp:30
Functioncheckpoint_sequential
r"""A helper function for checkpointing sequential models. Sequential models execute a list of modules/functions in order (sequentially). The
models/checkpoint.py:261
Methoddepth_to_color
(self, depth)
loaders/pipelines/loading.py:444
Functiondetailed_results_table_tex
Renders a detailed results table in tex. :param metrics_path: path to a serialized DetectionMetrics file. :param output_path: path to the
tools/render.py:380
Functiondist_pr_curve
Plot the PR curves for different distance thresholds. :param md_list: DetectionMetricDataList instance. :param metrics: DetectionMetrics
tools/render.py:304
Methoddn_loss_single
(self, cls_scores, bbox_preds, known_bbox
models/racformer_head.py:264
Methodencode
(self)
models/bbox/coders/nms_free_coder.py:34
Functionforward
(input)
models/checkpoint.py:318
Methodforward
Calls either forward_train or forward_test depending on whether return_loss=True. Note this setting will change the expected inputs. W
models/racformer.py:383
Methodforward
GPU version of multi-scale deformable attention. Args: value (Tensor): The value has shape (bs, num_keys, mum
models/multi_scale_deformable_attn_function.py:19
Methodforward
GPU version of multi-scale deformable attention. Args: value (Tensor): The value has shape (bs, num_keys, mum
models/multi_scale_deformable_attn_function.py:94
Methodforward
(self, x)
models/utils.py:14
Methodforward
(self, mlvl_feats, lss_bev_feats, radar_bev_feats, img_metas)
models/racformer_head.py:82
Methodforward
(ctx, run_function, preserve_rng_state, *args)
models/checkpoint.py:76
Methodforward
Forward Function of MultiScaleDeformAttention. Args: query (Tensor): Query of Transformer with shape (num_que
models/bev_self_attention.py:115
Methodforward
(self, query_bbox, query_feat, mlvl_feats, lss_bev_feats, radar_bev_feats, attn_mask, img_metas)
models/racformer_transformer.py:50
Methodforward
(self, query_bbox, query_feat, mlvl_feats, lss_bev_feats, radar_bev_feats, attn_mask, img_metas)
models/racformer_transformer.py:91
Methodforward
query_bbox: [B, Q, 10] [cx, cy, cz, w, h, d, rot.sin, rot.cos, vx, vy]
models/racformer_transformer.py:235
Methodforward
(self, query_bbox, query_feat, pre_attn_mask)
models/racformer_transformer.py:313
Methodforward
(self, query_ray, query_feat, mlvl_feats, img_metas, d_region=0.1)
models/racformer_transformer.py:419
Methodforward
(self, query_ray, query_feat, bev_feats, img_metas, d_region=0.1)
models/racformer_transformer.py:538
Methodforward
(self, x, query)
models/racformer_transformer.py:608
Methodforward
(self, bev_feats)
models/racformer_transformer.py:655
Methodforward
(self, x)
models/racformer_transformer.py:667
Methodforward
(self, x, h_prev)
models/racformer_transformer.py:705
Methodforward
(self, feats)
models/necks/lss_fpn.py:88
Methodforward
Transform image-view feature into bird-eye-view feature. Args: input (list(torch.tensor)): of (image-view feature, rots, trans,
models/necks/view_transformer_racformer.py:306
Methodforward
(self, x)
models/necks/view_transformer_racformer.py:348
Methodforward
(self, x)
models/necks/view_transformer_racformer.py:414
Methodforward
(self, x)
models/necks/view_transformer_racformer.py:456
Methodforward
(self, x, x_se)
models/necks/view_transformer_racformer.py:474
Methodforward
(self, x, radar_feats, rcs_embedding, mlp_input)
models/necks/view_transformer_racformer.py:546
Methodforward
(self, x, radar_depth, radar_rcs, img_metas, mlp_input)
models/necks/view_transformer_racformer.py:680
Methodforward
Forward function.
models/necks/fpn.py:155
Methodforward
(self, input: torch.Tensor, target: torch.Tensor)
models/necks/focalloss.py:176
← previousnext →201–300 of 355, ranked by callers