MCPcopy Create free account

hub / github.com/InternRobotics/EmbodiedScan / functions

Functions577 in github.com/InternRobotics/EmbodiedScan

↓ 85 callersMethodcat
Concatenate a list of Points into a single Points. Args: points_list (Sequence[:obj:`BasePoints`]): List of points. Retu
embodiedscan/structures/points/base_points.py:404
↓ 29 callersMethodcat
Concatenate a list of Boxes into a single Boxes. Args: boxes_list (Sequence[:obj:`BaseInstance3DBoxes`]): List of boxes.
embodiedscan/structures/bbox_3d/base_box3d.py:391
↓ 25 callersMethodnumpy
Reload ``numpy`` from self.tensor.
embodiedscan/structures/points/base_points.py:425
↓ 23 callersMethodto
Convert current points to a specific device. Args: device (str or :obj:`torch.device`): The name of the device. Returns:
embodiedscan/structures/points/base_points.py:429
↓ 22 callersMethodcpu
Convert current points to cpu device. Returns: :obj:`BasePoints`: A new points object on the cpu device.
embodiedscan/structures/points/base_points.py:444
↓ 13 callersMethodload
Load basic information of a given RGBD frame.
embodiedscan/converter/generate_image_scannet.py:25
↓ 13 callersFunctionto_tensor
Convert objects of various python types to :obj:`torch.Tensor`. Supported types are: :class:`numpy.ndarray`, :class:`torch.Tensor`, :class:`S
embodiedscan/datasets/transforms/formatting.py:15
↓ 11 callersMethodclone
Clone the boxes. Returns: :obj:`BaseInstance3DBoxes`: Box object with the same properties as self.
embodiedscan/structures/bbox_3d/base_box3d.py:455
↓ 10 callersMethodclone
Clone the points. Returns: :obj:`BasePoints`: Point object with the same properties as self.
embodiedscan/structures/points/base_points.py:466
↓ 10 callersFunctionrotation_3d_in_euler
Rotate points by angles according to axis. Args: points (np.ndarray | torch.Tensor | list | tuple ): Points of shape (N, M, 3
embodiedscan/structures/bbox_3d/utils.py:33
↓ 9 callersMethod__setattr__
setattr is only used to set data. The value must have the attribute of `__len__` and have the same length of `PointData`.
embodiedscan/utils/typing_config.py:77
↓ 9 callersFunctionlimit_period
Limit the value into a period for periodic function. Args: val (np.ndarray or Tensor): The value to be converted. offset (float):
embodiedscan/structures/bbox_3d/utils.py:14
↓ 8 callersFunctionrotation_3d_in_axis
Rotate points by angles according to axis. Args: points (np.ndarray or Tensor): Points with shape (N, M, 3). angles (np.ndarray o
embodiedscan/structures/bbox_3d/utils.py:90
↓ 6 callersFunctiondraw_camera
Draw the camera pose in the form of a cone. Args: camera_pose (numpy.ndarray): 4x4 camera pose from camera to world. camera_size
embodiedscan/visualization/utils.py:129
↓ 6 callersFunctionget_box_type
Get the type and mode of box structure. We temporarily only support EulerDepthInstance3DBoxes to support 9-DoF box operations and will co
embodiedscan/structures/bbox_3d/utils.py:208
↓ 6 callersMethodget_color
Get the color of a given category label. Args: label (str): Category name. Returns: list: RGB value of the m
embodiedscan/visualization/color_selector.py:903
↓ 6 callersFunctionget_proj_mat_by_coord_type
Obtain image features using points. Args: img_meta (dict): Meta information. coord_type (str): 'DEPTH' or 'CAMERA' or 'LIDAR'. Ca
embodiedscan/structures/bbox_3d/utils.py:417
↓ 6 callersFunctionrun_subprocess
(command)
install.py:7
↓ 6 callersMethodto
Convert current boxes to a specific device. Args: device (str or :obj:`torch.device`): The name of the device. Returns:
embodiedscan/structures/bbox_3d/base_box3d.py:417
↓ 5 callersFunction_9dof_to_box
Convert 9-DoF box from array/tensor to open3d.OrientedBoundingBox. Args: box (numpy.ndarray|torch.Tensor|List[float]): 9-DoF
embodiedscan/visualization/utils.py:43
↓ 5 callersFunctionbatch_point_sample
Batch version of point_sample. Args: img_meta (dict): Meta info. img_features (Tensor): B x C x H x W image features. poi
embodiedscan/models/layers/fusion_layers/point_fusion.py:208
↓ 5 callersMethodconvert_point
Convert points from ``src`` mode to ``dst`` mode. Args: box (Sequence[float] or np.ndarray or Tensor or :obj:`BasePoints`):
embodiedscan/structures/bbox_3d/coord_3d_mode.py:171
↓ 5 callersMethodflip
Flip the points along given BEV direction. Args: bev_direction (str): Flip direction (horizontal or vertical). De
embodiedscan/structures/points/base_points.py:206
↓ 5 callersFunctionpoints_cam2img
Project points in camera coordinates to image coordinates. Args: points_3d (Tensor or np.ndarray): Points in shape (N, 3). proj_m
embodiedscan/structures/bbox_3d/utils.py:244
↓ 5 callersFunctionreduce_mean
Obtain the mean of tensor on different GPUs.
embodiedscan/utils/dist_utils.py:4
↓ 5 callersMethodrotate
Rotate points with the given rotation matrix or angle. Args: rotation (Tensor or np.ndarray or float): Rotation matrix or angle.
embodiedscan/structures/points/base_points.py:168
↓ 4 callersMethod_bbox_pred_to_bbox
Transform predicted bbox parameters to bbox. Args: points (Tensor): Final locations of shape (N, 3) bbox_pred (Tensor
embodiedscan/models/dense_heads/fcaf3d_head.py:564
↓ 4 callersFunctionchamfer_distance
Calculate Chamfer Distance of two sets. Args: src (Tensor): Source set with shape [B, N, C] to calculate Chamfer Distance.
embodiedscan/models/losses/chamfer_distance.py:13
↓ 4 callersMethoddetach
Detach the points. Returns: :obj:`BasePoints`: Point object with the same properties as self.
embodiedscan/structures/points/base_points.py:477
↓ 4 callersMethodoverlaps
Calculate 3D overlaps of two boxes. Note: This function calculates the overlaps between ``boxes1`` and ``boxes2``, ``
tools/eval_script_portable.py:185
↓ 4 callersMethodoverlaps
Calculate 3D overlaps of two boxes. Note: This function calculates the overlaps between ``boxes1`` and ``boxes2``, ``
embodiedscan/structures/bbox_3d/euler_box3d.py:104
↓ 4 callersFunctionpoint_sample
Obtain image features using points. Args: img_meta (dict): Meta info. img_features (Tensor): 1 x C x H x W image features.
embodiedscan/models/layers/fusion_layers/point_fusion.py:110
↓ 4 callersMethodtransform
Transform function to sample points to in indoor scenes. Args: input_dict (dict): Result dict from loading pipeline. Ret
embodiedscan/datasets/transforms/points.py:119
↓ 4 callersMethodtranslate
Translate points with the given translation vector. Args: trans_vector (Tensor or np.ndarray): Translation vector of size 3
embodiedscan/structures/points/base_points.py:215
↓ 3 callersMethod_remove_prefix
(self, key: str)
embodiedscan/datasets/transforms/formatting.py:82
↓ 3 callersFunctioncamera_to_lidar
Convert points in camera coordinate to lidar coordinate. Note: This function is for KITTI only. Args: points (np.ndarray, sh
embodiedscan/structures/ops/box_np_ops.py:14
↓ 3 callersMethodconvert
Convert input array to target data type. Args: input_array (np.ndarray or torch.Tensor or list or tuple or int or
embodiedscan/utils/array_converter.py:261
↓ 3 callersMethodextract_feat
Directly extract features from the backbone+neck. Args: batch_inputs_dict (dict): The model input dict which includes
embodiedscan/models/detectors/sparse_featfusion_single_stage.py:86
↓ 3 callersMethodextract_feat
Extract 3d features from the backbone -> fpn -> 3d projection. -> 3d neck -> bbox_head. Args: batch_inputs_dict (dict):
embodiedscan/models/detectors/dense_fusion_occ.py:120
↓ 3 callersMethodextract_feat
Directly extract features from the backbone+neck. Args: batch_inputs_dict (dict): The model input dict which includes
embodiedscan/models/detectors/sparse_featfusion_grounder.py:176
↓ 3 callersMethodextract_feat
Directly extract features from the backbone+neck. Args: batch_inputs_dict (dict): The model input dict which includes
embodiedscan/models/detectors/embodied_det3d.py:90
↓ 3 callersMethodextract_feat
Extract 3d features from the backbone -> fpn -> 3d projection. -> 3d neck -> bbox_head. Args: batch_inputs_dict (dict):
embodiedscan/models/detectors/embodied_occ.py:118
↓ 3 callersFunctionget_points_type
Get the class of points according to coordinate type. Args: points_type (str): The type of points coordinate. The valid value are
embodiedscan/structures/points/__init__.py:10
↓ 3 callersMethodindex_to_str
Convert the sample index to string.
embodiedscan/converter/generate_image_scannet.py:130
↓ 3 callersFunctioninstall_requires
(fname)
install.py:97
↓ 3 callersMethodnumpy
Reload ``numpy`` from self.tensor.
embodiedscan/structures/bbox_3d/base_box3d.py:413
↓ 3 callersMethodpack_single_results
Method to pack the single input data. when the value in this dict is a list, it usually is in Augmentations Testing. Args:
embodiedscan/datasets/transforms/formatting.py:122
↓ 3 callersMethodsave_mat_to_file
Save a matrix to file.
embodiedscan/converter/generate_image_scannet.py:135
↓ 3 callersMethodscale
Scale the points with horizontal and vertical scaling factors. Args: scale_factors (float): Scale factors to scale the points.
embodiedscan/structures/points/base_points.py:308
↓ 2 callersMethod__init__
(self, transforms, n_images, ordered=False)
embodiedscan/datasets/transforms/multiview.py:28
↓ 2 callersMethod__init__
(self, input_channel, embed_dims=256)
embodiedscan/models/layers/ground_transformer/decoder.py:23
↓ 2 callersMethod_bbox_pred_to_bbox
Transform predicted bbox parameters to bbox. Args: points (Tensor): Final locations of shape (N, 3) bbox_pred (Tensor
embodiedscan/models/dense_heads/fcaf3d_head.py:1455
↓ 2 callersFunction_box_add_thickness
(box, thickness)
embodiedscan/visualization/utils.py:33
↓ 2 callersMethod_remove_dontcare
Remove annotations that do not need to be cared. -1 indicates dontcare in MMDet3d. Args: ann_info (dict): Dict of annota
embodiedscan/datasets/embodiedscan_dataset.py:289
↓ 2 callersMethodanchors_single_range
Generate anchors in a single range. Args: feature_size (list[float] | tuple[float]): Feature map size. It is eith
embodiedscan/models/task_modules/anchor/anchor_3d_generator.py:164
↓ 2 callersFunctionapply_3d_transformation
Apply transformation to input point cloud. Args: pcd (Tensor): The point cloud to be transformed. coord_type (str): 'DEPTH' or 'C
embodiedscan/models/layers/fusion_layers/point_fusion.py:20
↓ 2 callersFunctionbbox_to_corners
Transform bbox parameters to the 8 corners. Args: bbox (Tensor): 3D box of shape (N, 6) or (N, 7) or (N, 9). Returns: Tensor
embodiedscan/models/losses/chamfer_distance.py:160
↓ 2 callersFunctioncenter_to_corner_box2d
Convert kitti locations, dimensions and angles to corners. format: center(xy), dims(xy), angles(counterclockwise when positive) Args:
embodiedscan/structures/ops/box_np_ops.py:96
↓ 2 callersFunctioncenter_to_corner_box3d
Convert kitti locations, dimensions and angles to corners. Args: centers (np.ndarray): Locations in kitti label file with shape (N, 3).
embodiedscan/structures/ops/box_np_ops.py:171
↓ 2 callersMethodclose
Close the visualizer. Args: vis (open3d.visualization.VisualizerWithKeyCallback): Visualizer.
embodiedscan/visualization/continuous_drawer.py:176
↓ 2 callersMethodconvert_box
Convert boxes from ``src`` mode to ``dst`` mode. Args: box (Sequence[float] or np.ndarray or Tensor or :obj:`Base
embodiedscan/structures/bbox_3d/coord_3d_mode.py:131
↓ 2 callersFunctioncorners_nd
Generate relative box corners based on length per dim and origin point. Args: dims (np.ndarray, shape=[N, ndim]): Array of length per dim
embodiedscan/structures/ops/box_np_ops.py:63
↓ 2 callersFunctioncross_product
(a, b)
embodiedscan/models/dense_heads/grounding_head.py:833
↓ 2 callersFunctioncross_product
(a, b)
embodiedscan/models/dense_heads/fcaf3d_head.py:1734
↓ 2 callersMethodcuda
Convert current points to cuda device. Returns: :obj:`BasePoints`: A new points object on the cuda device.
embodiedscan/structures/points/base_points.py:455
↓ 2 callersMethoddevice
torch.device: The device of the points are on.
embodiedscan/structures/points/base_points.py:489
↓ 2 callersMethoddevice
torch.device: The device of the boxes are on.
embodiedscan/structures/bbox_3d/base_box3d.py:480
↓ 2 callersMethodforward
Forward function. Args: mlvl_feats (list[Tensor]): Multi-level features. input_metas (list[dict]): Input meta infos.
embodiedscan/models/dense_heads/imvoxel_occ_head.py:73
↓ 2 callersMethodforward_single
Scatters points into voxels. Args: points (torch.Tensor): Points to be reduced into voxels. coors (torch.Tensor): Cor
embodiedscan/models/data_preprocessors/voxelize.py:269
↓ 2 callersMethodforward_transformer
(self, point_feats: List[Tensor], scores: List[Tensor]
embodiedscan/models/detectors/sparse_featfusion_grounder.py:312
↓ 2 callersFunctiongeo_scal_loss
Geometric scene-class affinity loss. Only consider empty and nonempty probabilities. Args: pred (Tensor): Prediction maps. s
embodiedscan/models/losses/occ_loss.py:39
↓ 2 callersMethodget_positive_map
(self, tokenized, tokens_positive)
embodiedscan/models/detectors/sparse_featfusion_grounder.py:623
↓ 2 callersMethodgrid_anchors
Generate grid anchors in multiple feature levels. Args: featmap_sizes (list[tuple]): List of feature map sizes in
embodiedscan/models/task_modules/anchor/anchor_3d_generator.py:92
↓ 2 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.
embodiedscan/structures/bbox_3d/base_box3d.py:525
↓ 2 callersFunctionnormalize_vector
(vector)
embodiedscan/models/dense_heads/grounding_head.py:827
↓ 2 callersFunctionnormalize_vector
(vector)
embodiedscan/models/dense_heads/fcaf3d_head.py:1728
↓ 2 callersFunctionocc_multiscale_supervision
Produce multi-scale occupancy supervision from ground truth. Args: gt_occ (list[Tensor]): Ground truth occupancy. ratio (int): Do
embodiedscan/models/losses/occ_loss.py:7
↓ 2 callersFunctionortho_6d_2_Mat
x_raw, y_raw: both tensors (batch, 3).
embodiedscan/models/dense_heads/grounding_head.py:838
↓ 2 callersFunctionortho_6d_2_Mat
x_raw, y_raw: both tensors batch*3.
embodiedscan/models/dense_heads/fcaf3d_head.py:1739
↓ 2 callersMethodparse_ann_info
Process the `instances` in data info to `ann_info`. Args: info (dict): Info dict. Returns: dict: Processed `
embodiedscan/datasets/mv_3dvg_dataset.py:468
↓ 2 callersMethodparse_ann_info
Process the `instances` in data info to `ann_info`. Args: info (dict): Info dict. Returns: dict: Processed `
embodiedscan/datasets/embodiedscan_dataset.py:159
↓ 2 callersFunctionparse_require_file
(fpath)
setup.py:54
↓ 2 callersFunctionparse_requirements
Parse the package dependencies listed in a requirements file but strips specific versioning information. Args: fname (str): path to r
setup.py:4
↓ 2 callersFunctionpoints_img2cam
Project points in image coordinates to camera coordinates. Args: points (Tensor or np.ndarray): 2.5D points in 2D images with shape
embodiedscan/structures/bbox_3d/utils.py:336
↓ 2 callersFunctionpoints_in_convex_polygon_3d_jit
Check points is in 3d convex polygons. Args: points (np.ndarray): Input points with shape of (num_points, 3). polygon_surfaces (n
embodiedscan/structures/ops/box_np_ops.py:690
↓ 2 callersMethodpreprocess_img
(self, _batch_img: Tensor)
embodiedscan/models/data_preprocessors/data_preprocessor.py:249
↓ 2 callersMethodrandom_flip_data_3d
Flip 3D data randomly. `random_flip_data_3d` should take these situations into consideration: - 1. LIDAR-based 3d detection
embodiedscan/datasets/transforms/augmentation.py:141
↓ 2 callersMethodset_template
Set template array. Args: array (np.ndarray or torch.Tensor or list or tuple or int or float): Template array.
embodiedscan/utils/array_converter.py:220
↓ 2 callersMethodsimple_process
Perform normalization, padding and bgr2rgb conversion for img data based on ``BaseDataPreprocessor``, and voxelize point cloud if `voxel`
embodiedscan/models/data_preprocessors/data_preprocessor.py:160
↓ 1 callersMethod__init__
(self, iou_thr: Union[float, List[float]] = [0.5], collect_device: str = 'cp
embodiedscan/eval/metrics/det_metric.py:181
↓ 1 callersMethod__init__
(self, imdecode_backend: str = 'cv2', ignore_empty: bool = False,
embodiedscan/datasets/transforms/loading.py:41
↓ 1 callersMethod__init__
(self, in_channels, out_channels, stride=1)
embodiedscan/models/necks/imvoxel_neck.py:113
↓ 1 callersMethod__init__
(self, num_classes: int, embed_dims: int = 256, num_pred_la
embodiedscan/models/dense_heads/grounding_head.py:106
↓ 1 callersMethod__init__
(self, num_classes: int, in_channels: Tuple[int], out_chann
embodiedscan/models/dense_heads/fcaf3d_head.py:62
↓ 1 callersMethod__init__
(self, voxel_size: List, point_cloud_range: List, average_points: bool)
embodiedscan/models/data_preprocessors/voxelize.py:261
↓ 1 callersMethod__init__
(self, ranges: List[List[float]], sizes: List[List[float]] = [[3.9, 1.6, 1.5
embodiedscan/models/task_modules/anchor/anchor_3d_generator.py:42
↓ 1 callersMethod__init__
( self, mode: str = 'l2', group: str = 'g8', reduction: str = 'mean',
embodiedscan/models/losses/chamfer_distance.py:223
↓ 1 callersMethod__init__
(self, weight)
embodiedscan/models/losses/match_cost.py:99
↓ 1 callersMethod__len__
int: Number of points in the current object.
embodiedscan/structures/points/base_points.py:395
↓ 1 callersMethod_focal_loss_cost
Args: cls_pred (Tensor): Predicted classification logits, shape (num_queries, num_class). gt_labels (
embodiedscan/models/losses/match_cost.py:142
next →1–100 of 577, ranked by callers