Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/TRI-ML/dd3d
/ functions
Functions
487 in github.com/TRI-ML/dd3d
⨍
Functions
487
◇
Types & classes
89
Method
_read_calibration_file
Reads a calibration file and creates corresponding pose, camera calibration objects. Reference frame (world frame) is camera 0. Retur
tridet/data/datasets/kitti_3d/build.py:101
Function
apply_hflip_box3d
Horizontally flip 3D box. CAVEAT: This function makes assumption about the object symmetry wrt *y=0* plane. new quaternion: [quat.z, -quat.y
tridet/data/augmentations/flip_transform.py:28
Function
apply_hflip_depth
(hflip_tfm, depth)
tridet/data/augmentations/flip_transform.py:18
Function
apply_hflip_intrinsics
(hflip_tfm, intrinsics)
tridet/data/augmentations/flip_transform.py:8
Function
apply_imcrop_box3d
(crop_tfm, box3d)
tridet/data/augmentations/crop_transform.py:27
Function
apply_imcrop_depth
(crop_tfm, depth)
tridet/data/augmentations/crop_transform.py:21
Function
apply_imcrop_intrinsics
(crop_tfm, intrinsics)
tridet/data/augmentations/crop_transform.py:8
Function
apply_imresize_box3d
(resize_tfm, box3d)
tridet/data/augmentations/resize_transform.py:74
Function
apply_imresize_depth
(resize_tfm, depth)
tridet/data/augmentations/resize_transform.py:24
Function
apply_imresize_intrinsics
(resize_tfm, intrinsics)
tridet/data/augmentations/resize_transform.py:13
Function
apply_no_op_box3d
(blend_tfm, box3d)
tridet/data/augmentations/color_transform.py:18
Function
apply_no_op_depth
(blend_tfm, depth)
tridet/data/augmentations/color_transform.py:14
Function
apply_no_op_intrinsics
(blend_tfm, intrinsics)
tridet/data/augmentations/color_transform.py:10
Function
apply_vflip_box3d
(vflip_tfm, box3d)
tridet/data/augmentations/flip_transform.py:57
Function
apply_vflip_depth
(vflip_tfm, depth)
tridet/data/augmentations/flip_transform.py:23
Function
apply_vflip_intrinsics
(vflip_tfm, intrinsics)
tridet/data/augmentations/flip_transform.py:13
Method
blocks
(self)
tridet/modeling/backbone/omni_scripts/ops.py:516
Method
box2d_on
(self)
tridet/utils/tasks.py:76
Method
box3d_on
(self)
tridet/utils/tasks.py:80
Function
broadcast_from_master
If distributed, only the master executes the function and broadcast the results to other workers. Usage: @broadcast_from_master def foo(a
tridet/utils/comm.py:19
Function
build_dla_fpn_backbone
Args: cfg: a detectron2 CfgNode Returns: backbone (Backbone): backbone module, must be a subclass of :class:`Backbone`.
tridet/modeling/feature_extractor/dla.py:463
Function
build_fcos_dla_fpn_backbone_p6
Args: cfg: a detectron2 CfgNode Returns: backbone (Backbone): backbone module, must be a subclass of :class:`Backbone`.
tridet/modeling/feature_extractor/dla.py:509
Function
build_fcos_dla_fpn_backbone_p67
Args: cfg: a detectron2 CfgNode Returns: backbone (Backbone): backbone module, must be a subclass of :class:`Backbone`.
tridet/modeling/feature_extractor/dla.py:537
Function
build_fcos_vovnet_fpn_backbone_p6
Args: cfg: a detectron2 CfgNode Returns: backbone (Backbone): backbone module, must be a subclass of :class:`Backbone`.
tridet/modeling/feature_extractor/vovnet.py:429
Function
build_monocular_kitti3d_dataset
( mv3d_split, root_dir, class_names=VALID_CLASS_NAMES, sensors=('camera_2', ), box2d_from_box3d=False, max
tridet/data/datasets/kitti_3d/build.py:295
Function
build_nuscenes_dataset
(name, root_dir, min_num_lidar_points=3, min_box_visibility=0.2)
tridet/data/datasets/nuscenes/build.py:323
Function
build_vovnet_fpn_backbone
Args: cfg: a detectron2 CfgNode Returns: backbone (Backbone): backbone module, must be a subclass of :class:`Backbone`.
tridet/modeling/feature_extractor/vovnet.py:389
Method
cat
(cls, boxes_list, dim=0)
tridet/structures/boxes3d.py:94
Function
clean_kitti_data
Function for filtering KITTI data by difficulty and class. We filter with the following heuristics: If a ground truth matches the current
tridet/evaluators/kitti_3d_evaluator.py:635
Method
clip_norm
(v, x)
tridet/visualizers/bev.py:184
Method
clone
tridet/structures/boxes3d.py:137
Function
compute_features_locations
Adapted from AdelaiDet: https://github.com/aim-uofa/AdelaiDet/blob/master/adet/utils/comm.py Key differnece: offset is configurable.
tridet/utils/tensor2d.py:6
Method
corners
(self)
tridet/structures/boxes3d.py:49
Method
depth_on
(self)
tridet/utils/tasks.py:84
Method
device
(self)
tridet/modeling/dd3d/dense_depth.py:124
Method
device
(self)
tridet/modeling/dd3d/core.py:64
Method
device
(self)
tridet/structures/image_list.py:90
Method
device
(self)
tridet/structures/boxes3d.py:147
Method
dilation
(self)
tridet/modeling/backbone/omni_scripts/ops.py:63
Function
distance_kernel_eval
(N, K, dev_loc, dev_query_loc, dev_distances, criterion=-1)
tridet/evaluators/rotate_iou.py:390
Function
dla102
(cfg)
tridet/modeling/feature_extractor/dla.py:394
Function
dla102x
(cfg)
tridet/modeling/feature_extractor/dla.py:402
Function
dla102x2
(cfg)
tridet/modeling/feature_extractor/dla.py:410
Function
dla169
(cfg)
tridet/modeling/feature_extractor/dla.py:418
Function
dla34
(cfg)
tridet/modeling/feature_extractor/dla.py:359
Function
dla46_c
(cfg)
tridet/modeling/feature_extractor/dla.py:364
Function
dla46x_c
(cfg)
tridet/modeling/feature_extractor/dla.py:370
Function
dla60
(cfg)
tridet/modeling/feature_extractor/dla.py:382
Function
dla60x
(cfg)
tridet/modeling/feature_extractor/dla.py:388
Function
dla60x_c
(cfg)
tridet/modeling/feature_extractor/dla.py:376
Method
evaluate
(self)
tridet/evaluators/kitti_3d_evaluator.py:150
Method
evaluate
(self)
tridet/evaluators/nuscenes_evaluator.py:249
Method
expand_ratio
(self)
tridet/modeling/backbone/omni_scripts/ops.py:319
Method
expand_ratio
(self)
tridet/modeling/backbone/omni_scripts/ops.py:377
Method
forward
(self, x)
tridet/modeling/backbone/omni_scripts/backbone_with_fpn.py:95
Method
forward
(self, feature_dict: Dict[str, torch.Tensor])
tridet/modeling/backbone/omni_scripts/fpn.py:266
Method
forward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:74
Method
forward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:132
Method
forward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:184
Method
forward
(self, x: torch.Tensor)
tridet/modeling/backbone/omni_scripts/ops.py:207
Method
forward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:262
Method
forward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:322
Method
forward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:384
Method
forward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:398
Method
forward
(self, feature_dict: dict)
tridet/modeling/backbone/omni_scripts/ops.py:431
Method
forward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:535
Method
forward
(self, x, residual=None)
tridet/modeling/feature_extractor/dla.py:50
Method
forward
(self, x, residual=None)
tridet/modeling/feature_extractor/dla.py:86
Method
forward
(self, x, residual=None)
tridet/modeling/feature_extractor/dla.py:128
Method
forward
(self, *x)
tridet/modeling/feature_extractor/dla.py:160
Method
forward
(self, x, residual=None, children=None)
tridet/modeling/feature_extractor/dla.py:233
Method
forward
(self, x)
tridet/modeling/feature_extractor/dla.py:346
Method
forward
(self, c5)
tridet/modeling/feature_extractor/dla.py:501
Method
forward
(self, x)
tridet/modeling/feature_extractor/vovnet.py:169
Method
forward
(self, x)
tridet/modeling/feature_extractor/vovnet.py:180
Method
forward
(self, x)
tridet/modeling/feature_extractor/vovnet.py:218
Method
forward
(self, x)
tridet/modeling/feature_extractor/vovnet.py:357
Method
forward
(self, x)
tridet/modeling/feature_extractor/vovnet.py:423
Method
forward
(self, x, cams)
tridet/modeling/dd3d/depth.py:41
Method
forward
(self, x)
tridet/modeling/dd3d/dense_depth.py:91
Method
forward
(self, batched_inputs)
tridet/modeling/dd3d/dense_depth.py:130
Method
forward
(self, batched_inputs)
tridet/modeling/dd3d/core.py:70
Method
forward
(self, batched_inputs)
tridet/modeling/dd3d/nuscenes_dd3d.py:337
Method
forward
(self, depth_pred, depth_gt, masks=None)
tridet/modeling/dd3d/dense_depth_loss.py:28
Method
forward
(self, x)
tridet/modeling/dd3d/fcos2d.py:130
Method
forward
(self, x)
tridet/modeling/dd3d/fcos3d.py:160
Method
forward
(self, x)
tridet/data/build.py:182
Method
forward
(self, x)
tridet/layers/separable_conv2d.py:104
Method
forward
(self, x: List[torch.Tensor])
tridet/layers/conv_bn_fpn_layers.py:79
Method
forward
Args: pred: Nx4 predicted bounding boxes target: Nx4 target bounding boxes weight: N loss weight for each
tridet/layers/iou_loss.py:21
Method
forward
(self, input)
tridet/layers/normalization.py:18
Method
forward
(self, input)
tridet/layers/normalization.py:27
Method
forward
(self, x)
tridet/layers/normalization.py:36
Method
forward
(self, x)
tridet/layers/normalization.py:52
Method
from_config
(cls, cfg, levels, channels, block)
tridet/modeling/feature_extractor/dla.py:322
Method
from_config
(cls, cfg)
tridet/modeling/dd3d/disentangled_box3d_loss.py:20
Method
from_config
(cls, cfg)
tridet/modeling/dd3d/dense_depth_loss.py:20
Method
from_config
(cls, cfg)
tridet/utils/tasks.py:65
Method
from_config
(cls, cfg, is_train: bool = True)
tridet/data/dataset_mappers/dataset_mapper.py:88
Method
from_vectors
Parameters ---------- vecs: Iterable[np.ndarray] Iterable of 10D pose representation. intrinsics: np.nda
tridet/structures/boxes3d.py:67
← previous
next →
301–400 of 487, ranked by callers