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
get_feature_dims
(self, feature_id_list: Union[str, List[str]])
tridet/modeling/backbone/omni_scripts/ops.py:504
Function
get_num_groups_gn
(out_channels)
tridet/modeling/feature_extractor/dla.py:17
Method
get_transform
(self, image)
tridet/data/augmentations/color_transform.py:35
Method
get_transform
(self, image)
tridet/data/augmentations/color_transform.py:41
Method
get_transform
(self, image)
tridet/data/augmentations/resize_transform.py:86
Method
get_transform
(self, image)
tridet/data/augmentations/flip_transform.py:72
Method
get_transform
(self, image)
tridet/data/augmentations/crop_transform.py:38
Method
groups
(self)
tridet/modeling/backbone/omni_scripts/ops.py:67
Method
groups
(self)
tridet/modeling/backbone/omni_scripts/ops.py:381
Method
has_dense_prediction_task
(self)
tridet/utils/tasks.py:88
Method
has_detection_task
(self)
tridet/utils/tasks.py:92
Function
if_wandb_initialized
(fn)
tridet/utils/wandb.py:52
Function
image_box_overlap
(boxes, query_boxes, criterion=-1)
tridet/evaluators/rotate_iou.py:361
Method
image_paths
(self)
tridet/structures/image_list.py:65
Method
in_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:47
Method
in_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:169
Method
in_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:247
Method
in_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:299
Method
in_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:357
Method
intrinsics
(self)
tridet/structures/image_list.py:58
Function
kaiming_uniform_groups_
torch.nn.init.kaiming_uniform_()' with 'groups'. If 'mode=="fan_out"', fan is divided by 'groups', yielding larger std of weights.
tridet/layers/separable_conv2d.py:17
Method
kernel_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:55
Method
kernel_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:181
Method
kernel_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:255
Method
kernel_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:307
Method
kernel_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:365
Function
line_segment_intersection_v1
(pts1, pts2, i, j, temp_pts)
tridet/evaluators/rotate_iou.py:122
Function
master_only
If distributed, only the master executes the function. Usage: @master_only def foo(a, b): ...
tridet/utils/comm.py:53
Method
matrix
Returns a 4x4 homogeneous matrix of the form [R t; 0 1] Returns ---------- result: np.ndarray 4x4 homogeneous mat
tridet/structures/pose.py:89
Function
maybe_download_from_s3
(src_path)
tridet/utils/s3.py:152
Function
maybe_sync_dir_from_s3
(src_path, excludes=None)
tridet/utils/s3.py:166
Method
mid_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:315
Method
mid_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:373
Method
n_extra_stage
(self)
tridet/modeling/backbone/omni_scripts/backbone_with_fpn.py:77
Method
n_stages
(self)
tridet/modeling/backbone/omni_scripts/ops.py:472
Function
neg_box_distance
(loc, query_loc, criterion=-1, device_id=0)
tridet/evaluators/rotate_iou.py:415
Method
on_job_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:27
Method
on_run_end
tridet/utils/hydra/callbacks.py:100
Method
on_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:20
Method
on_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:39
Method
on_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:57
Method
on_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:70
Method
on_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:86
Method
on_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:96
Method
out_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:51
Method
out_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:251
Method
out_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:303
Method
out_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:361
Method
output_shape
(self)
tridet/layers/conv_bn_fpn_layers.py:69
Method
pool_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:173
Method
pool_type
(self)
tridet/modeling/backbone/omni_scripts/ops.py:177
Method
process
Process the pair of inputs and outputs. If they contain batches, the pairs can be consumed one-by-one using `zip`: .. code-b
tridet/evaluators/kitti_3d_evaluator.py:66
Method
process
Process the pair of inputs and outputs. If they contain batches, the pairs can be consumed one-by-one using `zip`: .. code-b
tridet/evaluators/nuscenes_evaluator.py:147
Function
register_kitti_3d_metadata
(dataset_name, valid_class_names=VALID_CLASS_NAMES, coco_cache_dir='/tmp/')
tridet/data/datasets/kitti_3d/build.py:303
Function
register_nuscenes_metadata
(dataset_name)
tridet/data/datasets/nuscenes/build.py:331
Method
render_radar_point_cloud
Render radar point cloud in BEV perspective. Parameters ---------- radar_point_cloud: numpy array with shape (N, 3)
tridet/visualizers/bev.py:143
Method
reset
Preparation for a new round of evaluation. Should be called before starting a round of evaluation.
tridet/evaluators/kitti_3d_evaluator.py:55
Method
reset
Preparation for a new round of evaluation. Should be called before starting a round of evaluation.
tridet/evaluators/nuscenes_evaluator.py:138
Function
resize_depth_preserve
Adapted from: https://github.com/TRI-ML/packnet-sfm_internal/blob/919ab604ae2319e4554d3b588877acfddf877f9c/packnet_sfm/datasets/augmentat
tridet/data/augmentations/resize_transform.py:31
Function
rotate_iou_kernel_eval
(N, K, dev_boxes, dev_query_boxes, dev_iou, criterion=-1)
tridet/evaluators/rotate_iou.py:261
Method
rotation
Return the rotation component of the pose as a Quaternion object. Returns ---------- self.quat: Quaternion Rotati
tridet/structures/pose.py:114
Method
rotation_matrix
Returns the 3x3 rotation matrix (R) Returns ---------- result: np.ndarray 3x3 rotation matrix
tridet/structures/pose.py:102
Method
size_divisibility
(self)
tridet/modeling/backbone/omni_scripts/backbone_with_fpn.py:81
Method
size_divisibility
(self)
tridet/modeling/feature_extractor/dla.py:318
Method
split
(self, split_sizes, dim=0)
tridet/structures/boxes3d.py:109
Method
stage_depth_info
(self)
tridet/modeling/backbone/omni_scripts/ops.py:476
Method
stage_width_info
(self)
tridet/modeling/backbone/omni_scripts/ops.py:487
Method
stride
(self)
tridet/modeling/backbone/omni_scripts/ops.py:59
Method
stride
(self)
tridet/modeling/backbone/omni_scripts/ops.py:259
Method
stride
(self)
tridet/modeling/backbone/omni_scripts/ops.py:311
Method
stride
(self)
tridet/modeling/backbone/omni_scripts/ops.py:369
Method
task_names
(self)
tridet/utils/tasks.py:96
Method
tasks
(self)
tridet/utils/tasks.py:61
Method
to
(self, *args, **kwargs)
tridet/structures/boxes3d.py:150
Method
translation
Return the translation component of the pose as a np.ndarray. Returns ---------- self.tvec: np.ndarray Translatio
tridet/structures/pose.py:125
Function
trivial_batch_collator
A batch collator that does nothing.
tridet/data/build.py:152
Method
tvec
(self)
tridet/structures/boxes3d.py:44
Method
tvec
(self)
tridet/structures/boxes3d.py:170
Method
use_bias
(self)
tridet/modeling/backbone/omni_scripts/ops.py:71
Method
visualize
(self, x)
tridet/visualizers/d2_visualizer.py:173
Method
visualize
Parameters ---------- x: Dict One 'dataset_dict'. Returns ------- viz_images: Dict[np.a
tridet/visualizers/box3d_visualizer.py:267
Method
visualize
(self, x)
tridet/visualizers/box3d_visualizer.py:358
Function
worker_init_reset_seed
(worker_id)
tridet/data/build.py:159
Function
wrapped_fn
(*args, **kwargs)
tridet/utils/wandb.py:54
Function
wrapped_fn
(*args, **kwargs)
tridet/utils/comm.py:61
Function
wrapper
(*args, **kwargs)
tridet/utils/comm.py:27
Method
zero_last_gamma
(self)
tridet/modeling/backbone/omni_scripts/ops.py:525
← previous
401–487 of 487, ranked by callers