MCPcopy Create free account

hub / github.com/VDIGPKU/HENet / functions

Functions6,925 in github.com/VDIGPKU/HENet

↓ 1,152 callersMethodcat
Concatenate a list of Points into a single Points. Args: points_list (list[:obj:`BasePoints`]): List of points. Returns:
mmdet3d/core/points/base_points.py:357
↓ 910 callersFunctionprint
(*args, **kwargs)
detr2/projects/DDETRS/ddetrs/util/misc.py:354
↓ 693 callersMethodget
Returns a record from the layer in constant runtime. :param layer_name: Name of the layer that we are interested in. :param t
tools/data_converter/map_api.py:203
↓ 431 callersMethodto
Convert current points to a specific device. Args: device (str | :obj:`torch.device`): The name of the device. Returns:
mmdet3d/core/points/base_points.py:379
↓ 364 callersMethodupdate
(self, trainer, model)
mmdet3d/core/hook/ema.py:48
↓ 349 callersMethodmax
(self)
detr2/projects/DDETRS/ddetrs/util/misc.py:77
↓ 259 callersMethodflatten
(cls, obj)
detr2/detectron2/export/flatten.py:37
↓ 251 callersMethodto
(self, device: torch.device)
detr2/detectron2/structures/boxes.py:166
↓ 215 callersMethodload
Load a config file. Args: filename: absolute path or relative path w.r.t. the current working directory keys
detr2/detectron2/config/lazy.py:183
↓ 169 callersMethodclone
Clone the Points. Returns: :obj:`BasePoints`: Box object with the same properties as self.
mmdet3d/core/points/base_points.py:395
↓ 156 callersMethodcopy
Create a copy of self. :return: A copy.
mmdet3d/core/bbox/structures/nuscenes_box.py:374
↓ 151 callersMethodcat
Concatenates a list of Boxes into a single Boxes Arguments: boxes_list (list[Boxes]) Returns: Boxes
detr2/detectron2/structures/boxes.py:277
↓ 131 callersMethoddump
Returns: str: a yaml string representation of the config
detr2/detectron2/config/config.py:87
↓ 114 callersFunctionbuild_loss
Build loss function.
mmdet3d/models/builder.py:74
↓ 113 callersMethodget
Returns the field called `name`.
detr2/detectron2/structures/instances.py:94
↓ 107 callersMethodall
Returns all EvalBoxes in a list.
mmdet3d/datasets/nuscenes_styled_eval_utils.py:527
↓ 99 callersMethodpad
(self, img)
mmdet3d/datasets/pipelines/transform_3d_focalformer3d.py:366
↓ 96 callersMethodlog
(self, runner)
mmdet3d/core/hook/logger.py:82
↓ 92 callersMethodgather
(self, image_ids)
tools/data_converter/kitti_data_utils.py:502
↓ 88 callersMethodto
(self, device, non_blocking=False)
detr2/projects/DDETRS/ddetrs/util/misc.py:324
↓ 87 callersMethodrotate
Rotate points with the given rotation matrix or angle. Args: rotation (float | np.ndarray | torch.Tensor): Rotation matrix
mmdet3d/core/points/base_points.py:145
↓ 80 callersMethodnorm
Normalize the feature maps to have zero mean and unit variances. Args: feat (torch.Tensor): The original feature map with shape
mmdet3d/models/losses/pkd_loss.py:37
↓ 79 callersMethodupdate
(self, trajs, gt_trajs, gt_trajs_mask, fut_boxes)
mmdet3d/datasets/evaluation/planning/planning_eval.py:114
↓ 77 callersMethodclip
Clip (in place) the boxes by limiting x coordinates to the range [0, width] and y coordinates to the range [0, height]. Args
detr2/detectron2/structures/boxes.py:181
↓ 68 callersMethoddecode
Decode regression into locations, dimensions, orientations. Args: reg (Tensor): Batch regression for each predict center2d point.
mmdet3d/core/bbox/coders/smoke_bbox_coder.py:52
↓ 65 callersMethodcat
Concatenate a list of Boxes into a single Boxes. Args: boxes_list (list[:obj:`BaseInstance3DBoxes`]): List of boxes. Ret
mmdet3d/core/bbox/structures/base_box3d.py:341
↓ 61 callersMethodtranslate
Translate points with the given translation vector. Args: trans_vector (np.ndarray, torch.Tensor): Translation ve
mmdet3d/core/points/base_points.py:182
↓ 55 callersMethoddevice
(self)
mmdet3d/models/backbones/sam.py:681
↓ 53 callersFunctionget_norm
Args: norm (str or callable): either one of BN, SyncBN, FrozenBN, GN; or a callable that takes a channel number and returns
detr2/detectron2/layers/batch_norm.py:121
↓ 51 callersFunctionbbox3d2result
Convert detection results to a list of numpy arrays. Args: bboxes (torch.Tensor): Bounding boxes with shape (N, 5). labels (torch
mmdet3d/core/bbox/transforms.py:50
↓ 51 callersMethodsave
Save a config object to a yaml file. Note that when the config dictionary contains complex objects (e.g. lambda), it can't be
detr2/detectron2/config/lazy.py:238
↓ 47 callersMethodget_bboxes
Generate bboxes from bbox head predictions. Args: preds_dicts (tuple[list[dict]]): Prediction results. img_metas (list
mmdet3d/models/dense_heads/farhead.py:1225
↓ 44 callersFunctionbuild_dataset
(cfg, default_args=None)
mmdet3d/datasets/builder.py:34
↓ 40 callersMethodhas
(self, name)
detr2/detectron2/export/c10.py:72
↓ 37 callersMethodloss
Loss function. Args: gt_bboxes_list (list[Tensor]): Ground truth bboxes for each image with shape (num_gts, 4) in
mmdet3d/models/dense_heads/farhead.py:1114
↓ 34 callersMethodextract_feat
Extract features from points.
mmdet3d/models/detectors/sassd.py:41
↓ 34 callersMethodflip
Flip the points along given BEV direction. Args: bev_direction (str): Flip direction (horizontal or vertical).
mmdet3d/core/points/base_points.py:174
↓ 31 callersMethodclone
Clone the Boxes. Returns: :obj:`BaseInstance3DBoxes`: Box object with the same properties as self.
mmdet3d/core/bbox/structures/base_box3d.py:379
↓ 31 callersMethodload_state_dict
(self, state_dict)
detr2/detectron2/engine/hooks.py:369
↓ 30 callersFunctionbuild_model
A function warpper for building 3D detector or segmentor according to cfg. Should be deprecated in the future.
mmdet3d/models/builder.py:116
↓ 30 callersMethodstate_dict
(self)
detr2/detectron2/engine/hooks.py:364
↓ 29 callersFunctioncat
Efficient version of torch.cat that avoids a copy if there is only a single element in a list
detr2/detectron2/layers/wrappers.py:38
↓ 29 callersFunctionload_checkpoint
Load checkpoint from a file or URI. Args: model (Module): Module to load checkpoint. filename (str): Accept local filepath, URL, `
mmdet3d/models/codetr/checkpoint.py:282
↓ 28 callersFunctionbuild_dataloader
Build PyTorch DataLoader. In distributed training, each GPU/process has a dataloader. In non-distributed training, there is only one dataloade
mmdet3d/datasets/builder.py:61
↓ 27 callersFunctionbuild_head
Build head.
mmdet3d/models/builder.py:66
↓ 26 callersMethodbev_encoder
(self, x)
mmdet3d/models/detectors/bevdet.py:147
↓ 26 callersMethodprepare_bev_feat
(self, img, rot, tran, intrin, post_rot, post_tran, bda, mlp_input, ret_2d_feat=False
mmdet3d/models/detectors/bevdet.py:849
↓ 26 callersMethodupdate
Args: predictions: D2 Instances for predictions of the current frame Return: D2 Instances for predictions of
detr2/detectron2/tracking/base_tracker.py:32
↓ 25 callersMethodencode
Encode CameraInstance3DBoxes by locations, dimensions, orientations. Args: locations (Tensor): Center location for 3D boxes.
mmdet3d/core/bbox/coders/smoke_bbox_coder.py:26
↓ 24 callersFunctioncompat_cfg
This function would modify some filed to keep the compatibility of config. For example, it will move some args which will be deprecated to th
mmdet3d/utils/compat_cfg.py:8
↓ 23 callersMethodassign
(self, bboxes, gt_bboxes, gt_labels, cls_pred, train_cfg)
mmdet3d/core/bbox/assigners/hungarian_assigner_3d.py:29
↓ 23 callersFunctionbbox_cxcywh_to_xyxy
Convert bbox coordinates from (cx, cy, w, h) to (x1, y1, x2, y2). Args: bbox (Tensor): Shape (n, 4) for bboxes. Returns: Ten
mmdet3d/datasets/eval_recall.py:7
↓ 23 callersMethodconvert_to
Convert self to ``dst`` mode. Args: dst (:obj:`CoordMode`): The target Box mode. rt_mat (np.ndarray | torch.Tensor, o
mmdet3d/core/points/base_points.py:250
↓ 23 callersMethodevaluate
Evaluation in Lyft protocol. Args: results (list[dict]): Testing results of the dataset. metric (str | list[str], opt
mmdet3d/datasets/lyft_dataset.py:366
↓ 23 callersMethodnorm1
nn.Module: normalization layer after the first convolution layer
mmdet3d/models/bricks/res_block.py:61
↓ 23 callersMethodsample
Sample specific number of ground truths. Args: num (int): Sampled number. Returns: list[dict]: Sampled groun
mmdet3d/datasets/pipelines/dbsampler.py:68
↓ 22 callersMethodadd
(self, name)
mmdet3d/models/nerf/nerf_head.py:91
↓ 22 callersMethoddevice
(self)
detr2/detectron2/structures/boxes.py:297
↓ 22 callersMethodto
Convert current boxes to a specific device. Args: device (str | :obj:`torch.device`): The name of the device. Returns:
mmdet3d/core/bbox/structures/base_box3d.py:363
↓ 21 callersMethoddecoder
(self, replacement, add_prefix_space)
mmdet3d/models/internvl_model/internlm2/tokenization_internlm2_fast.py:54
↓ 21 callersMethodnorm2
nn.Module: normalization layer after the second convolution layer
mmdet3d/models/bricks/res_block.py:66
↓ 21 callersFunctionto_device
This function is a replacement of .to(another_device) such that it allows the casting to be traced properly by explicitly calling the underly
detr2/detectron2/export/shared.py:25
↓ 20 callersFunctionlimit_period
Limit the value into a period for periodic function. Args: val (torch.Tensor | np.ndarray): The value to be converted. offset (fl
mmdet3d/core/bbox/structures/utils.py:11
↓ 19 callersMethodget_bboxes
Convert instance masks to axis-aligned bounding boxes. Args: points (np.array): Scene points of shape (n, 6). pts_ins
tools/data_converter/s3dis_data_utils.py:115
↓ 19 callersMethodget_mlp_input
(self, rot, tran, intrin, post_rot, post_tran, bda)
mmdet3d/models/necks/view_transformer.py:440
↓ 19 callersFunctionpoints_cam2img
Project points in camera coordinates to image coordinates. Args: points_3d (torch.Tensor | np.ndarray): Points in shape (N, 3) pr
mmdet3d/core/bbox/structures/utils.py:176
↓ 19 callersFunctionrotation_3d_in_axis
Rotate points by angles according to axis. Args: points (np.ndarray | torch.Tensor | list | tuple ): Points of shape (N, M, 3
mmdet3d/core/bbox/structures/utils.py:29
↓ 18 callersMethodprepare_inputs
(self, inputs, openad=False)
mmdet3d/models/detectors/bevdet.py:176
↓ 17 callersMethodput
(self, value)
mmdet3d/models/internvl_model/internlm2/modeling_internlm2.py:1245
↓ 16 callersFunctionbox_cxcywh_to_xyxy
(x)
detr2/projects/DDETRS/ddetrs/util/box_ops.py:17
↓ 16 callersMethodget_depth_loss
(self, depth_labels, depth_preds)
mmdet3d/models/necks/view_transformer.py:1142
↓ 15 callersMethod_load_layer
Returns a list of records corresponding to the layer name. :param layer_name: Name of the layer that will be loaded. :return:
tools/data_converter/map_api.py:112
↓ 15 callersFunctionalias
(x, name, is_backward=False)
detr2/detectron2/export/shared.py:592
↓ 15 callersFunctionbuild_backbone
Build backbone.
mmdet3d/models/builder.py:34
↓ 15 callersMethodclone
Clone the Boxes. Returns: Boxes
detr2/detectron2/structures/boxes.py:157
↓ 15 callersMethodconvert
Convert input array to target data type. Args: input_array (tuple | list | np.ndarray | torch.Tensor | int | floa
mmdet3d/core/utils/array_converter.py:253
↓ 15 callersMethodnorm
Normalize the feature maps to have zero mean and unit variances. Args: feat (torch.Tensor): The original feature map with shape
mmdet3d/models/dense_heads/bevmaepp_head.py:221
↓ 15 callersMethodpre_pipeline
Initialization before data preparation. Args: results (dict): Dict before data preprocessing. - img_fields (list
mmdet3d/datasets/custom_3d.py:193
↓ 15 callersMethodremove
Remove the field called `name`.
detr2/detectron2/structures/instances.py:88
↓ 15 callersFunctionreplace_feature
(out, new_features)
mmdet3d/ops/sparse_block.py:14
↓ 14 callersFunctionbuild_sampler
(cfg, default_args)
mmdet3d/datasets/samplers/sampler.py:13
↓ 14 callersFunctiondecode_bbox
(bboxes, pc_range=None)
mmdet3d/core/bbox/utils.py:72
↓ 14 callersFunctioninverse_sigmoid
Inverse function of sigmoid. Args: x (Tensor): The tensor to do the inverse. eps (float): EPS avoid numerical
mmdet3d/models/codetr/transformer.py:16
↓ 14 callersMethodnorm3
nn.Module: normalization layer after the third convolution layer
mmdet3d/models/bricks/res_block.py:271
↓ 14 callersFunctionxywhr2xyxyr
Convert a rotated boxes in XYWHR format to XYXYR format. Args: boxes_xywhr (torch.Tensor | np.ndarray): Rotated boxes in XYWHR format.
mmdet3d/core/bbox/structures/utils.py:121
↓ 13 callersFunctionbuild_neck
Build neck.
mmdet3d/models/builder.py:42
↓ 13 callersFunctioncheck_set_pb_arg
(pb, arg_name, arg_attr, arg_value, allow_override=False)
detr2/detectron2/export/shared.py:204
↓ 13 callersFunctioncheckpoint
r"""Checkpoint a model or part of the model Checkpointing works by trading compute for memory. Rather than storing all intermediate activatio
mmdet3d/models/sparsebev/checkpoint.py:166
↓ 13 callersFunctionconv2d
Helper for building a conv2d layer.
detr2/detectron2/modeling/backbone/regnet.py:28
↓ 13 callersFunctiondigit_version
(version_str)
mmdet3d/__init__.py:9
↓ 13 callersMethodget_ann_info
Get annotation info according to the given index. Args: index (int): Index of the annotation data to get. Returns:
mmdet3d/datasets/lyft_dataset.py:188
↓ 13 callersMethodget_reuse
(self, key, do_shift, name, allow_missing=True)
mmdet3d/ops/sst/sst_ops.py:317
↓ 13 callersMethodnew_box
Create a new box object with data. The new box and its tensor has the similar properties as self and self.tensor, respectively.
mmdet3d/core/bbox/structures/base_box3d.py:489
↓ 12 callersMethod__init__
(self, config, has_relative_attention_bias=False)
detr2/projects/DDETRS/ddetrs/models/text/modeling_t5.py:702
↓ 12 callersFunction_extend_matrix
(mat)
tools/data_converter/kitti_data_utils.py:160
↓ 12 callersMethod_transpose_and_gather_feat
Given feats and indexes, returns the transposed and gathered feats. Args: feat (torch.Tensor): Features to be transposed and gath
mmdet3d/core/bbox/coders/centerpoint_bbox_coders.py:98
↓ 12 callersMethodcompress
(self, x)
mmdet3d/models/backbones/radar_encoder_adapter.py:817
↓ 12 callersFunctionconvbn
(in_planes, out_planes, kernel_size, stride, pad, dilat
mmdet3d/models/backbones/mono_submodule.py:10
↓ 12 callersMethodconvert_point
Convert points from `src` mode to `dst` mode. Args: point (tuple | list | np.ndarray | torch.Tensor | :obj:`BaseP
mmdet3d/core/bbox/structures/coord_3d_mode.py:136
↓ 12 callersMethodnorm
(c)
detr2/detectron2/modeling/meta_arch/retinanet.py:350
next →1–100 of 6,925, ranked by callers