MCPcopy Create free account

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

Functions487 in github.com/TRI-ML/dd3d

Methodget_feature_dims
(self, feature_id_list: Union[str, List[str]])
tridet/modeling/backbone/omni_scripts/ops.py:504
Functionget_num_groups_gn
(out_channels)
tridet/modeling/feature_extractor/dla.py:17
Methodget_transform
(self, image)
tridet/data/augmentations/color_transform.py:35
Methodget_transform
(self, image)
tridet/data/augmentations/color_transform.py:41
Methodget_transform
(self, image)
tridet/data/augmentations/resize_transform.py:86
Methodget_transform
(self, image)
tridet/data/augmentations/flip_transform.py:72
Methodget_transform
(self, image)
tridet/data/augmentations/crop_transform.py:38
Methodgroups
(self)
tridet/modeling/backbone/omni_scripts/ops.py:67
Methodgroups
(self)
tridet/modeling/backbone/omni_scripts/ops.py:381
Methodhas_dense_prediction_task
(self)
tridet/utils/tasks.py:88
Methodhas_detection_task
(self)
tridet/utils/tasks.py:92
Functionif_wandb_initialized
(fn)
tridet/utils/wandb.py:52
Functionimage_box_overlap
(boxes, query_boxes, criterion=-1)
tridet/evaluators/rotate_iou.py:361
Methodimage_paths
(self)
tridet/structures/image_list.py:65
Methodin_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:47
Methodin_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:169
Methodin_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:247
Methodin_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:299
Methodin_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:357
Methodintrinsics
(self)
tridet/structures/image_list.py:58
Functionkaiming_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
Methodkernel_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:55
Methodkernel_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:181
Methodkernel_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:255
Methodkernel_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:307
Methodkernel_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:365
Functionline_segment_intersection_v1
(pts1, pts2, i, j, temp_pts)
tridet/evaluators/rotate_iou.py:122
Functionmaster_only
If distributed, only the master executes the function. Usage: @master_only def foo(a, b): ...
tridet/utils/comm.py:53
Methodmatrix
Returns a 4x4 homogeneous matrix of the form [R t; 0 1] Returns ---------- result: np.ndarray 4x4 homogeneous mat
tridet/structures/pose.py:89
Functionmaybe_download_from_s3
(src_path)
tridet/utils/s3.py:152
Functionmaybe_sync_dir_from_s3
(src_path, excludes=None)
tridet/utils/s3.py:166
Methodmid_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:315
Methodmid_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:373
Methodn_extra_stage
(self)
tridet/modeling/backbone/omni_scripts/backbone_with_fpn.py:77
Methodn_stages
(self)
tridet/modeling/backbone/omni_scripts/ops.py:472
Functionneg_box_distance
(loc, query_loc, criterion=-1, device_id=0)
tridet/evaluators/rotate_iou.py:415
Methodon_job_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:27
Methodon_run_end
tridet/utils/hydra/callbacks.py:100
Methodon_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:20
Methodon_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:39
Methodon_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:57
Methodon_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:70
Methodon_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:86
Methodon_run_start
(self, config, **kwargs)
tridet/utils/hydra/callbacks.py:96
Methodout_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:51
Methodout_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:251
Methodout_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:303
Methodout_channels
(self)
tridet/modeling/backbone/omni_scripts/ops.py:361
Methodoutput_shape
(self)
tridet/layers/conv_bn_fpn_layers.py:69
Methodpool_size
(self)
tridet/modeling/backbone/omni_scripts/ops.py:173
Methodpool_type
(self)
tridet/modeling/backbone/omni_scripts/ops.py:177
Methodprocess
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
Methodprocess
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
Functionregister_kitti_3d_metadata
(dataset_name, valid_class_names=VALID_CLASS_NAMES, coco_cache_dir='/tmp/')
tridet/data/datasets/kitti_3d/build.py:303
Functionregister_nuscenes_metadata
(dataset_name)
tridet/data/datasets/nuscenes/build.py:331
Methodrender_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
Methodreset
Preparation for a new round of evaluation. Should be called before starting a round of evaluation.
tridet/evaluators/kitti_3d_evaluator.py:55
Methodreset
Preparation for a new round of evaluation. Should be called before starting a round of evaluation.
tridet/evaluators/nuscenes_evaluator.py:138
Functionresize_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
Functionrotate_iou_kernel_eval
(N, K, dev_boxes, dev_query_boxes, dev_iou, criterion=-1)
tridet/evaluators/rotate_iou.py:261
Methodrotation
Return the rotation component of the pose as a Quaternion object. Returns ---------- self.quat: Quaternion Rotati
tridet/structures/pose.py:114
Methodrotation_matrix
Returns the 3x3 rotation matrix (R) Returns ---------- result: np.ndarray 3x3 rotation matrix
tridet/structures/pose.py:102
Methodsize_divisibility
(self)
tridet/modeling/backbone/omni_scripts/backbone_with_fpn.py:81
Methodsize_divisibility
(self)
tridet/modeling/feature_extractor/dla.py:318
Methodsplit
(self, split_sizes, dim=0)
tridet/structures/boxes3d.py:109
Methodstage_depth_info
(self)
tridet/modeling/backbone/omni_scripts/ops.py:476
Methodstage_width_info
(self)
tridet/modeling/backbone/omni_scripts/ops.py:487
Methodstride
(self)
tridet/modeling/backbone/omni_scripts/ops.py:59
Methodstride
(self)
tridet/modeling/backbone/omni_scripts/ops.py:259
Methodstride
(self)
tridet/modeling/backbone/omni_scripts/ops.py:311
Methodstride
(self)
tridet/modeling/backbone/omni_scripts/ops.py:369
Methodtask_names
(self)
tridet/utils/tasks.py:96
Methodtasks
(self)
tridet/utils/tasks.py:61
Methodto
(self, *args, **kwargs)
tridet/structures/boxes3d.py:150
Methodtranslation
Return the translation component of the pose as a np.ndarray. Returns ---------- self.tvec: np.ndarray Translatio
tridet/structures/pose.py:125
Functiontrivial_batch_collator
A batch collator that does nothing.
tridet/data/build.py:152
Methodtvec
(self)
tridet/structures/boxes3d.py:44
Methodtvec
(self)
tridet/structures/boxes3d.py:170
Methoduse_bias
(self)
tridet/modeling/backbone/omni_scripts/ops.py:71
Methodvisualize
(self, x)
tridet/visualizers/d2_visualizer.py:173
Methodvisualize
Parameters ---------- x: Dict One 'dataset_dict'. Returns ------- viz_images: Dict[np.a
tridet/visualizers/box3d_visualizer.py:267
Methodvisualize
(self, x)
tridet/visualizers/box3d_visualizer.py:358
Functionworker_init_reset_seed
(worker_id)
tridet/data/build.py:159
Functionwrapped_fn
(*args, **kwargs)
tridet/utils/wandb.py:54
Functionwrapped_fn
(*args, **kwargs)
tridet/utils/comm.py:61
Functionwrapper
(*args, **kwargs)
tridet/utils/comm.py:27
Methodzero_last_gamma
(self)
tridet/modeling/backbone/omni_scripts/ops.py:525
← previous401–487 of 487, ranked by callers