MCPcopy Create free account

hub / github.com/InternRobotics/EmbodiedScan / functions

Functions577 in github.com/InternRobotics/EmbodiedScan

Method__call__
Calculate 3D IoU using cuda implementation. Note: This function calculate the IoU of 3D boxes based on their volumes.
embodiedscan/structures/ops/iou3d_calculator.py:70
Method__call__
Calculate IoU between 2D bboxes. Args: bboxes1 (Tensor): shape (B, m, 6) in <x1, y1, z1, x2, y2, z2> format or em
embodiedscan/structures/ops/iou3d_calculator.py:185
Method__call__
(self, results)
embodiedscan/datasets/transforms/multiview.py:256
Method__call__
Compute match cost. Args: pred_instances (:obj:`InstanceData`): Instances of model predictions. It includes ``pri
embodiedscan/models/losses/match_cost.py:23
Method__call__
Compute match cost. Args: bbox_pred (Tensor): Predicted boxes with normalized coordinates (cx,cy,l,w,cz,h,sin(φ),
embodiedscan/models/losses/match_cost.py:53
Method__call__
Compute match cost. Args: pred_logits (Tensor): Shape [num_query, C]. gt_logits (Tensor): Shape [num_gt, C].
embodiedscan/models/losses/match_cost.py:82
Method__call__
(self, pred_instances: InstanceData, gt_instances: InstanceData)
embodiedscan/models/losses/match_cost.py:102
Method__call__
Compute match cost. Args: pred_instances (:obj:`InstanceData`): Predicted instances which must contain ``scores``
embodiedscan/models/losses/match_cost.py:186
Method__call__
Compute match cost. Args: pred_instances (:obj:`InstanceData`): Predicted instances which must contain ``scores``
embodiedscan/models/losses/match_cost.py:239
Method__getitem__
Args: item (int or slice or np.ndarray or Tensor): Index of boxes. Note: The following usage are allowed:
tools/eval_script_portable.py:147
Method__getitem__
Args: item (str, int, list, :obj:`slice`, :obj:`numpy.ndarray`, :obj:`torch.LongTensor`, :obj:`torch.BoolTensor`)
embodiedscan/utils/typing_config.py:98
Method__getitem__
Args: item (int or tuple or slice or np.ndarray or Tensor): Index of points. Note: The follo
embodiedscan/structures/points/base_points.py:316
Method__getitem__
Args: item (int or slice or np.ndarray or Tensor): Index of boxes. Note: The following usage are allowed:
embodiedscan/structures/bbox_3d/base_box3d.py:348
Method__init__
(self, tensor, box_dim=9, with_yaw=True, o
tools/eval_script_portable.py:101
Method__init__
(self, data_root: Union[dict, List], ann_file: Union[dict, List, str],
embodiedscan/explorer.py:35
Method__init__
(self, template_array: Optional[TemplateArrayType] = None)
embodiedscan/utils/array_converter.py:215
Method__init__
Creates a line represented as sequence of cylinder triangular meshes. Arguments: points {ndarray} -- Numpy array of ponts
embodiedscan/utils/line_mesh.py:44
Method__init__
(self, image, verbose=False)
embodiedscan/visualization/img_drawer.py:18
Method__init__
(self, classes=EMBODIED_CATE, init_file=None, verbose=False)
embodiedscan/visualization/color_selector.py:860
Method__init__
(self, dataset, dir, scene, classes, id_to_index, color_selector, start_idx, pcd_downsample,
embodiedscan/visualization/continuous_drawer.py:30
Method__init__
(self, dataset, dir, scene, classes, id_to_index, color_selector, start_idx)
embodiedscan/visualization/continuous_drawer.py:204
Method__init__
(self, dataset, dir, scene, classes, id_to_index, color_selector, start_idx)
embodiedscan/visualization/continuous_drawer.py:371
Method__init__
Creates a line represented as sequence of cylinder triangular meshes. Arguments: points {ndarray} -- Numpy array of ponts
embodiedscan/visualization/line_mesh.py:44
Method__init__
(self, coordinate='lidar')
embodiedscan/structures/ops/iou3d_calculator.py:21
Method__init__
(self, coordinate)
embodiedscan/structures/ops/iou3d_calculator.py:66
Method__init__
(self, tensor: Union[Tensor, np.ndarray, Sequence[Sequence[float]]], points_
embodiedscan/structures/points/lidar_points.py:30
Method__init__
(self, tensor: Union[Tensor, np.ndarray, Sequence[Sequence[float]]], points_
embodiedscan/structures/points/cam_points.py:30
Method__init__
(self, tensor: Union[Tensor, np.ndarray, Sequence[Sequence[float]]], points_
embodiedscan/structures/points/base_points.py:34
Method__init__
(self, tensor: Union[Tensor, np.ndarray, Sequence[Sequence[float]]], points_
embodiedscan/structures/points/depth_points.py:30
Method__init__
(self, tensor, box_dim=9, origin=(0.5, 0.5, 0.5))
embodiedscan/structures/bbox_3d/euler_box3d.py:24
Method__init__
(self, tensor, box_dim=9, with_yaw=True, o
embodiedscan/structures/bbox_3d/euler_depth_box3d.py:41
Method__init__
( self, tensor: Union[Tensor, np.ndarray, Sequence[Sequence[float]]], box_dim: int = 7
embodiedscan/structures/bbox_3d/base_box3d.py:42
Method__init__
(self, collect_device: str = 'cpu', prefix: Optional[str] = None,
embodiedscan/eval/metrics/occupancy_metric.py:33
Method__init__
(self, iou_thr: List[float] = [0.25, 0.5], collect_device: str = 'cpu',
embodiedscan/eval/metrics/det_metric.py:36
Method__init__
(self, iou_thr: List[float] = [0.25, 0.5], collect_device: str = 'cpu',
embodiedscan/eval/metrics/grounding_metric.py:36
Method__init__
(self, data_root: str, ann_file: str, vg_file: str,
embodiedscan/datasets/mv_3dvg_dataset.py:129
Method__init__
(self, data_root: str, ann_file: str, metainfo: Optional[di
embodiedscan/datasets/embodiedscan_dataset.py:43
Method__init__
(self, sync_2d: bool = True, flip_2d: bool = True, flip_3d:
embodiedscan/datasets/transforms/augmentation.py:58
Method__init__
(self, rot_range: Union[List[float], int, float] = [-0.7853
embodiedscan/datasets/transforms/augmentation.py:289
Method__init__
(self, with_bbox_3d: bool = True, with_label_3d: bool = True,
embodiedscan/datasets/transforms/loading.py:181
Method__init__
(self, coord_type: str = 'CAMERA', use_color: bool = False)
embodiedscan/datasets/transforms/points.py:23
Method__init__
(self, num_points: int, sample_range: Optional[float] = None,
embodiedscan/datasets/transforms/points.py:111
Method__init__
(self, point_cloud_range: List[float])
embodiedscan/datasets/transforms/points.py:243
Method__init__
(self, coord_type: str = 'DEPTH', save_slices: bool = False)
embodiedscan/datasets/transforms/multiview.py:131
Method__init__
(self)
embodiedscan/datasets/transforms/multiview.py:176
Method__init__
( self, keys: dict, meta_keys: dict = ( 'img_path', 'ori_shape', 'img_shap
embodiedscan/datasets/transforms/formatting.py:64
Method__init__
(self, name: str = 'visualizer', save_dir: str = None, vis_
embodiedscan/visualizer/base_visualizer.py:29
Method__init__
( self, num_classes: int, # 1 in_channels: Tuple[int], out_ch
embodiedscan/models/necks/mink_neck.py:40
Method__init__
(self, in_channels: List[int], out_channels: int, kernel_si
embodiedscan/models/necks/channel_mapper.py:56
Method__init__
(self, in_channels, out_channels, n_blocks)
embodiedscan/models/necks/imvoxel_neck.py:18
Method__init__
(self, depth: int, in_channels: int, num_stages: int = 4,
embodiedscan/models/backbones/mink_resnet.py:40
Method__init__
(self, max_text_len: int = 256, log_scale: Optional[Union[str, float]] = Non
embodiedscan/models/dense_heads/grounding_head.py:42
Method__init__
(self, *args, num_classes=21, volume_h=40,
embodiedscan/models/dense_heads/imvoxel_occ_head.py:33
Method__init__
(self, num_classes: int, in_channels: Tuple[int], out_chann
embodiedscan/models/dense_heads/fcaf3d_head.py:860
Method__init__
(self, point_cloud_range: List, max_num_points: int, voxel_
embodiedscan/models/data_preprocessors/voxelize.py:129
Method__init__
(self, voxel: bool = False, voxel_type: str = 'hard', voxel
embodiedscan/models/data_preprocessors/data_preprocessor.py:90
Method__init__
(self, backbone: ConfigType, backbone_3d: ConfigType, bbox_
embodiedscan/models/detectors/sparse_featfusion_single_stage.py:53
Method__init__
(self, backbone: ConfigType, backbone_3d: ConfigType, neck:
embodiedscan/models/detectors/dense_fusion_occ.py:58
Method__init__
(self, backbone: ConfigType, backbone_3d: ConfigType, bbox_
embodiedscan/models/detectors/sparse_featfusion_grounder.py:60
Method__init__
(self, backbone: ConfigType, backbone_3d: ConfigType, bbox_
embodiedscan/models/detectors/embodied_det3d.py:54
Method__init__
(self, backbone: ConfigType, backbone_3d: ConfigType, neck:
embodiedscan/models/detectors/embodied_occ.py:56
Method__init__
(self, align_corner: bool = False, **kwargs)
embodiedscan/models/task_modules/anchor/anchor_3d_generator.py:267
Method__init__
( self, match_costs: Union[List[Union[dict, ConfigDict]], dict, Confi
embodiedscan/models/task_modules/assigners/hungarian_assigner.py:41
Method__init__
(self, self_attn_cfg: OptConfigType = dict(embed_dims=256,
embodiedscan/models/layers/ground_transformer/decoder.py:39
Method__init__
(self, num_layers: int, layer_cfg: ConfigType, post_norm_cf
embodiedscan/models/layers/ground_transformer/decoder.py:197
Method__init__
(self, img_channels: Union[List[int], int], pts_channels: int,
embodiedscan/models/layers/fusion_layers/point_fusion.py:350
Method__init__
(self, mode: str = 'l2', reduction: str = 'mean', loss_src_
embodiedscan/models/losses/chamfer_distance.py:96
Method__init__
(self, weight: Union[float, int] = 1.)
embodiedscan/models/losses/match_cost.py:19
Method__init__
(self, alpha: Union[float, int] = 0.25, gamma: Union[float, int] = 2,
embodiedscan/models/losses/match_cost.py:130
Method__init__
(self, reduction: str = 'mean', loss_weight: float = 1.0)
embodiedscan/models/losses/rotated_iou_loss.py:45
Method__init__
(self, filename, fast=False)
embodiedscan/converter/generate_image_scannet.py:58
Method__iter__
Yield a point as a Tensor at a time. Returns: Iterator[Tensor]: A point of shape (points_dim, ).
embodiedscan/structures/points/base_points.py:493
Method__iter__
Yield a box as a Tensor at a time. Returns: Iterator[Tensor]: A box of shape (box_dim, ).
embodiedscan/structures/bbox_3d/base_box3d.py:484
Method__len__
int: Number of boxes in the current object.
tools/eval_script_portable.py:143
Method__len__
int: The length of `PointData`.
embodiedscan/utils/typing_config.py:185
Method__len__
int: Number of boxes in the current object.
embodiedscan/structures/bbox_3d/base_box3d.py:382
Method__repr__
str: Return a string that describes the module.
embodiedscan/structures/ops/iou3d_calculator.py:50
Method__repr__
str: return a string that describes the module
embodiedscan/structures/ops/iou3d_calculator.py:92
Method__repr__
str: a string describing the module
embodiedscan/structures/ops/iou3d_calculator.py:206
Method__repr__
str: Return a string that describes the object.
embodiedscan/structures/points/base_points.py:399
Method__repr__
str: Return a string that describes the object.
embodiedscan/structures/bbox_3d/base_box3d.py:386
Method__repr__
str: Return a string that describes the module.
embodiedscan/datasets/transforms/augmentation.py:244
Method__repr__
str: Return a string that describes the module.
embodiedscan/datasets/transforms/augmentation.py:468
Method__repr__
(self)
embodiedscan/datasets/transforms/loading.py:83
Method__repr__
str: Return a string that describes the module.
embodiedscan/datasets/transforms/loading.py:507
Method__repr__
str: Return a string that describes the module.
embodiedscan/datasets/transforms/points.py:215
Method__repr__
str: Return a string that describes the module.
embodiedscan/datasets/transforms/points.py:279
Method__repr__
str: Return a string that describes the module.
embodiedscan/datasets/transforms/formatting.py:286
Method__repr__
(self)
embodiedscan/models/data_preprocessors/voxelize.py:175
Method__repr__
(self)
embodiedscan/models/data_preprocessors/voxelize.py:321
Method__repr__
(self)
embodiedscan/models/task_modules/anchor/anchor_3d_generator.py:70
Method_bbox_pred_to_bbox
Transform predicted bbox parameters to bbox. Args: points (Tensor): Final locations of shape (N, num_query, 3) bbox_p
embodiedscan/models/dense_heads/grounding_head.py:267
Method_bbox_to_loss
Transform box to the axis-aligned or rotated iou loss format. Args: bbox (Tensor): 3D box of shape (N, 6) or (N, 7). Ret
embodiedscan/models/dense_heads/fcaf3d_head.py:543
Method_bbox_to_loss
Transform box to the axis-aligned or rotated iou loss format. Args: bbox (Tensor): 3D box of shape (N, 6) or (N, 7). Ret
embodiedscan/models/dense_heads/fcaf3d_head.py:1434
Method_flip_on_direction
Function to flip images, bounding boxes, semantic segmentation map and keypoints. Add the override feature that if 'flip' is already
embodiedscan/datasets/transforms/augmentation.py:203
Method_forward
Network forward process. Usually includes backbone, neck and head forward without any post-processing. Args: batch_input
embodiedscan/models/detectors/sparse_featfusion_single_stage.py:332
Method_forward
Network forward process. Usually includes backbone, neck and head forward without any post-processing. Args: batch_inputs
embodiedscan/models/detectors/dense_fusion_occ.py:332
Method_forward
Network forward process. Usually includes backbone, neck and head forward without any post-processing. Args: batch_input
embodiedscan/models/detectors/sparse_featfusion_grounder.py:682
Method_forward
Network forward process. Usually includes backbone, neck and head forward without any post-processing. Args: batch_input
embodiedscan/models/detectors/embodied_det3d.py:318
Method_forward
Network forward process. Usually includes backbone, neck and head forward without any post-processing. Args: batch_inputs
embodiedscan/models/detectors/embodied_occ.py:320
← previousnext →301–400 of 577, ranked by callers