MCPcopy Create free account

hub / github.com/AIR-DISCOVER/Omni-PQ / functions

Functions370 in github.com/AIR-DISCOVER/Omni-PQ

↓ 1 callersFunctioncompute_class_consistency_loss
(end_points, ema_end_points, map_ind, prefix="last_")
models/utils/mean_teacher_consistency_util.py:99
↓ 1 callersFunctioncompute_class_consistency_loss_quad
(end_points, ema_end_points, map_ind, prefix="last_")
models/utils/mean_teacher_consistency_util.py:114
↓ 1 callersFunctioncompute_objectness_loss
Compute objectness loss for the proposals.
models/loss_helper_pq.py:47
↓ 1 callersFunctioncompute_physical_constraints_loss
(end_points,config)
models/loss_helper_pq.py:357
↓ 1 callersFunctioncompute_quad_score_loss
(end_points,num_layer = 6)
models/loss_helper_pq.py:197
↓ 1 callersFunctioncompute_quad_score_loss_ema
(end_points, num_layer=6)
models/ema_loss_helper_pq.py:21
↓ 1 callersFunctioncompute_size_consistency_loss
(end_points, ema_end_points, map_ind, config, prefix="last_")
models/utils/mean_teacher_consistency_util.py:130
↓ 1 callersFunctioncompute_size_consistency_loss_quad
(end_points, ema_end_points, map_ind, config, prefix="last_")
models/utils/mean_teacher_consistency_util.py:168
↓ 1 callersFunctioncompute_vote_loss
(end_points)
models/loss_helper_pq.py:24
↓ 1 callersFunctionconvert_box_to_trimesh_fmt
(box)
utils/pc_util.py:371
↓ 1 callersFunctiondecode_scores
(base_xyz,objectness_scores, center, heading_scores, heading_residuals_normalized, size_scores, size_residuals
models/pq_transformer.py:35
↓ 1 callersFunctiondemo_nn_distance
()
utils/nn_distance.py:63
↓ 1 callersFunctiondistance_loss
(end_points, config, query_points_obj_topk, pc_loss, num_layer)
models/utils/distance_util.py:396
↓ 1 callersFunctiondistance_loss_gamma_mixture
(end_points, config, query_points_obj_topk, pc_loss, num_layer)
models/utils/distance_util.py:310
↓ 1 callersFunctiondump_pc
Dump point clouds Args: point_clouds (_type_): _description_ dump_dir (str, optional): _description_. Defaults to "./dump/".
models/dump_helper.py:38
↓ 1 callersFunctiondump_results_quad
Dump results. Args: end_points: dict {..., pred_mask} pred_mask is a binary mask array of size (batch_size, num_
models/dump_helper_quad.py:53
↓ 1 callersFunctionerror_pdf
(func, data_arr, steps=50000)
fit.py:60
↓ 1 callersFunctionevaluate_one_epoch
(test_loader, DATASET_CONFIG, CONFIG_DICT, AP_IOU_THRESHOLDS, model, criterion, config, ema=False)
eval.py:369
↓ 1 callersFunctionexport_one_scan
(scan_name, output_filename_prefix)
scannet/batch_load_scannet_data.py:24
↓ 1 callersFunctionextract_pc_in_box3d
pc: (N,3), box3d: (8,3)
models/utils/ap_util.py:10
↓ 1 callersFunctionfit_gamma
(arr, a1=2, b1=200, a2=50, b2=50, weight=0.5, step=10, save=None, quiet=True)
fit.py:152
↓ 1 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 Tensor version.
models/ap_helper_pq.py:35
↓ 1 callersFunctionflip_axis_to_depth
(pc)
models/ap_helper_pq.py:49
↓ 1 callersMethodforward_post
(self,src,pos = None)
models/transformer.py:137
↓ 1 callersFunctiongamma_mixture_guide_criterion
(end_points, DATASET_CONFIG, config, **kwargs)
models/utils/gamma_mixture_loss_util.py:130
↓ 1 callersFunctionget_3d_box_tensor
(box_size, heading_angle, center)
utils/box_util.py:235
↓ 1 callersFunctionget_arkit_pc_loss
(end_points, batch_data_unlabeled, config)
models/utils/arkit_loss_util.py:5
↓ 1 callersMethodget_ceiling_and_floor
(self, pred_corners)
models/ap_helper_pq.py:673
↓ 1 callersFunctionget_color2
(p)
models/dump_helper_quad.py:182
↓ 1 callersFunctionget_consistency_loss
Args: end_points: dict { center, size_scores, size_residuals_normalized, sem_cls_scores, flip
models/utils/mean_teacher_consistency_util.py:201
↓ 1 callersFunctionget_current_consistency_weight
(epoch)
train.py:441
↓ 1 callersFunctionget_iou
Calculate the Intersection over Union (IoU) of two 2D bounding boxes. Parameters ---------- bb1 : dict Keys: {'x1', 'x2', 'y
utils/box_util.py:120
↓ 1 callersFunctionget_iou_main
(get_iou_func, args)
utils/eval_det.py:74
↓ 1 callersFunctionget_loader
(args)
eval.py:211
↓ 1 callersFunctionget_loader
(args)
train.py:211
↓ 1 callersFunctionget_loss_teacher
(end_points, ema_end_points, config, query_points_obj_topk=5, pc_loss=False, num_layer=6)
models/loss_helper_pq.py:502
↓ 1 callersFunctionget_next_weak_batch
()
train.py:311
↓ 1 callersFunctionget_quads
(mapping_name, height=2.00, center_z = 1.00)
ARKitScenes/arkitscenes_utils.py:11
↓ 1 callersFunctionget_verts_tensor
(center, width, height, normal_vector)
models/ap_helper_pq.py:298
↓ 1 callersFunctionhuber_loss
Args: error: Torch tensor (d1,d2,...,dk) Returns: loss: Torch tensor (d1,d2,...,dk) x = error = pred - gt or dist(pred,g
utils/nn_distance.py:15
↓ 1 callersMethodimage_summary
Log a list of images.
utils/tf_logger.py:28
↓ 1 callersFunctionin_hull
(p, hull)
models/utils/ap_util.py:4
↓ 1 callersMethodinit_bn_momentum
(self)
models/pq_transformer.py:275
↓ 1 callersMethodinit_weights
(self)
models/pq_transformer.py:269
↓ 1 callersFunctioninitiate_environment
initiate randomness. :param config: :return:
eval.py:136
↓ 1 callersFunctioninitiate_environment
initiate randomness. :param config: :return:
train.py:141
↓ 1 callersFunctionis_valid_mapping_name
(mapping_name)
ARKitScenes/arkitscenes_dataset.py:35
↓ 1 callersMethodjudge
(self, arr)
fit.py:140
↓ 1 callersFunctionmain
(args)
eval.py:296
↓ 1 callersFunctionmain
(args)
train.py:347
↓ 1 callersFunctionmain
()
scannet/load_scannet_data.py:129
↓ 1 callersFunctionnms_2d
(boxes, overlap_threshold)
utils/nms.py:13
↓ 1 callersFunctionnms_3d_faster_samecls
(boxes, overlap_threshold, old_type=False)
utils/nms.py:115
↓ 1 callersFunctionnot_door_or_window
(sem_class)
models/loss_helper_pq.py:352
↓ 1 callersFunctionparse_groundtruths
Parse groundtruth labels to OBB parameters. Args: end_points: dict {center_label, heading_class_label, heading_residual_
models/ap_helper_pq.py:221
↓ 1 callersFunctionparse_option
()
eval.py:37
↓ 1 callersFunctionparse_option
()
train.py:42
↓ 1 callersFunctionpoint_cloud_to_image
input is Nx3 points output is imgsize*imgsize*num_sample*3 assumes points are in range [-radius, radius] samples num_sample p
utils/pc_util.py:157
↓ 1 callersFunctionpoint_cloud_to_volume
input is Nx3 points. output is vsize*vsize*vsize assumes points are in range [-radius, radius]
utils/pc_util.py:67
↓ 1 callersFunctionpoint_cloud_to_volume_v2
input is Nx3 points output is vsize*vsize*vsize*num_sample*3 assumes points are in range [-radius, radius] samples num_sample
utils/pc_util.py:107
↓ 1 callersFunctionprocess
()
ARKitScenes/dataset/compute_normal_for_pc.py:45
↓ 1 callersFunctionprocess
()
scannet/compute_normal_for_pc.py:7
↓ 1 callersFunctionpyplot_draw_point_cloud
points is a Nx3 numpy array
utils/pc_util.py:244
↓ 1 callersFunctionquad_point_mixture_metric
(quad_center, normal_vector, quad_size, quad_score, downsampled_pc, downsampled_normal, prefix="last_", config
models/utils/gamma_mixture_loss_util.py:27
↓ 1 callersFunctionread_aggregation
(filename)
scannet/load_scannet_data.py:24
↓ 1 callersFunctionread_segmentation
(filename)
scannet/load_scannet_data.py:43
↓ 1 callersFunctionrepresents_int
if string s represents an int.
scannet/scannet_utils.py:24
↓ 1 callersMethodreset
(self)
models/ap_helper_pq.py:638
↓ 1 callersFunctionrotate_quad
(rectangle, rot_mat)
scannet/model_util_scannet.py:97
↓ 1 callersFunctionroty
Rotation about the y-axis.
utils/box_util.py:185
↓ 1 callersFunctionroty_batch
Rotation about the y-axis. t: (x1,x2,...xn) return: (x1,x2,...,xn,3,3)
utils/box_util.py:201
↓ 1 callersFunctionroty_tensor
(t)
utils/box_util.py:193
↓ 1 callersMethodsame_point
(self,pred,gt)
models/ap_helper_pq.py:647
↓ 1 callersFunctionsample_and_group
Input: npoint: radius: nsample: xyz: input points position data, [B, N, 3] points: input points data, [B,
models/utils/pointnet_util.py:117
↓ 1 callersFunctionsample_and_group_all
Input: xyz: input points position data, [B, N, 3] points: input points data, [B, N, D] Return: new_xyz: sampled point
models/utils/pointnet_util.py:157
↓ 1 callersFunctionsave_transformed_scene
(scan_name)
scannet/viz_transformed_ply.py:11
↓ 1 callersMethodscalar_summary
Log a scalar variable.
utils/tf_logger.py:21
↓ 1 callersFunctionsigmoid
Numpy function for sigmoid
models/ap_helper_pq.py:70
↓ 1 callersMethodsigmoid_cross_entropy_with_logits
PyTorch Implementation for tf.nn.sigmoid_cross_entropy_with_logits: max(x, 0) - x * z + log(1 + exp(-abs(x))) in https://www.
models/utils/losses.py:37
↓ 1 callersFunctionsigmoid_rampup
Exponential rampup from https://arxiv.org/abs/1610.02242
eval.py:356
↓ 1 callersFunctionsigmoid_rampup
Exponential rampup from https://arxiv.org/abs/1610.02242
train.py:444
↓ 1 callersFunctionsoftmax
Numpy function for softmax
models/dump_helper_quad.py:16
↓ 1 callersFunctionsoftmax
Numpy function for softmax
models/dump_helper.py:18
↓ 1 callersMethodstep
(self, epoch=None)
pointnet2/pytorch_utils.py:291
↓ 1 callersFunctiontest_interpolation_grad
()
pointnet2/pointnet2_test.py:18
↓ 1 callersFunctiontrain_one_epoch
(epoch, train_loader, DATASET_CONFIG, CONFIG_DICT, model, criterion, optimizer, scheduler, config)
train.py:456
↓ 1 callersFunctionupdate_ema_variables
(model, ema_model, alpha, global_step)
train.py:435
↓ 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 met
utils/eval_det.py:24
↓ 1 callersFunctionvolume_to_point_cloud
vol is occupancy grid (value = 0 or 1) of size vsize*vsize*vsize return Nx3 numpy array.
utils/pc_util.py:79
↓ 1 callersMethodwith_pos_embed
(self, tensor, pos = None)
models/transformer.py:134
↓ 1 callersFunctionwrite_lines_as_cylinders
Create lines represented as cylinders connecting pairs of 3D points Args: pcl: (N x 2 x 3 numpy array): N pairs of xyz pos
utils/pc_util.py:533
↓ 1 callersFunctionwrite_oriented_bbox_single
(box, out_filename, color=None)
utils/pc_util.py:483
FunctionPYBIND11_MODULE
pointnet2/_ext_src/src/bindings.cpp:11
Method__call__
(self, t)
fit.py:23
Method__call__
(self, t)
fit.py:35
Method__call__
(self, distance, out)
models/utils/distance_util.py:14
Method__call__
(self, idx, distance, out)
models/utils/distance_util.py:37
Method__getitem__
(self, idx, **kwargs)
ARKitScenes/arkitscenes_dataset.py:83
Method__init__
(self, dataloader_A, dataloader_B)
double_loader.py:3
Method__init__
(self,data=1)
double_loader.py:33
← previousnext →101–200 of 370, ranked by callers