MCPcopy Create free account

hub / github.com/OpenDriveLab/OccNet / functions

Functions203 in github.com/OpenDriveLab/OccNet

⨍ Functions 203◇ Types & classes 44↳ Endpoints 1
Method__init__
(self, num_feature_levels=4, num_cams=6, two_stage_num_p
projects/mmdet3d_plugin/bevformer/modules/transformer.py:38
Method__init__
(self, attn_cfgs=None, ffn_cfgs=dict( type='FFN',
projects/mmdet3d_plugin/bevformer/modules/custom_base_transformer_layer.py:72
Method__init__
(self, *args, pc_range=None, num_points_in_pillar=4, return_intermediate=False, dataset_type='nuscenes',
projects/mmdet3d_plugin/bevformer/modules/encoder.py:40
Method__init__
(self, embed_dims=256, num_heads=8, num_levels=4,
projects/mmdet3d_plugin/bevformer/modules/spatial_cross_attention.py:204
Method__init__
(self, *args, with_box_refine=False, as_two_stage=False,
projects/mmdet3d_plugin/bevformer/dense_heads/bevformer_occ_head.py:45
Method__init__
(self, model, eval_model=None, batch_processor=None,
projects/mmdet3d_plugin/bevformer/runner/epoch_based_runner.py:33
Method__init__
(self, use_grid_mask=False, pts_voxel_layer=None, pts_vo
projects/mmdet3d_plugin/bevformer/detectors/bevformer_occ.py:27
Method__init__
(self, every_n_inters=1)
projects/mmdet3d_plugin/bevformer/hooks/custom_hooks.py:8
Method__init__
(self, pc_range, voxel_size=None, post_center_range=None,
projects/mmdet3d_plugin/core/bbox/coders/nms_free_coder.py:22
Method__init__
(self, weight=1.)
projects/mmdet3d_plugin/core/bbox/match_costs/match_cost.py:12
Method__init__
(self, cls_cost=dict(type='ClassificationCost', weight=1.), reg_cost=dict(ty
projects/mmdet3d_plugin/core/bbox/assigners/hungarian_assigner_3d.py:42
Method__init__
(self, *args, dynamic_intervals=None, **kwargs)
projects/mmdet3d_plugin/core/evaluation/eval_hooks.py:31
Method__init__
(self, data_infos, dataset_type=None)
projects/mmdet3d_plugin/datasets/ego_pose_extractor.py:15
Method__init__
Figure out a list of sample data tokens for training.
projects/mmdet3d_plugin/datasets/nuscenes_ego_pose_loader.py:14
Method__init__
(self, dataset, samples_per_gpu=1, num_replicas=None,
projects/mmdet3d_plugin/datasets/samplers/group_sampler.py:33
Method__init__
(self, dataset=None, num_replicas=None, rank=None,
projects/mmdet3d_plugin/datasets/samplers/distributed_sampler.py:11
Method__init__
(self, size=None, size_divisor=None, pad_val=0)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:23
Method__init__
(self, mean, std, to_rgb=True)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:76
Method__init__
(self, brightness_delta=32, contrast_range=(0.5, 1.5), satu
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:122
Method__init__
(self, keys, meta_keys=('filename', 'ori_shape', 'img_shape', 'lidar2img','e
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:248
Method__init__
(self, scales=[])
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:299
Method__init__
(self)
projects/mmdet3d_plugin/datasets/pipelines/loading.py:18
Method__init__
(self, inplace=True)
projects/mmdet3d_plugin/models/backbones/vovnet.py:157
Method__init__
(self, channel, reduction=4)
projects/mmdet3d_plugin/models/backbones/vovnet.py:166
Method__init__
( self, in_ch, stage_ch, concat_ch, layer_per_block, module_name, SE=False, identity=False, depthwise=
projects/mmdet3d_plugin/models/backbones/vovnet.py:181
Method__init__
( self, in_ch, stage_ch, concat_ch, block_per_stage, layer_per_block, stage_num, SE=False, depthwise=F
projects/mmdet3d_plugin/models/backbones/vovnet.py:234
Method__init__
(self, use_h, use_w, rotate = 1, offset=False, ratio = 0.5, mode=0, prob = 1.)
projects/mmdet3d_plugin/models/utils/grid_mask.py:71
Method__init__
(self, num_feats, row_num_embed=50, col_num_embed=50,
projects/mmdet3d_plugin/models/utils/positional_encoding.py:24
Method__init__
(self, num_pos_feats=64, pos_norm=True)
projects/mmdet3d_plugin/models/utils/position_embedding.py:6
Method__init__
(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8, weight_decay=1e-2, amsgrad=False)
projects/mmdet3d_plugin/models/opt/adamw.py:38
Method__iter__
(self)
projects/mmdet3d_plugin/datasets/samplers/group_sampler.py:62
Method__iter__
(self)
projects/mmdet3d_plugin/datasets/samplers/distributed_sampler.py:22
Method__len__
(self)
tools/ray_iou/ego_pose_extractor.py:37
Method__len__
(self)
projects/mmdet3d_plugin/datasets/ego_pose_extractor.py:37
Method__len__
(self)
projects/mmdet3d_plugin/datasets/nuscenes_ego_pose_loader.py:74
Method__len__
(self)
projects/mmdet3d_plugin/datasets/samplers/group_sampler.py:105
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:57
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:96
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:191
Method__repr__
str: Return a string that describes the module.
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:285
Method__repr__
(self)
projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py:328
Method__repr__
str: Return a string that describes the module.
projects/mmdet3d_plugin/datasets/pipelines/loading.py:34
Method__repr__
str: a string that describes the module
projects/mmdet3d_plugin/models/utils/positional_encoding.py:64
Method__setstate__
(self, state)
projects/mmdet3d_plugin/models/opt/adamw.py:54
Method_do_evaluate
perform evaluation and save ckpt.
projects/mmdet3d_plugin/core/evaluation/eval_hooks.py:54
Method_initialize_weights
(self)
projects/mmdet3d_plugin/models/backbones/vovnet.py:336
Method_is_power_of_2
(n)
projects/mmdet3d_plugin/bevformer/modules/decoder.py:182
Method_is_power_of_2
(n)
projects/mmdet3d_plugin/bevformer/modules/temporal_self_attention.py:79
Method_is_power_of_2
(n)
projects/mmdet3d_plugin/bevformer/modules/spatial_cross_attention.py:226
Methodafter_train_iter
(self, runner)
projects/mmdet3d_plugin/bevformer/hooks/custom_hooks.py:11
Methodafter_train_iter
(self, runner)
projects/mmdet3d_plugin/models/hooks/hooks.py:8
Methodassign
Computes one-to-one matching based on the weighted costs. This method assign each query prediction to a ground truth or background. Th
projects/mmdet3d_plugin/core/bbox/assigners/hungarian_assigner_3d.py:52
Methodbackward
GPU version of backward function. Args: grad_output (Tensor): Gradient of output tensor of forward.
projects/mmdet3d_plugin/bevformer/modules/multi_scale_deformable_attn_function.py:57
Methodbackward
GPU version of backward function. Args: grad_output (Tensor): Gradient of output tensor of forward.
projects/mmdet3d_plugin/bevformer/modules/multi_scale_deformable_attn_function.py:133
Methodbefore_train_epoch
Evaluate the model only at the start of training by epoch.
projects/mmdet3d_plugin/core/evaluation/eval_hooks.py:45
Methodbefore_train_iter
(self, runner)
projects/mmdet3d_plugin/core/evaluation/eval_hooks.py:50
Functioncustom_encode_mask_results
Encode bitmap mask to RLE code. Semantic Masks only Args: mask_results (list | tuple[list]): bitmap mask results. In mask s
projects/mmdet3d_plugin/bevformer/apis/test.py:25
Methodforward
Forward function for `Detr3DTransformerDecoder`. Args: query (Tensor): Input query with shape `(num_query, bs,
projects/mmdet3d_plugin/bevformer/modules/decoder.py:66
Methodforward
Forward Function of MultiScaleDeformAttention. Args: query (Tensor): Query of Transformer with shape (num_que
projects/mmdet3d_plugin/bevformer/modules/decoder.py:231
Methodforward
Forward Function of MultiScaleDeformAttention. Args: query (Tensor): Query of Transformer with shape (num_que
projects/mmdet3d_plugin/bevformer/modules/temporal_self_attention.py:128
Methodforward
GPU version of multi-scale deformable attention. Args: value (Tensor): The value has shape (bs, num_keys, mum
projects/mmdet3d_plugin/bevformer/modules/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
projects/mmdet3d_plugin/bevformer/modules/multi_scale_deformable_attn_function.py:94
Methodforward
Forward function for `Detr3DTransformer`. Args: mlvl_feats (list(Tensor)): Input queries from different level.
projects/mmdet3d_plugin/bevformer/modules/transformer_occ.py:245
Methodforward
Forward function for `Detr3DTransformer`. Args: mlvl_feats (list(Tensor)): Input queries from different level.
projects/mmdet3d_plugin/bevformer/modules/transformer.py:203
Methodforward
Forward function for `TransformerDecoderLayer`. **kwargs contains some specific arguments of attentions. Args: query (T
projects/mmdet3d_plugin/bevformer/modules/custom_base_transformer_layer.py:167
Methodforward
Forward function for `TransformerDecoder`. Args: bev_query (Tensor): Input BEV query with shape `(num_query, bs
projects/mmdet3d_plugin/bevformer/modules/encoder.py:154
Methodforward
Forward function for `TransformerDecoderLayer`. **kwargs contains some specific arguments of attentions. Args: quer
projects/mmdet3d_plugin/bevformer/modules/encoder.py:287
Methodforward
Forward Function of Detr3DCrossAtten. Args: query (Tensor): Query of Transformer with shape (num_query, bs, emb
projects/mmdet3d_plugin/bevformer/modules/spatial_cross_attention.py:76
Methodforward
Forward Function of MultiScaleDeformAttention. Args: query (Tensor): Query of Transformer with shape ( bs, num_
projects/mmdet3d_plugin/bevformer/modules/spatial_cross_attention.py:273
Methodforward
Forward function. Args: mlvl_feats (tuple[Tensor]): Features from the upstream network, each is a 5D-tensor wit
projects/mmdet3d_plugin/bevformer/dense_heads/bevformer_occ_head.py:100
Methodforward
Calls either forward_train or forward_test depending on whether return_loss=True. Note this setting will change the expected inputs.
projects/mmdet3d_plugin/bevformer/detectors/bevformer_occ.py:144
Methodforward
(self, x)
projects/mmdet3d_plugin/models/backbones/vovnet.py:161
Methodforward
(self, x)
projects/mmdet3d_plugin/models/backbones/vovnet.py:172
Methodforward
(self, x)
projects/mmdet3d_plugin/models/backbones/vovnet.py:210
Methodforward
(self, x)
projects/mmdet3d_plugin/models/backbones/vovnet.py:341
Methodforward
(self, x)
projects/mmdet3d_plugin/models/utils/grid_mask.py:85
Methodforward
Forward function for `LearnedPositionalEncoding`. Args: mask (Tensor): ByteTensor mask. Non-zero values representing
projects/mmdet3d_plugin/models/utils/positional_encoding.py:39
Methodforward
(self, tensor)
projects/mmdet3d_plugin/models/utils/position_embedding.py:15
Methodforward_dummy
(self, img)
projects/mmdet3d_plugin/bevformer/detectors/bevformer_occ.py:140
Methodget_data_info
Get data info according to the given index. Args: index (int): Index of the sample data to get. Returns: dic
projects/mmdet3d_plugin/datasets/nuscenes_occ.py:52
Functioninit
tools/ray_iou/lib/dvr/dvr.cpp:61
Methodinit_weights
Initialize the transformer weights.
projects/mmdet3d_plugin/bevformer/modules/transformer.py:86
Methodinit_weights
Initialize weights of the DeformDETR head.
projects/mmdet3d_plugin/bevformer/dense_heads/bevformer_occ_head.py:91
Methodload_annotations
Load annotations from ann_file. Args: ann_file (str): Path of the annotation file. Returns: list[dict]: List
projects/mmdet3d_plugin/datasets/nuscenes_occ.py:28
Functionmain
(sem_pred_list, sem_gt_list, flow_pred_list, flow_gt_list, lidar_origin_list)
projects/mmdet3d_plugin/datasets/ray_metrics.py:200
Functionmeshgrid3d
(occ_size, pc_range)
tools/ray_iou/ray_casting.py:105
Functionmeshgrid3d
(occ_size, pc_range)
projects/mmdet3d_plugin/datasets/ray_metrics.py:49
Functionmiddle
(fn)
projects/mmdet3d_plugin/models/utils/bricks.py:8
Methodobtain_history_bev
Obtain history BEV features iteratively. To save GPU memory, gradients are not calculated.
projects/mmdet3d_plugin/bevformer/detectors/bevformer_occ.py:159
Functionrender
tools/ray_iou/lib/dvr/dvr.cpp:51
Functionrender_forward
tools/ray_iou/lib/dvr/dvr.cpp:39
Functionrotz
Rotation about the z-axis.
utils/vis.py:60
Methodrun_iter
(self, data_batch, train_mode, **kwargs)
projects/mmdet3d_plugin/bevformer/runner/epoch_based_runner.py:57
Functionrun_time
(name)
projects/mmdet3d_plugin/models/utils/bricks.py:7
Functionsave_tensor
(tensor, path, pad_value=254.0,)
projects/mmdet3d_plugin/models/utils/visual.py:15
Methodset_epoch
(self, epoch)
projects/mmdet3d_plugin/datasets/samplers/group_sampler.py:108
Methodset_prob
(self, epoch, max_epoch)
projects/mmdet3d_plugin/models/utils/grid_mask.py:18
Methodset_prob
(self, epoch, max_epoch)
projects/mmdet3d_plugin/models/utils/grid_mask.py:82
Functionsingle_gpu_test
Test model with single gpu. This method tests model with single gpu and gives the 'show' option. By setting ``show=True``, it saves the vi
projects/mmdet3d_plugin/bevformer/apis/test.py:142
Methodstep
Performs a single optimization step. Args: closure (callable, optional): A closure that reevaluates the model and
projects/mmdet3d_plugin/models/opt/adamw.py:60
← previousnext →101–200 of 203, ranked by callers