MCPcopy Create free account

hub / github.com/antonioo-c/PiMAE / functions

Functions1,397 in github.com/antonioo-c/PiMAE

↓ 2 callersFunction_unellipsify
Pretrain/utils_detr/box_intersection.c:13256
↓ 2 callersFunctionabs_py_ssize_t
Pretrain/utils_detr/box_intersection.c:16828
↓ 2 callersFunctionadd_detr_config
Add config for DETR.
Downstream/detr/d2/detr/config.py:6
↓ 2 callersFunctionall_gather
Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]
Downstream/detr/util/misc.py:89
↓ 2 callersFunctionall_reduce_sum
(tensor)
Downstream/3detr/utils/dist.py:67
↓ 2 callersFunctionall_reduce_sum
(tensor)
Pretrain/utils_detr/dist.py:67
↓ 2 callersFunctionbox3d_vol
corners: (8,3) no assumption on axis direction
Downstream/3detr/utils/box_ops3d.py:165
↓ 2 callersFunctionbox3d_vol
corners: (8,3) no assumption on axis direction
Downstream/3detr/utils/box_util.py:102
↓ 2 callersFunctionbox3d_vol
corners: (8,3) no assumption on axis direction
Pretrain/utils_detr/box_ops3d.py:165
↓ 2 callersFunctionbox3d_vol
corners: (8,3) no assumption on axis direction
Pretrain/utils_detr/box_util.py:102
↓ 2 callersFunctionbox_xyxy_to_cxcywh
(x)
Downstream/detr/util/box_ops.py:16
↓ 2 callersFunctionbuild_backbone
(args)
Downstream/detr/models/backbone.py:250
↓ 2 callersFunctionbuild_encoder
(args)
Downstream/detr/models/backbone.py:235
↓ 2 callersFunctionbuild_model
(args)
Downstream/detr/models/__init__.py:5
↓ 2 callersFunctionbuild_model_from_cfg
Build a dataset, defined by `dataset_name`. Args: cfg (eDICT): Returns: Dataset: a constructed dataset specified by data
Pretrain/models/build.py:20
↓ 2 callersMethodclass2angle
Inverse function to angle2class
Downstream/3detr/datasets/sunrgbd.py:97
↓ 2 callersMethodclass2angle_batch
(self, pred_cls, residual, to_label_format=True)
Downstream/3detr/datasets/sunrgbd.py:107
↓ 2 callersFunctioncomputeIntersection
()
Downstream/3detr/utils/box_ops3d.py:40
↓ 2 callersFunctioncomputeIntersection
()
Downstream/3detr/utils/box_util.py:52
↓ 2 callersFunctioncomputeIntersection
()
Pretrain/utils_detr/box_ops3d.py:40
↓ 2 callersFunctioncomputeIntersection
()
Pretrain/utils_detr/box_util.py:52
↓ 2 callersFunctionconvert_coco_poly_to_mask
(segmentations, height, width)
Downstream/detr/datasets/coco.py:33
↓ 2 callersFunctionconvert_oriented_box_to_trimesh_fmt
(box)
Downstream/3detr/utils/pc_util.py:197
↓ 2 callersFunctionconvert_oriented_box_to_trimesh_fmt
(box)
Pretrain/utils_detr/pc_util.py:197
↓ 2 callersFunctionconvert_to_xywh
(boxes)
Downstream/detr/datasets/coco_eval.py:165
↓ 2 callersFunctioncopy_strided_to_strided
Pretrain/utils_detr/box_intersection.c:17321
↓ 2 callersFunctiondownload_label_map
(out_dir)
Downstream/3detr/datasets/download_scannet.py:102
↓ 2 callersFunctiondownload_release
(release_scans, out_dir, file_types, use_v1_sens)
Downstream/3detr/datasets/download_scannet.py:46
↓ 2 callersFunctiondownload_scan
(scan_id, out_dir, file_types, use_v1_sens)
Downstream/3detr/datasets/download_scannet.py:72
↓ 2 callersFunctionenclosing_box3d_vol
volume of enclosing axis-aligned box
Downstream/3detr/utils/box_util.py:466
↓ 2 callersFunctionenclosing_box3d_vol
volume of enclosing axis-aligned box
Pretrain/utils_detr/box_util.py:466
↓ 2 callersFunctioneval_det_cls
Generic functions to compute precision/recall for object detection for a single class. Input: pred: map of {img_id: [(bbox, score)]} w
Downstream/3detr/utils/eval_det.py:66
↓ 2 callersFunctioneval_det_cls
Generic functions to compute precision/recall for object detection for a single class. Input: pred: map of {img_id: [(bbox, score)]} w
Pretrain/utils_detr/eval_det.py:66
↓ 2 callersFunctionevaluate
(model, criterion, postprocessors, data_loader, base_ds, device, output_dir)
Downstream/detr/engine.py:69
↓ 2 callersFunctionflip_axis_to_camera
Flip X-right,Y-forward,Z-up to X-right,Y-down,Z-forward Input and output are both (N,3) array
Pretrain/utils/sunrgbd_utils.py:22
↓ 2 callersFunctionflip_axis_to_camera
Flip X-right,Y-forward,Z-up to X-right,Y-down,Z-forward Input and output are both (N,3) array
Pretrain/datasets/sunrgbd_utils.py:26
↓ 2 callersFunctionflip_axis_to_camera_np
Flip X-right,Y-forward,Z-up to X-right,Y-down,Z-forward Input and output are both (N,3) array
Downstream/3detr/utils/box_util.py:255
↓ 2 callersFunctionflip_axis_to_camera_tensor
Flip X-right,Y-forward,Z-up to X-right,Y-down,Z-forward Input and output are both (N,3) array
Downstream/3detr/utils/box_util.py:288
↓ 2 callersFunctionflip_axis_to_depth
(pc)
Pretrain/utils/sunrgbd_utils.py:31
↓ 2 callersFunctionflip_axis_to_depth
(pc)
Pretrain/datasets/sunrgbd_utils.py:36
↓ 2 callersMethodforward_img_decoder
(self, x, ids_restore)
Pretrain/models/pimae.py:233
↓ 2 callersMethodforward_pc_decoder
(self, x_vis, center, mask)
Pretrain/models/pimae.py:175
↓ 2 callersFunctionget_3d_box_batch_np
(box_size, angle, center)
Downstream/3detr/utils/box_util.py:265
↓ 2 callersFunctionget_3d_box_batch_tensor
(box_size, angle, center)
Downstream/3detr/utils/box_util.py:313
↓ 2 callersFunctionget_clones
(module, N)
Downstream/3detr/models/helpers.py:115
↓ 2 callersFunctionget_clones
(module, N)
Downstream/detr/models/helpers.py:115
↓ 2 callersFunctionget_clones
(module, N)
Pretrain/models/transformer.py:20
↓ 2 callersFunctionget_config
(args, logger=None)
Pretrain/utils/config.py:47
↓ 2 callersMethodget_loss
(self, loss, outputs, targets, indices, num_boxes, **kwargs)
Downstream/detr/models/detr.py:209
↓ 2 callersFunctionget_release_scans
(release_file)
Downstream/3detr/datasets/download_scannet.py:36
↓ 2 callersFunctionget_root_logger
Get root logger and add a keyword filter to it. The logger will be initialized if it has not been initialized. By default a StreamHandler will
Pretrain/utils/logger.py:6
↓ 2 callersMethodget_target
(self, idx)
Downstream/3detr/datasets/sunrgbd.py:213
↓ 2 callersFunctionget_world_size
()
Downstream/MonoDETR/utils/misc.py:389
↓ 2 callersFunctionheading2rotmat
(heading_angle)
Downstream/3detr/utils/pc_util.py:188
↓ 2 callersFunctionheading2rotmat
(heading_angle)
Pretrain/utils_detr/pc_util.py:188
↓ 2 callersFunctionhelper_computeIntersection
(cp1: torch.Tensor, cp2: torch.Tensor, s: torch.Tensor, e: torch.Tensor)
Downstream/3detr/utils/box_ops3d.py:78
↓ 2 callersFunctionhelper_computeIntersection
( cp1: torch.Tensor, cp2: torch.Tensor, s: torch.Tensor, e: torch.Tensor )
Downstream/3detr/utils/box_util.py:387
↓ 2 callersFunctionhelper_computeIntersection
(cp1: torch.Tensor, cp2: torch.Tensor, s: torch.Tensor, e: torch.Tensor)
Pretrain/utils_detr/box_ops3d.py:78
↓ 2 callersFunctionhelper_computeIntersection
( cp1: torch.Tensor, cp2: torch.Tensor, s: torch.Tensor, e: torch.Tensor )
Pretrain/utils_detr/box_util.py:387
↓ 2 callersFunctionis_main_process
()
Downstream/detr/util/misc.py:398
↓ 2 callersFunctionloadPretrain
(model,path,first_depth,second_depth, is_distributed=False)
Downstream/3detr/main_load.py:355
↓ 2 callersFunctionloadPretrain
(model,path,first_depth,second_depth, prefix='', is_distributed=False)
Pretrain/tutorial_load.py:212
↓ 2 callersFunctionlog_args_to_file
(args, pre='args', logger=None)
Pretrain/utils/config.py:6
↓ 2 callersFunctionlog_config_to_file
(cfg, pre='cfg', logger=None)
Pretrain/utils/config.py:10
↓ 2 callersMethodlog_every
(self, iterable, print_freq, header=None)
Downstream/detr/util/misc.py:194
↓ 2 callersFunctionmake_coco_transforms
(image_set)
Downstream/detr/datasets/coco.py:120
↓ 2 callersMethodnew_align
(self, calib_Rtilt, calib_K, scale, img_size, points)
Pretrain/models/pimae.py:341
↓ 2 callersFunctionnorm_cdf
(x)
Pretrain/models/multimae_utils.py:51
↓ 2 callersMethodpatchify
imgs: (N, 3, H, W) x: (N, L, patch_size**2 *3)
Pretrain/models/models_mae.py:126
↓ 2 callersFunctionplot_polys
(plist,scale=500.0)
Downstream/3detr/utils/box_ops3d.py:718
↓ 2 callersFunctionplot_polys
(plist,scale=500.0)
Pretrain/utils_detr/box_ops3d.py:718
↓ 2 callersFunctionpoly_area
Ref: http://stackoverflow.com/questions/24467972/calculate-area-of-polygon-given-x-y-coordinates
Downstream/3detr/utils/box_ops3d.py:130
↓ 2 callersFunctionpoly_area
Ref: http://stackoverflow.com/questions/24467972/calculate-area-of-polygon-given-x-y-coordinates
Downstream/3detr/utils/box_util.py:84
↓ 2 callersFunctionpoly_area
Ref: http://stackoverflow.com/questions/24467972/calculate-area-of-polygon-given-x-y-coordinates
Pretrain/utils_detr/box_ops3d.py:130
↓ 2 callersFunctionpoly_area
Ref: http://stackoverflow.com/questions/24467972/calculate-area-of-polygon-given-x-y-coordinates
Pretrain/utils_detr/box_util.py:84
↓ 2 callersFunctionpolygon_clip
Clip a polygon with another polygon. Ref: https://rosettacode.org/wiki/Sutherland-Hodgman_polygon_clipping#Python Args: subjectPolygon:
Downstream/3detr/utils/box_ops3d.py:18
↓ 2 callersFunctionpolygon_clip
Clip a polygon with another polygon. Ref: https://rosettacode.org/wiki/Sutherland-Hodgman_polygon_clipping#Python Args: subjectPolygon:
Pretrain/utils_detr/box_ops3d.py:18
↓ 2 callersMethodprepare
(self, predictions, iou_type)
Downstream/detr/datasets/coco_eval.py:69
↓ 2 callersFunctionproject_upright_depth_to_camera
project point cloud from depth coord to camera coordinate Input: (N,3) Output: (N,3)
Pretrain/utils/matcher.py:22
↓ 2 callersMethodproject_upright_depth_to_image
Input: (N,3) Output: (N,2) UV and (N,) depth
Pretrain/utils/sunrgbd_utils.py:99
↓ 2 callersFunctionreduce_dict
Args: input_dict (dict): all the values will be reduced average (bool): whether to do average or sum Reduce the values in the
Downstream/3detr/utils/dist.py:86
↓ 2 callersFunctionroty_batch
Rotation about the y-axis. t: (x1,x2,...xn) return: (x1,x2,...,xn,3,3)
Downstream/3detr/utils/box_util.py:220
↓ 2 callersFunctionroty_batch
Rotation about the y-axis. t: (x1,x2,...xn) return: (x1,x2,...,xn,3,3)
Pretrain/utils_detr/box_util.py:220
↓ 2 callersFunctionrotz
Rotation about the z-axis.
Pretrain/utils/sunrgbd_utils.py:149
↓ 2 callersFunctionrotz
Rotation about the z-axis.
Pretrain/datasets/sunrgbd_utils.py:157
↓ 2 callersMethodsingle_output_forward
(self, outputs, targets)
Downstream/3detr/criterion.py:319
↓ 2 callersMethodstep
Perform NMS on predicted boxes and threshold them according to score. Convert GT boxes
Downstream/3detr/utils/ap_calculator.py:324
↓ 2 callersMethodstep
(self, closure=None)
Downstream/detr/train_net.py:83
↓ 2 callersMethodstep_meter
(self, outputs, targets)
Downstream/3detr/utils/ap_calculator.py:311
↓ 2 callersMethodsynchronize_between_processes
(self)
Downstream/detr/util/misc.py:187
↓ 2 callersFunctionunpatchify
x: (N, L, patch_size**2 *3) imgs: (N, 3, H, W)
Pretrain/tools/mae_visualize.py:58
↓ 2 callersMethodupdate
(self, value, n=1)
Downstream/MonoDETR/utils/misc.py:43
↓ 2 callersMethodupdate
(self, value, n=1)
Downstream/detr/util/misc.py:40
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
Downstream/3detr/models/transformer.py:241
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
Downstream/detr/models/transformer_orig.py:173
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
Downstream/detr/models/transformer_3detr.py:241
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
Downstream/detr/models/transformer.py:173
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
Downstream/detr/models/backbone.py:64
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
Pretrain/tutorial_load.py:137
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
Pretrain/models/transformer.py:283
← previousnext →201–300 of 1,397, ranked by callers