MCPcopy Create free account

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

Functions487 in github.com/TRI-ML/dd3d

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
Functionapply_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
Functionapply_hflip_depth
(hflip_tfm, depth)
tridet/data/augmentations/flip_transform.py:18
Functionapply_hflip_intrinsics
(hflip_tfm, intrinsics)
tridet/data/augmentations/flip_transform.py:8
Functionapply_imcrop_box3d
(crop_tfm, box3d)
tridet/data/augmentations/crop_transform.py:27
Functionapply_imcrop_depth
(crop_tfm, depth)
tridet/data/augmentations/crop_transform.py:21
Functionapply_imcrop_intrinsics
(crop_tfm, intrinsics)
tridet/data/augmentations/crop_transform.py:8
Functionapply_imresize_box3d
(resize_tfm, box3d)
tridet/data/augmentations/resize_transform.py:74
Functionapply_imresize_depth
(resize_tfm, depth)
tridet/data/augmentations/resize_transform.py:24
Functionapply_imresize_intrinsics
(resize_tfm, intrinsics)
tridet/data/augmentations/resize_transform.py:13
Functionapply_no_op_box3d
(blend_tfm, box3d)
tridet/data/augmentations/color_transform.py:18
Functionapply_no_op_depth
(blend_tfm, depth)
tridet/data/augmentations/color_transform.py:14
Functionapply_no_op_intrinsics
(blend_tfm, intrinsics)
tridet/data/augmentations/color_transform.py:10
Functionapply_vflip_box3d
(vflip_tfm, box3d)
tridet/data/augmentations/flip_transform.py:57
Functionapply_vflip_depth
(vflip_tfm, depth)
tridet/data/augmentations/flip_transform.py:23
Functionapply_vflip_intrinsics
(vflip_tfm, intrinsics)
tridet/data/augmentations/flip_transform.py:13
Methodblocks
(self)
tridet/modeling/backbone/omni_scripts/ops.py:516
Methodbox2d_on
(self)
tridet/utils/tasks.py:76
Methodbox3d_on
(self)
tridet/utils/tasks.py:80
Functionbroadcast_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
Functionbuild_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
Functionbuild_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
Functionbuild_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
Functionbuild_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
Functionbuild_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
Functionbuild_nuscenes_dataset
(name, root_dir, min_num_lidar_points=3, min_box_visibility=0.2)
tridet/data/datasets/nuscenes/build.py:323
Functionbuild_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
Methodcat
(cls, boxes_list, dim=0)
tridet/structures/boxes3d.py:94
Functionclean_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
Methodclip_norm
(v, x)
tridet/visualizers/bev.py:184
Methodclone
tridet/structures/boxes3d.py:137
Functioncompute_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
Methodcorners
(self)
tridet/structures/boxes3d.py:49
Methoddepth_on
(self)
tridet/utils/tasks.py:84
Methoddevice
(self)
tridet/modeling/dd3d/dense_depth.py:124
Methoddevice
(self)
tridet/modeling/dd3d/core.py:64
Methoddevice
(self)
tridet/structures/image_list.py:90
Methoddevice
(self)
tridet/structures/boxes3d.py:147
Methoddilation
(self)
tridet/modeling/backbone/omni_scripts/ops.py:63
Functiondistance_kernel_eval
(N, K, dev_loc, dev_query_loc, dev_distances, criterion=-1)
tridet/evaluators/rotate_iou.py:390
Functiondla102
(cfg)
tridet/modeling/feature_extractor/dla.py:394
Functiondla102x
(cfg)
tridet/modeling/feature_extractor/dla.py:402
Functiondla102x2
(cfg)
tridet/modeling/feature_extractor/dla.py:410
Functiondla169
(cfg)
tridet/modeling/feature_extractor/dla.py:418
Functiondla34
(cfg)
tridet/modeling/feature_extractor/dla.py:359
Functiondla46_c
(cfg)
tridet/modeling/feature_extractor/dla.py:364
Functiondla46x_c
(cfg)
tridet/modeling/feature_extractor/dla.py:370
Functiondla60
(cfg)
tridet/modeling/feature_extractor/dla.py:382
Functiondla60x
(cfg)
tridet/modeling/feature_extractor/dla.py:388
Functiondla60x_c
(cfg)
tridet/modeling/feature_extractor/dla.py:376
Methodevaluate
(self)
tridet/evaluators/kitti_3d_evaluator.py:150
Methodevaluate
(self)
tridet/evaluators/nuscenes_evaluator.py:249
Methodexpand_ratio
(self)
tridet/modeling/backbone/omni_scripts/ops.py:319
Methodexpand_ratio
(self)
tridet/modeling/backbone/omni_scripts/ops.py:377
Methodforward
(self, x)
tridet/modeling/backbone/omni_scripts/backbone_with_fpn.py:95
Methodforward
(self, feature_dict: Dict[str, torch.Tensor])
tridet/modeling/backbone/omni_scripts/fpn.py:266
Methodforward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:74
Methodforward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:132
Methodforward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:184
Methodforward
(self, x: torch.Tensor)
tridet/modeling/backbone/omni_scripts/ops.py:207
Methodforward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:262
Methodforward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:322
Methodforward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:384
Methodforward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:398
Methodforward
(self, feature_dict: dict)
tridet/modeling/backbone/omni_scripts/ops.py:431
Methodforward
(self, x)
tridet/modeling/backbone/omni_scripts/ops.py:535
Methodforward
(self, x, residual=None)
tridet/modeling/feature_extractor/dla.py:50
Methodforward
(self, x, residual=None)
tridet/modeling/feature_extractor/dla.py:86
Methodforward
(self, x, residual=None)
tridet/modeling/feature_extractor/dla.py:128
Methodforward
(self, *x)
tridet/modeling/feature_extractor/dla.py:160
Methodforward
(self, x, residual=None, children=None)
tridet/modeling/feature_extractor/dla.py:233
Methodforward
(self, x)
tridet/modeling/feature_extractor/dla.py:346
Methodforward
(self, c5)
tridet/modeling/feature_extractor/dla.py:501
Methodforward
(self, x)
tridet/modeling/feature_extractor/vovnet.py:169
Methodforward
(self, x)
tridet/modeling/feature_extractor/vovnet.py:180
Methodforward
(self, x)
tridet/modeling/feature_extractor/vovnet.py:218
Methodforward
(self, x)
tridet/modeling/feature_extractor/vovnet.py:357
Methodforward
(self, x)
tridet/modeling/feature_extractor/vovnet.py:423
Methodforward
(self, x, cams)
tridet/modeling/dd3d/depth.py:41
Methodforward
(self, x)
tridet/modeling/dd3d/dense_depth.py:91
Methodforward
(self, batched_inputs)
tridet/modeling/dd3d/dense_depth.py:130
Methodforward
(self, batched_inputs)
tridet/modeling/dd3d/core.py:70
Methodforward
(self, batched_inputs)
tridet/modeling/dd3d/nuscenes_dd3d.py:337
Methodforward
(self, depth_pred, depth_gt, masks=None)
tridet/modeling/dd3d/dense_depth_loss.py:28
Methodforward
(self, x)
tridet/modeling/dd3d/fcos2d.py:130
Methodforward
(self, x)
tridet/modeling/dd3d/fcos3d.py:160
Methodforward
(self, x)
tridet/data/build.py:182
Methodforward
(self, x)
tridet/layers/separable_conv2d.py:104
Methodforward
(self, x: List[torch.Tensor])
tridet/layers/conv_bn_fpn_layers.py:79
Methodforward
Args: pred: Nx4 predicted bounding boxes target: Nx4 target bounding boxes weight: N loss weight for each
tridet/layers/iou_loss.py:21
Methodforward
(self, input)
tridet/layers/normalization.py:18
Methodforward
(self, input)
tridet/layers/normalization.py:27
Methodforward
(self, x)
tridet/layers/normalization.py:36
Methodforward
(self, x)
tridet/layers/normalization.py:52
Methodfrom_config
(cls, cfg, levels, channels, block)
tridet/modeling/feature_extractor/dla.py:322
Methodfrom_config
(cls, cfg)
tridet/modeling/dd3d/disentangled_box3d_loss.py:20
Methodfrom_config
(cls, cfg)
tridet/modeling/dd3d/dense_depth_loss.py:20
Methodfrom_config
(cls, cfg)
tridet/utils/tasks.py:65
Methodfrom_config
(cls, cfg, is_train: bool = True)
tridet/data/dataset_mappers/dataset_mapper.py:88
Methodfrom_vectors
Parameters ---------- vecs: Iterable[np.ndarray] Iterable of 10D pose representation. intrinsics: np.nda
tridet/structures/boxes3d.py:67
← previousnext →301–400 of 487, ranked by callers