MCPcopy Create free account

hub / github.com/TRI-ML/dd3d / functions

Functions487 in github.com/TRI-ML/dd3d

↓ 34 callersMethodcat
(cls, boxes_list, dim=0)
tridet/structures/boxes3d.py:221
↓ 29 callersMethodto
(self, *args, **kwargs)
tridet/structures/boxes3d.py:283
↓ 23 callersFunctionget_norm
(norm, out_channels, norm_kwargs={})
tridet/layers/normalization.py:71
↓ 17 callersMethodcopy
Return a copy of this pose object. Returns ---------- result: Pose Copied pose object.
tridet/structures/pose.py:35
↓ 11 callersMethodinverse
Returns a new Pose that corresponds to the inverse of this one. Returns ---------- result: Pose Inverted
tridet/structures/pose.py:76
↓ 11 callersFunctionmake_divisible
This function is taken from the original tf repo. It ensures that all layers have a channel number that is divisible by 8 It can be seen here
tridet/modeling/backbone/omni_scripts/utils.py:33
↓ 9 callersMethod__init__
( self, inputs: Dict, merge_mode: str, post_input_op: Optional[nn.Module],
tridet/modeling/backbone/omni_scripts/ops.py:411
↓ 9 callersFunctionval2list
Repeat `val` for `repeat_time` times and return the list or val if list/tuple.
tridet/modeling/backbone/omni_scripts/utils.py:15
↓ 9 callersMethodvectorize
(self)
tridet/structures/boxes3d.py:142
↓ 8 callersMethodfrom_vectors
Parameters ---------- vecs: Iterable[np.ndarray] Iterable of 10D pose representation. intrinsics: np.nda
tridet/structures/boxes3d.py:176
↓ 8 callersMethodsplit
(self, split_sizes, dim=0)
tridet/structures/boxes3d.py:238
↓ 7 callersMethodclone
tridet/structures/boxes3d.py:276
↓ 6 callersMethod__init__
configurable is experimental. out_features (list[str]): name of the layers whose outputs should be returned in f
tridet/modeling/feature_extractor/dla.py:252
↓ 6 callersFunctiondo_test
(cfg, model, is_last=False, use_tta=False)
scripts/train.py:197
↓ 6 callersMethodfrom_tensors
Args: tensors: a tuple or list of `torch.Tensor`, each of shape (Hi, Wi) or (C_1, ..., C_K, Hi, Wi) where K >= 1.
tridet/structures/image_list.py:94
↓ 5 callersMethod__init__
Args: input_ch(int) : the number of input channel out_features (list[str]): name of the layers whose outputs should
tridet/modeling/feature_extractor/vovnet.py:277
↓ 5 callersMethodfrom_matrix
Initialize pose from 4x4 transformation matrix Parameters ---------- transformation_matrix: np.ndarray 4x4 contai
tridet/structures/pose.py:136
↓ 5 callersFunctionis_distributed
()
tridet/utils/comm.py:15
↓ 4 callersMethod__init__
(self, init_value=1.0)
tridet/layers/normalization.py:14
↓ 4 callersFunctiondiv_up
(m, n)
tridet/evaluators/rotate_iou.py:21
↓ 4 callersFunctionget_dataset_mapper
(cfg, is_train=True)
tridet/data/dataset_mappers/__init__.py:7
↓ 4 callersFunctionnuscenes_sample_aggregate
Parameters ---------- instances: List[Instances] Predicted instances. group_idxs: dict Mapping from nuScene's `sampl
tridet/modeling/dd3d/postprocessing.py:58
↓ 4 callersFunctionreduce_sum
Adapted from AdelaiDet: https://github.com/aim-uofa/AdelaiDet/blob/master/adet/utils/comm.py
tridet/utils/comm.py:91
↓ 4 callersFunctionsync_dir
Sync a directory from source to target (either local to s3, s3 to s3, s3 to local) Parameters ---------- source: str Directo
tridet/utils/s3.py:97
↓ 4 callersFunctiontrangle_area
(a, b, c)
tridet/evaluators/rotate_iou.py:26
↓ 3 callersFunction_to_tensor
(x, dim)
tridet/structures/boxes3d.py:20
↓ 3 callersFunctionbev_nms
Parameters ---------- boxes3d: Boxes3D 3D boxes in camera frame. scores: Tensor 1D score vector. Must be of same si
tridet/layers/bev_nms.py:99
↓ 3 callersFunctionbuild_dla_backbone
Parameters ---------- cfg: CfgNode [description] input_shape: ShapeSpec [description]
tridet/modeling/feature_extractor/dla.py:446
↓ 3 callersFunctionbuild_feature_extractor_all_fuse
(return_list=False, width_mult=1.0, depth_mult=1.0)
tridet/modeling/backbone/omni_scripts/backbone_with_fpn.py:12
↓ 3 callersFunctiondraw_boxes3d_bev
( boxes3d, extrinsics, class_ids, image_width, intrinsics=None, color=(0, 255, 0),
tridet/visualizers/box3d_visualizer.py:166
↓ 3 callersFunctiondraw_boxes3d_cam
Parameters ---------- img: np.ndarray RGB image. (H, W, 3) boxes3d: tridet.structures.Boxes3D or np.ndarray If np.nd
tridet/visualizers/box3d_visualizer.py:108
↓ 3 callersFunctionflatten_dict
Almost identical to detectron2.evaluation.testing:flatten_result_dict()', but using 'OrderedDict' -------------------------------------------
tridet/utils/wandb.py:85
↓ 3 callersFunctionget_group_idxs
(sample_tokens, num_images_per_sample, inverse=False)
tridet/modeling/dd3d/postprocessing.py:111
↓ 3 callersFunctionmosaic
Creates a mosaic from list of images. Parameters ---------- items: list of np.ndarray List of images to mosaic. scale: float
tridet/utils/visualization.py:111
↓ 3 callersMethodoutput_shape
(self)
tridet/modeling/feature_extractor/vovnet.py:369
↓ 3 callersFunctionproject_points3d
(Xw, K)
tridet/utils/geometry.py:77
↓ 3 callersFunctionsmooth_l1_loss
Smooth L1 loss defined in the Fast R-CNN paper as: | 0.5 * x ** 2 / beta if abs(x) < beta smoothl1(x) = |
tridet/layers/smooth_l1_loss.py:9
↓ 3 callersFunctionunproject_points2d
Parameters ---------- points2d: Tensor xy coordinates. shape=(N, ..., 2) E.g., (N, 2) or (N, K, 2) or (N, H, W, 2) i
tridet/utils/geometry.py:86
↓ 3 callersMethodvisualize
Parameters ---------- x: Dict One 'dataset_dict'. Returns ------- viz_images: Dict[np.ar
tridet/visualizers/d2_visualizer.py:109
↓ 3 callersFunctionwandb_is_initialized
()
tridet/utils/wandb.py:43
↓ 2 callersMethod__init__
(self, cfg)
tridet/modeling/dd3d/nuscenes_dd3d.py:200
↓ 2 callersMethod_format
(idx, kitti_format, is_prediction)
tridet/evaluators/kitti_3d_evaluator.py:285
↓ 2 callersMethod_make_conv_level
(self, inplanes, planes, convs, stride=1, dilation=1, norm="BN")
tridet/modeling/feature_extractor/dla.py:327
↓ 2 callersMethod_transpose
This function is used to transpose image first training targets to level first ones :return: level first training targets
tridet/modeling/dd3d/prepare_targets.py:214
↓ 2 callersFunctionaws_credential_is_available
()
tridet/utils/s3.py:53
↓ 2 callersFunctionbev_frustum_crop
(bev_vis)
tridet/visualizers/box3d_visualizer.py:225
↓ 2 callersFunctionbuild_activation
(act_func_name: Union[str, nn.Module], **kwargs)
tridet/modeling/backbone/omni_scripts/act.py:24
↓ 2 callersFunctionbuild_block
( block_str: str, in_channels: int, out_channels: int, channel_att: Optional[str], act_func: str )
tridet/modeling/backbone/omni_scripts/fpn.py:13
↓ 2 callersFunctionbuild_feature_extractor
Build a backbone from `cfg.FE.BUILDER` Returns: an instance of :class:`Backbone`
tridet/modeling/feature_extractor/__init__.py:13
↓ 2 callersMethodbuild_input
( feature_id: str, in_channels: int, extra_input: List[Tuple[str, int]], input_mode: str, mid_channe
tridet/modeling/backbone/omni_scripts/fpn.py:237
↓ 2 callersFunctionbuild_test_dataloader
(cfg, dataset_name, mapper)
tridet/data/build.py:68
↓ 2 callersFunctionbuild_train_dataloader
(cfg, mapper)
tridet/data/build.py:29
↓ 2 callersFunctionbuild_vovnet_backbone
Create a VoVNet instance from config. Returns: VoVNet: a :class:`VoVNet` instance.
tridet/modeling/feature_extractor/vovnet.py:377
↓ 2 callersFunctioncollect_dataset_dicts
Build D2 dataset (i.e. List[Dict]), given a dataset implementing recipe for building its item. This is useful when a __getitem__() takes much tim
tridet/data/build.py:163
↓ 2 callersMethodcompute_locations
(self, features)
tridet/modeling/dd3d/core.py:172
↓ 2 callersFunctionconv1x1
1x1 convolution with padding
tridet/modeling/feature_extractor/vovnet.py:144
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
tridet/modeling/feature_extractor/vovnet.py:124
↓ 2 callersFunctionconvert_3d_box_to_kitti
Convert a single 3D bounding box (GenericBoxes3D) to KITTI convention. i.e. for evaluation. We assume the box is in the reference frame of camera_
tridet/evaluators/kitti_3d_evaluator.py:205
↓ 2 callersFunctioncreate_instances
(predictions, image_size, score_threshold, metadata, score_key="score")
tridet/visualizers/d2_visualizer.py:36
↓ 2 callersMethodeval_metric
(self, gt_annos, dt_annos, metric, overlap_thresholds)
tridet/evaluators/kitti_3d_evaluator.py:444
↓ 2 callersFunctionfloat_to_uint8_color
(float_clr)
tridet/utils/visualization.py:105
↓ 2 callersFunctionget_dataloader_visualizer
(cfg, visualizer_name, dataset_name)
tridet/visualizers/__init__.py:17
↓ 2 callersFunctionget_fpn_out_channels
(output_shape)
tridet/modeling/dd3d/utils.py:2
↓ 2 callersMethodget_mAP
Get mAP from precision. Parameters ---------- precision: np.ndarray Numpy array of precision curves at different
tridet/evaluators/kitti_3d_evaluator.py:367
↓ 2 callersFunctionget_same_padding
(kernel_size: Union[int, Tuple[int, int]])
tridet/modeling/backbone/omni_scripts/utils.py:56
↓ 2 callersMethodget_sample_region
(self, boxes, num_loc_list, loc_xs, loc_ys)
tridet/modeling/dd3d/prepare_targets.py:179
↓ 2 callersMethodget_shards
Shard number into evenly sized parts. `Remaining` values are put into the last shard. Parameters ---------- num: int
tridet/evaluators/kitti_3d_evaluator.py:320
↓ 2 callersMethodget_transform
(self, image)
tridet/data/augmentations/color_transform.py:29
↓ 2 callersFunctionkaiming_normal_groups_
torch.nn.init.kaiming_normal_()' with 'groups'. If 'mode=="fan_out"', fan is divided by 'groups', yielding larger std of weights.
tridet/layers/separable_conv2d.py:34
↓ 2 callersMethodnms_and_top_k
(self, instances_per_im, score_key_for_nms="scores")
tridet/modeling/dd3d/fcos2d.py:346
↓ 2 callersFunctionpoint_in_quadrilateral
(pt_x, pt_y, corners)
tridet/evaluators/rotate_iou.py:161
↓ 2 callersFunctionpredictions_to_boxes3d
( quat, proj_ctr, depth, size, locations, inv_intrinsics, canon_box_sizes, min
tridet/modeling/dd3d/fcos3d.py:16
↓ 2 callersMethodpreprocess_image
(self, x)
tridet/modeling/dd3d/core.py:67
↓ 2 callersFunctionrbbox_to_corners
(corners, rbbox)
tridet/evaluators/rotate_iou.py:205
↓ 2 callersFunctionregister_datasets
(cfg)
tridet/data/datasets/__init__.py:12
↓ 2 callersFunctionrotate_iou_gpu_eval
rotated box iou running in gpu. 500x faster than cpu version (take 5ms in one example with numba.cuda code). convert from [this project](
tridet/evaluators/rotate_iou.py:292
↓ 2 callersFunctions3_copy
Copy single file from local to s3, s3 to local, or s3 to s3. Parameters ---------- source_path: str Path of file to copy tar
tridet/utils/s3.py:61
↓ 2 callersFunctionsave_vis
(np_arrays_dict, output_dir, filename, step=None)
tridet/utils/visualization.py:25
↓ 2 callersFunctionsetup
(cfg)
tridet/utils/setup.py:70
↓ 2 callersFunctionsync_output_dir_s3
(cfg, output_dir=None)
tridet/utils/s3.py:128
↓ 2 callersFunctionwandb_credential_is_available
()
tridet/utils/wandb.py:17
↓ 1 callersMethod__init__
(self, cfg)
tridet/modeling/dd3d/dense_depth.py:105
↓ 1 callersMethod_add_feat
(feat: list)
tridet/modeling/backbone/omni_scripts/ops.py:452
↓ 1 callersFunction_as_tensor
An equivalent of `torch.as_tensor`, but works under tracing if input is a list of tensor. `torch.as_tensor` will record a constant in tracing
tridet/structures/image_list.py:14
↓ 1 callersMethod_batch_inference
Execute inference on a list of inputs, using batch size = self.batch_size, instead of the length of the list. Inputs & outpu
tridet/modeling/dd3d/test_time_augmentation.py:118
↓ 1 callersMethod_batch_inference
Execute inference on a list of inputs, using batch size = self.batch_size, instead of the length of the list. Inputs & outpu
tridet/modeling/dd3d/nuscenes_dd3d_tta.py:163
↓ 1 callersMethod_build_dataset_item_info
(self, name)
tridet/data/datasets/nuscenes/build.py:115
↓ 1 callersMethod_build_id
(self, scene_name, sample_idx, datum_name)
tridet/data/datasets/nuscenes/build.py:154
↓ 1 callersMethod_cat_feat
(feat: list)
tridet/modeling/backbone/omni_scripts/ops.py:448
↓ 1 callersMethod_compute_box2d_from_box3d
(self, box3d, K)
tridet/data/datasets/kitti_3d/build.py:272
↓ 1 callersFunction_compute_iou
Parameters ---------- box1, box2: (x1, y1, x2, y2)
tridet/data/datasets/nuscenes/build.py:80
↓ 1 callersFunction_create_empty_instances
(image_size)
tridet/data/transform_utils.py:73
↓ 1 callersMethod_get_2d_annotation
(self, label)
tridet/data/datasets/kitti_3d/build.py:268
↓ 1 callersMethod_get_3d_annotation
Convert KITTI annotation data frame to 3D bounding box annotations. Labels are provided in the reference frame of camera_2. NOTE: Anno
tridet/data/datasets/kitti_3d/build.py:239
↓ 1 callersMethod_get_augmented_inputs
(self, input)
tridet/modeling/dd3d/test_time_augmentation.py:185
↓ 1 callersMethod_get_augmented_inputs
(self, x)
tridet/modeling/dd3d/nuscenes_dd3d_tta.py:108
↓ 1 callersMethod_get_augmented_instances
(self, augmented_inputs, tfms, orig_shape)
tridet/modeling/dd3d/test_time_augmentation.py:190
↓ 1 callersMethod_get_augmented_instances
(self, augmented_inputs, tfms, orig_shape)
tridet/modeling/dd3d/nuscenes_dd3d_tta.py:113
↓ 1 callersMethod_get_ego_velocity
Velocity of ego-vehicle in m/s.
tridet/data/datasets/nuscenes/build.py:245
↓ 1 callersMethod_get_sample_data
(self, sample_id, sensor_name)
tridet/data/datasets/kitti_3d/build.py:169
next →1–100 of 487, ranked by callers