↓ 1 callersFunctiondecode_scores(base_xyz,objectness_scores, center, heading_scores, heading_residuals_normalized, size_scores, size_residuals
models/pq_transformer.py:35
↓ 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 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 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 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 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 callersFunctiontrain_one_epoch(epoch, train_loader, DATASET_CONFIG, CONFIG_DICT, model, criterion, optimizer, scheduler, config)
train.py:456