MCPcopy Create free account

hub / github.com/apple/ARKitScenes / functions

Functions146 in github.com/apple/ARKitScenes

↓ 5 callersFunctiondownload_file
(url, file_name, dst)
download_data.py:51
↓ 4 callersFunctioncreate_gradient_log_loss
(log_prediction_d, mask, log_gt)
depth_upsampling/losses/gradient_loss.py:42
↓ 4 callersFunctionimage_chw_to_hwc
revert image_hwc_to_chw function
depth_upsampling/data_utils.py:20
↓ 4 callersFunctionline_actor
(points)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:259
↓ 3 callersFunctionTrajStringToMatrix
convert traj_str into translation and rotation matrices Args: traj_str: A space-delimited file where each line represents a camera positi
threedod/benchmark_scripts/utils/tenFpsDataLoader.py:13
↓ 3 callersFunctionboxes_to_corners_3d
7 -------- 4 /| /| 6 -------- 5 . | | | | . 3 -------- 0 |/ |/ 2 -------- 1 A
threedod/benchmark_scripts/utils/box_utils.py:90
↓ 3 callersFunctionextract_gt
extract original label data Args: gt_fn: str (file name of "annotation.json") after loading, we got a dict with keys
threedod/benchmark_scripts/utils/tenFpsDataLoader.py:105
↓ 3 callersFunctionimage_hwc_to_chw
transpose the image from height, width, channel -> channel, height, width (pytorch format)
depth_upsampling/data_utils.py:12
↓ 3 callersFunctioninside
(p)
threedod/benchmark_scripts/utils/box_utils.py:184
↓ 3 callersMethodload_image
(path, shape, is_depth, sky_direction)
depth_upsampling/dataset.py:93
↓ 2 callersMethod__init__
( self, num_input_features: int, growth_rate: int, bn_size: int, drop_
depth_upsampling/models/mspf/blocks/dense_net.py:9
↓ 2 callersFunctionbatch_to_cuda
(batch)
depth_upsampling/data_utils.py:27
↓ 2 callersFunctionbox3d_vol
corners: (8,3) no assumption on axis direction
threedod/benchmark_scripts/utils/box_utils.py:232
↓ 2 callersFunctioncomputeIntersection
()
threedod/benchmark_scripts/utils/box_utils.py:187
↓ 2 callersFunctiondecide_pose
Args: pose: np.array (4, 4) Returns: index: int (0, 1, 2, 3) for upright, left, upside-down and right
threedod/benchmark_scripts/rectify_im.py:7
↓ 2 callersFunctioneval_det_cls
Generic functions to compute precision/recall for object detection for a single class. Args: pred: map of {img_id: [(bbox, score)]} wh
threedod/benchmark_scripts/utils/eval_utils.py:119
↓ 2 callersFunctionget_size
Args: box: 8x3 Returns: size: [dx, dy, dz]
threedod/benchmark_scripts/utils/box_utils.py:10
↓ 2 callersFunctionpoly_area
Ref: http://stackoverflow.com/questions/24467972/calculate-area-of-polygon-given-x-y-coordinates
threedod/benchmark_scripts/utils/box_utils.py:167
↓ 1 callersMethod__init__
( self, growth_rate: int = 32, block_config: Tuple[int, int, int, int] = (6, 12, 24, 1
depth_upsampling/models/mspf/densenet.py:16
↓ 1 callersMethod__init__
(self, in_channels, out_channels, bias=False, kernel_size=1, stride=1, padding=0, dilation=1, activation=None,
depth_upsampling/models/mspf/blocks/multi_scale_depth.py:10
↓ 1 callersMethod__init__
(self, in_channels, out_channels, kernel=5, stride=1, padding=2)
depth_upsampling/models/msg/blocks.py:5
↓ 1 callersMethod__init__
:param ply_file: path of ply file :param json_file: path of annotation result json file :param back_face_cull: see single sid
threedod/benchmark_scripts/show_3d_bbox_annotation.py:11
↓ 1 callersMethod_prepare
(self)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:40
↓ 1 callersFunctionaccumulate_wrapper
Args: loader: TenFpsDataLoader Returns: world_pc: (N, 3) xyz in world coordinate system world_sem: (N, d)
threedod/benchmark_scripts/data_prepare_offline.py:19
↓ 1 callersMethodadd_actor
(self)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:118
↓ 1 callersMethodany_requires_grad
(self, input: List[Tensor])
depth_upsampling/models/mspf/blocks/dense_net.py:42
↓ 1 callersMethodbackward
In the backward pass we receive a Tensor containing the gradient of the loss with respect to the output, and we need to compute the g
depth_upsampling/losses/gradient_loss.py:27
↓ 1 callersMethodbind_mouse_event
(self)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:115
↓ 1 callersMethodbn_function
(self, inputs: List[Tensor])
depth_upsampling/models/mspf/blocks/dense_net.py:37
↓ 1 callersFunctioncheck_file_type
(file)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:233
↓ 1 callersFunctioncompute_box_3d
(scale, transform, rotation)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:215
↓ 1 callersFunctioncompute_box_3d
Compute corners of a single box from rotation matrix Args: size: list of float [dx, dy, dz] center: np.array [x, y, z] rot
threedod/benchmark_scripts/utils/box_utils.py:38
↓ 1 callersFunctioncompute_errors
(gt, pred, valid_mask)
depth_upsampling/logs/eval.py:15
↓ 1 callersFunctionconvert_angle_axis_to_matrix3
Return a Matrix3 for the angle axis. Arguments: angle_axis {Point3} -- a rotation in angle axis form.
threedod/benchmark_scripts/utils/rotation.py:135
↓ 1 callersFunctionconvex_hull_intersection
Compute area of two convex hull's intersection area. p1,p2 are a list of (x,y) tuples of hull vertices. return a list of (x,y) for th
threedod/benchmark_scripts/utils/box_utils.py:219
↓ 1 callersFunctioncreate_gradient_log_loss_4_scales
(log_prediction, log_ground_truth, mask)
depth_upsampling/losses/gradient_loss.py:66
↓ 1 callersFunctiondiv_by_mask_sum
(loss: torch.Tensor, mask_sum: torch.Tensor)
depth_upsampling/losses/gradient_loss.py:6
↓ 1 callersFunctiondownload_data
(dataset, video_ids, dataset_splits, download_dir,
download_data.py:152
↓ 1 callersFunctiondownload_laser_scanner_point_clouds
(laser_scanner_point_cloud_id, visit_id, download_dir)
download_data.py:126
↓ 1 callersFunctiondownload_laser_scanner_point_clouds_for_video
(video_id, metadata, download_dir)
download_data.py:83
↓ 1 callersMethoddraw_bbox
(self, bbox)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:132
↓ 1 callersMethoddraw_lines
(self)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:128
↓ 1 callersFunctioneulerAnglesToRotationMatrix
Euler rotation matrix with clockwise logic. Rotation Args: theta: list of float [theta_x, theta_y, theta_z] Returns:
threedod/benchmark_scripts/utils/rotation.py:6
↓ 1 callersFunctioneval_log
(step, model, dataloader, tensorboard_writer)
depth_upsampling/logs/eval.py:27
↓ 1 callersFunctionexpand_channel_dim
expand image dimension to add a channel dimension
depth_upsampling/data_utils.py:5
↓ 1 callersFunctiongenerate_point
Generate 3D point coordinates and related rgb feature Args: rgb_image: (h, w, 3) rgb depth_image: (h, w) depth intrinsic:
threedod/benchmark_scripts/utils/tenFpsDataLoader.py:51
↓ 1 callersFunctionget_args
()
threedod/benchmark_scripts/show_3d_bbox_annotation.py:288
↓ 1 callersFunctionget_heading_angle
Args: box: (8, 3) Returns: heading_angle: float
threedod/benchmark_scripts/utils/box_utils.py:24
↓ 1 callersFunctionget_lines
Args: box: np.array (8, 3) 8 corners color: line color Returns: o3d.Linset()
threedod/benchmark_scripts/utils/visual_utils.py:89
↓ 1 callersFunctionget_loss
(network)
depth_upsampling/losses/__init__.py:14
↓ 1 callersFunctionget_metadata
(dataset, download_dir)
download_data.py:139
↓ 1 callersFunctionget_network
(network, upsampling_factor)
depth_upsampling/models/__init__.py:16
↓ 1 callersFunctionget_votes
Args: points: (N, 3) boxes: (m, 8, 3) Returns: votes: (N, 4)
threedod/benchmark_scripts/prepare_votes.py:11
↓ 1 callersFunctiongradient_loss
(outputs, inputs)
depth_upsampling/losses/gradient_loss.py:104
↓ 1 callersMethodinit_coordinate_axes
(self)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:121
↓ 1 callersFunctionl1_loss
(outputs, inputs)
depth_upsampling/losses/l1_loss.py:8
↓ 1 callersFunctionlaser_scanner_point_clouds_for_visit_id
(visit_id, download_dir)
download_data.py:103
↓ 1 callersFunctionload_json
(js_path)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:209
↓ 1 callersFunctionmain
(args)
depth_upsampling/train.py:26
↓ 1 callersFunctionpoints_in_boxes
Args: pc: np.array (n, 3+d) boxes: np.array (m, 8, 3) Returns: mask: np.array (n, m) of type bool
threedod/benchmark_scripts/utils/box_utils.py:127
↓ 1 callersFunctionpolygon_clip
Clip a polygon with another polygon. Ref: https://rosettacode.org/wiki/Sutherland-Hodgman_polygon_clipping#Python Args: subjectPolygon:
threedod/benchmark_scripts/utils/box_utils.py:172
↓ 1 callersFunctionraw_files
(video_id, assets, metadata)
download_data.py:23
↓ 1 callersMethodread_mesh
(self)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:52
↓ 1 callersMethodreset
(self)
threedod/benchmark_scripts/utils/eval_utils.py:113
↓ 1 callersFunctionrmse_loss
(outputs, inputs)
depth_upsampling/losses/rmse.py:7
↓ 1 callersMethodrotate_image
(img, direction)
depth_upsampling/dataset.py:79
↓ 1 callersFunctionrotate_points_along_z
Rotation clockwise Args: points: np.array of np.array (B, N, 3 + C) or (N, 3 + C) for single batch angle: np.array of
threedod/benchmark_scripts/utils/rotation.py:99
↓ 1 callersFunctionsample_vis
(dataset_path: str, split: str, sample_id: str, max_depth)
depth_upsampling/sample_vis.py:9
↓ 1 callersMethodset_actor
(self)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:86
↓ 1 callersMethodset_mapper
(self)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:70
↓ 1 callersMethodset_render
(self)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:105
↓ 1 callersFunctionst2_camera_intrinsics
(filename)
threedod/benchmark_scripts/utils/tenFpsDataLoader.py:46
↓ 1 callersMethodstep
Accumulate one batch of prediction and groundtruth. self.gt_map_cls[idx]: a list of (int, np.array (8, 3)) self.gt_map_cls[idx
threedod/benchmark_scripts/utils/eval_utils.py:34
↓ 1 callersFunctiontrain_log
(step, input_batch, output_batch, tensorboard_writer, **kwargs)
depth_upsampling/logs/train.py:8
↓ 1 callersMethodtransform_actor
(self)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:100
↓ 1 callersFunctionunzip_file
(file_name, dst, keep_zip=True)
download_data.py:69
↓ 1 callersFunctionvisualize_o3d
Visualize result with open3d Args: pc: np.array of shape (n, 3) point cloud boxes: a list of m boxes, each item a
threedod/benchmark_scripts/utils/visual_utils.py:29
↓ 1 callersFunctionvoc_ap
ap = voc_ap(rec, prec, [use_07_metric]) Compute VOC AP given precision and recall. If use_07_metric is true, uses the VOC 07 11 point meth
threedod/benchmark_scripts/utils/eval_utils.py:245
Method__call__
(self, batch)
depth_upsampling/transfroms/dilate_valid_mask.py:11
Method__call__
(self, sample)
depth_upsampling/transfroms/valid_depth_mask.py:11
Method__call__
(self, sample)
depth_upsampling/transfroms/random_fliplr.py:12
Method__call__
(self, sample)
depth_upsampling/transfroms/ascontiguousarray.py:8
Method__call__
(self, sample)
depth_upsampling/transfroms/random_crop.py:13
Method__call__
(self, sample)
depth_upsampling/transfroms/mod_crop.py:9
Method__call__
(self)
threedod/benchmark_scripts/show_3d_bbox_annotation.py:31
Method__getitem__
Args: index (int): Index Returns: tuple: (identifier, color, highres_depth, lowres_depth).
depth_upsampling/dataset.py:105
Method__getitem__
Returns: frame: a dict {frame_id}: str {depth}: (h, w) {image}: (h, w)
threedod/benchmark_scripts/utils/tenFpsDataLoader.py:297
Method__init__
( self, root: str, split: str = 'train', transform: Optional[C
depth_upsampling/dataset.py:33
Method__init__
(self, data_source, num_iter, start_itr=0, batch_size=128)
depth_upsampling/sampler.py:14
Method__init__
(self, dilation_radius: int)
depth_upsampling/transfroms/dilate_valid_mask.py:7
Method__init__
(self, gt_low_limit: float = None, gt_high_limit: float = None)
depth_upsampling/transfroms/valid_depth_mask.py:5
Method__init__
(self)
depth_upsampling/transfroms/random_fliplr.py:9
Method__init__
(self)
depth_upsampling/transfroms/ascontiguousarray.py:5
Method__init__
(self, height: int, width: int, upsample_factor: int = None)
depth_upsampling/transfroms/random_crop.py:7
Method__init__
(self, modulo: int)
depth_upsampling/transfroms/mod_crop.py:5
Method__init__
(self)
depth_upsampling/models/mspf/densenet.py:83
Method__init__
(self, upsample_factor)
depth_upsampling/models/mspf/MultiscaleConvDepthEncoder.py:12
Method__init__
( self, input_channels: List[int], output_channels: List[int], upsample_factor
depth_upsampling/models/mspf/MultiScaleDepthSR.py:20
Method__init__
(self, upsample_factor, decoder_channel_scale=2)
depth_upsampling/models/mspf/mspf.py:23
Method__init__
( self, num_layers: int, num_input_features: int, bn_size: int, growth
depth_upsampling/models/mspf/blocks/dense_net.py:82
next →1–100 of 146, ranked by callers