MCPcopy Create free account

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

Functions487 in github.com/TRI-ML/dd3d

Method__call__
(self, locations, gt_instances, feature_shapes)
tridet/modeling/dd3d/prepare_targets.py:28
Method__call__
(self, logits, box2d_reg, centerness, targets)
tridet/modeling/dd3d/fcos2d.py:168
Method__call__
(self, logits, box2d_reg, centerness, locations, image_sizes)
tridet/modeling/dd3d/fcos2d.py:251
Method__call__
( self, box3d_quat, box3d_ctr, box3d_depth, box3d_size, box3d_conf, dense_depth, inv_intrinsics, fcos2
tridet/modeling/dd3d/fcos3d.py:209
Method__call__
( self, box3d_quat, box3d_ctr, box3d_depth, box3d_size, box3d_conf, inv_intrinsics, pred_instances, fc
tridet/modeling/dd3d/fcos3d.py:315
Method__call__
(self, dataset_dict)
tridet/data/dataset_mappers/nuscenes_mapper.py:13
Method__call__
Args: dataset_dict (dict): Metadata of one image, in Detectron2 Dataset format. Returns: dict: a format that
tridet/data/dataset_mappers/dataset_mapper.py:100
Method__eq__
(self, other)
tridet/structures/pose.py:163
Method__getitem__
(self, idx)
tridet/data/datasets/kitti_3d/build.py:162
Method__getitem__
(self, idx)
tridet/data/datasets/kitti_3d/build.py:289
Method__getitem__
(self, idx)
tridet/data/datasets/nuscenes/build.py:283
Method__getitem__
Access the individual image in its original size. Args: idx: int or slice Returns: Tensor: an image
tridet/structures/image_list.py:71
Method__getitem__
tridet/structures/boxes3d.py:117
Method__getitem__
tridet/structures/boxes3d.py:248
Method__init__
( self, backbone: nn.Module, fpn: FPN, last_channels: int, act_func="relu", n_extra_stage=0, return_list=F
tridet/modeling/backbone/omni_scripts/backbone_with_fpn.py:61
Method__init__
( self, width_mult=1.0, channel_divisor=8, ks: Union[int, List[int], None] = None, expand_ratio: Uni
tridet/modeling/backbone/omni_scripts/fused_mb_nets.py:11
Method__init__
( self, # inputs inputs: List[Tuple[str, int, int, int]], input_mode="cat_conv", # middle middle_c
tridet/modeling/backbone/omni_scripts/fpn.py:74
Method__init__
( self, in_channels: int, out_channels: int, kernel_size=3, stride=1,
tridet/modeling/backbone/omni_scripts/ops.py:12
Method__init__
( self, pool_type="global_avg", kernel_size=2, stride=2, padding=0,
tridet/modeling/backbone/omni_scripts/ops.py:106
Method__init__
( self, in_channels: int, pool_size: List[int], pool_type: str = "max",
tridet/modeling/backbone/omni_scripts/ops.py:139
Method__init__
( self, mode="bilinear", size: Union[int, Tuple[int, int], List[int], None] = None,
tridet/modeling/backbone/omni_scripts/ops.py:194
Method__init__
( self, in_channels: int, out_channels: int, kernel_size=3, stride=1, act_func=("relu6", None) )
tridet/modeling/backbone/omni_scripts/ops.py:235
Method__init__
( self, in_channels: int, out_channels: int, kernel_size=3, stride=1,
tridet/modeling/backbone/omni_scripts/ops.py:271
Method__init__
( self, in_channels: int, out_channels: int, kernel_size=3, stride=1,
tridet/modeling/backbone/omni_scripts/ops.py:332
Method__init__
(self, conv, shortcut, post_residual_act=None)
tridet/modeling/backbone/omni_scripts/ops.py:391
Method__init__
(self, input_stem: nn.Module, stages: List[nn.Module])
tridet/modeling/backbone/omni_scripts/ops.py:466
Method__init__
(self, inplanes, planes, stride=1, dilation=1, norm="BN")
tridet/modeling/feature_extractor/dla.py:25
Method__init__
(self, inplanes, planes, stride=1, dilation=1, norm="BN")
tridet/modeling/feature_extractor/dla.py:68
Method__init__
(self, inplanes, planes, stride=1, dilation=1, norm="BN")
tridet/modeling/feature_extractor/dla.py:108
Method__init__
(self, in_channels, out_channels, kernel_size, residual, norm="BN")
tridet/modeling/feature_extractor/dla.py:147
Method__init__
( self, levels, block, in_channels, out_channels, stride=1,
tridet/modeling/feature_extractor/dla.py:171
Method__init__
(self, in_channels, out_channels, in_feature="res5")
tridet/modeling/feature_extractor/dla.py:491
Method__init__
(self, inplace=True)
tridet/modeling/feature_extractor/vovnet.py:165
Method__init__
(self, channel, reduction=4)
tridet/modeling/feature_extractor/vovnet.py:174
Method__init__
( self, in_ch, stage_ch, concat_ch, layer_per_block, module_name, SE=False, identity=False, depthwise=
tridet/modeling/feature_extractor/vovnet.py:189
Method__init__
( self, in_ch, stage_ch, concat_ch, block_per_stage, layer_per_block, stage_num, SE=False, depthwise=F
tridet/modeling/feature_extractor/vovnet.py:242
Method__init__
(self, in_channels, out_channels, in_features="res5")
tridet/modeling/feature_extractor/vovnet.py:415
Method__init__
(self, cfg)
tridet/modeling/dd3d/test_time_augmentation.py:32
Method__init__
Args: cfg (CfgNode): model (GeneralizedRCNN): a GeneralizedRCNN to apply TTA on. tta_mapper (callable): t
tridet/modeling/dd3d/test_time_augmentation.py:93
Method__init__
( self, net, input_shape, min_depth, max_depth, scale_depth_by
tridet/modeling/dd3d/depth.py:9
Method__init__
(self, smooth_l1_loss_beta, max_loss_per_group)
tridet/modeling/dd3d/disentangled_box3d_loss.py:15
Method__init__
(self, cfg, input_shape)
tridet/modeling/dd3d/dense_depth.py:18
Method__init__
(self, cfg, model, tta_mapper=None)
tridet/modeling/dd3d/nuscenes_dd3d_tta.py:22
Method__init__
(self, cfg)
tridet/modeling/dd3d/core.py:21
Method__init__
(self, cfg, input_shape)
tridet/modeling/dd3d/nuscenes_dd3d.py:25
Method__init__
(self, cfg)
tridet/modeling/dd3d/nuscenes_dd3d.py:269
Method__init__
(self, cfg)
tridet/modeling/dd3d/nuscenes_dd3d.py:301
Method__init__
(self, beta, min_depth=0., max_depth=100., loss_weight=1.0)
tridet/modeling/dd3d/dense_depth_loss.py:12
Method__init__
(self, cfg, input_shape)
tridet/modeling/dd3d/prepare_targets.py:12
Method__init__
(self, cfg, input_shape)
tridet/modeling/dd3d/fcos2d.py:31
Method__init__
(self, cfg)
tridet/modeling/dd3d/fcos2d.py:160
Method__init__
(self, cfg)
tridet/modeling/dd3d/fcos2d.py:243
Method__init__
(self, cfg, input_shape)
tridet/modeling/dd3d/fcos3d.py:56
Method__init__
(self, cfg)
tridet/modeling/dd3d/fcos3d.py:192
Method__init__
(self, cfg)
tridet/modeling/dd3d/fcos3d.py:303
Method__init__
(self, name, is_detection_task, is_dense_prediction_task)
tridet/utils/tasks.py:8
Method__init__
configurable is experimental.
tridet/utils/tasks.py:39
Method__init__
(self, cfg, dataset_name, inference_output_dir)
tridet/visualizers/d2_visualizer.py:73
Method__init__
(self, cfg, dataset_name)
tridet/visualizers/d2_visualizer.py:167
Method__init__
( self, metric_width=100.0, metric_height=100.0, pixels_per_meter=10.0,
tridet/visualizers/bev.py:61
Method__init__
(self, cfg, dataset_name, inference_output_dir)
tridet/visualizers/box3d_visualizer.py:238
Method__init__
(self, cfg, dataset_name)
tridet/visualizers/box3d_visualizer.py:352
Method__init__
(self)
tridet/data/build.py:178
Method__init__
Args: size (int): the total number of data of the underlying dataset to sample from
tridet/data/samplers/group_sampler.py:14
Method__init__
NOTE: 'configurable' interface is experimental. Args: is_train: whether it's used in training or inference a
tridet/data/dataset_mappers/dataset_mapper.py:54
Method__init__
(self, root_dir, mv3d_split, class_names, sensors, box2d_from_box3d=False, max_num_items=None)
tridet/data/datasets/kitti_3d/build.py:54
Method__init__
(self, root_dir, mv3d_split, class_names, sensors, box2d_from_box3d, max_num_items)
tridet/data/datasets/kitti_3d/build.py:282
Method__init__
(self, name, root_dir, datum_names=CAMERA_NAMES, min_num_lidar_points=3, min_box_visibility=0.2)
tridet/data/datasets/nuscenes/build.py:100
Method__init__
Arguments: tensor (Tensor): of shape (N, H, W) or (N, C_1, ..., C_K, H, W) where K >= 1 image_sizes (list[tuple[int,
tridet/structures/image_list.py:45
Method__init__
Initialize a Pose with Quaternion and 3D Position Parameters ---------- wxyz: np.float32 or Quaternion (default: np.float32([
tridet/structures/pose.py:10
Method__init__
(self, quat, tvec, size)
tridet/structures/boxes3d.py:38
Method__init__
(self, quat, proj_ctr, depth, size, inv_intrinsics)
tridet/structures/boxes3d.py:162
Method__init__
( self, in_channels, out_channels, kernel_size=3, stride=1, di
tridet/layers/separable_conv2d.py:53
Method__init__
( self, num_layers, input_shape, norm_kwargs={}, kernel_size=3,
tridet/layers/conv_bn_fpn_layers.py:15
Method__init__
(self, loc_loss_type='iou')
tridet/layers/iou_loss.py:17
Method__init__
(self, init_value=0.)
tridet/layers/normalization.py:23
Method__init__
(self, modules=None)
tridet/layers/normalization.py:32
Method__init__
(self, modules=None)
tridet/layers/normalization.py:48
Method__init__
(self, out_channels, num_bn_modules, **bn_kwargs)
tridet/layers/normalization.py:66
Method__init__
( self, dataset_name, iou_thresholds, only_prepare_submission=False, o
tridet/evaluators/kitti_3d_evaluator.py:34
Method__init__
(self, id_to_name, num_shards=50, sample_points=41)
tridet/evaluators/kitti_3d_evaluator.py:276
Method__init__
(self, nusc, config, result_path, eval_set, output_dir, verbose)
tridet/evaluators/nuscenes_evaluator.py:76
Method__init__
(self, nusc_root, dataset_name, output_dir)
tridet/evaluators/nuscenes_evaluator.py:132
Method__iter__
(self)
tridet/data/samplers/group_sampler.py:37
Method__len__
(self)
tridet/data/samplers/group_sampler.py:40
Method__len__
(self)
tridet/data/datasets/kitti_3d/build.py:159
Method__len__
(self)
tridet/data/datasets/kitti_3d/build.py:286
Method__len__
(self)
tridet/data/datasets/nuscenes/build.py:151
Method__len__
(self)
tridet/structures/image_list.py:68
Method__len__
(self)
tridet/structures/boxes3d.py:133
Method__len__
(self)
tridet/structures/boxes3d.py:272
Method__mul__
Left-multiply Pose with another Pose or 3D-Points. Parameters ---------- other: Pose or np.ndarray 1. Pose: Ident
tridet/structures/pose.py:45
Method__repr__
(self)
tridet/modeling/backbone/omni_scripts/ops.py:227
Method__repr__
(self)
tridet/visualizers/bev.py:92
Method__repr__
(self)
tridet/structures/pose.py:30
Method__rmul__
(self, other)
tridet/structures/pose.py:73
Method_freeze_backbone
(self, freeze_at)
tridet/modeling/feature_extractor/vovnet.py:344
Method_maybe_read_image
(dataset_dict)
tridet/modeling/dd3d/test_time_augmentation.py:139
Method_maybe_read_image
(dataset_dict)
tridet/modeling/dd3d/nuscenes_dd3d_tta.py:44
Method_parse_config
(config: dict)
tridet/modeling/backbone/omni_scripts/ops.py:83
← previousnext →201–300 of 487, ranked by callers