↓ 2 callersFunctionbatched_index_select
:param input: [B, N1, *]
:param dim: the dim to be selected
:param index: [B, N2]
:return: [B, N2, *] selected result
Affordance-DrivenHOIDiffusion/lib/utils/proc_output.py:174
↓ 2 callersMethodddpm_loop_loss_guidance(
self, sample_lhand, sample_rhand, sample_obj,
model, obj_feat, enc_none_text,
Affordance-DrivenHOIDiffusion/lib/networks/diffusion copy.py:340
↓ 2 callersFunctionfarthest_point_sample Input: xyz: pointcloud data, [B, N, 3] npoint: number of samples Return: centroids: sampled pointcloud index, [B, npo
HOIAffordanceMLLM/ReConV2/segmentation/models/pointnet2_utils.py:60
↓ 2 callersFunctionfarthest_point_sample Input: xyz: pointcloud data, [B, N, 3] npoint: number of samples Return: centroids: sampled pointcloud index, [B, npo
HOIAffordanceMLLM/llava/model/language_model/affordancellm.py:140
↓ 2 callersMethodforward_7d_batch(
self,
angles,
global_orient,
transl,
query_names: list,
fwd_
Affordance-DrivenHOIDiffusion/lib/models/object_arctic.py:42
↓ 2 callersFunctionget_NN
:param src_xyz: [B, N1, 3]
:param trg_xyz: [B, N2, 3]
:return: nn_dists, nn_dix: all [B, 3000] tensor for NN distance and index in N2
Affordance-DrivenHOIDiffusion/lib/utils/proc_output.py:136
↓ 2 callersFunctionget_distance_map_loss(
pred_ldist, pred_rdist,
targ_ldist, targ_rdist,
weight=None,
)
Affordance-DrivenHOIDiffusion/lib/utils/loss.py:107
↓ 2 callersFunctionget_hand_obj_dist_map(
pred_lhand, pred_rhand, pred_obj, obj_pc,
lhand_layer, rhand_layer, dataset_name, obj_pc_top_idx=N
Affordance-DrivenHOIDiffusion/lib/utils/proc_output.py:106
↓ 2 callersFunctionget_interior
:param src_face_normal: [B, 778, 3], surface normal of every vert in the source mesh
:param src_xyz: [B, 778, 3], source mesh vertices xyz
Affordance-DrivenHOIDiffusion/lib/utils/proc_output.py:154
↓ 2 callersFunctionget_joint_contact_loss(
pred_lhand, pred_rhand,
lhand_obj_cont_v, rhand_obj_cont_v,
lhand_layer, rhand_layer,
mask
Affordance-DrivenHOIDiffusion/lib/utils/loss.py:42
↓ 2 callersMethodget_loss(
self, input_lhand, input_rhand, refined_obj,
targ_lhand, targ_rhand,
obj_verts_
Affordance-DrivenHOIDiffusion/lib/networks/refiner.py:68
↓ 2 callersFunctionget_object_hand_info(
object_model,
clip_model, text,
obj_root, data_config,
mpnet=None,
)
Affordance-DrivenHOIDiffusion/lib/utils/eval.py:120
↓ 2 callersFunctionget_penetration_loss(
pred_X0_lhand, pred_X0_rhand, pred_X0_obj,
lhand_layer, rhand_layer, obj_pc_org,
valid_mask_lh
Affordance-DrivenHOIDiffusion/lib/utils/loss.py:202
↓ 2 callersFunctionget_relative_orientation_loss(
pred_lhand, pred_rhand, pred_obj,
targ_lhand, targ_rhand, targ_obj,
mask_lhand, ma
Affordance-DrivenHOIDiffusion/lib/utils/loss.py:121
↓ 2 callersFunctionknn_point Input: nsample: max sample number in local region xyz: all points, [B, N, C] new_xyz: query points, [B, S, C] Return:
HOIAffordanceMLLM/ReConV2/utils/knn.py:26
↓ 2 callersMethodprepare_inputs_labels_for_multimodal(
self, input_ids, attention_mask, past_key_values, labels, points
)
HOIAffordanceMLLM/llava/model/llava_arch.py:107
↓ 2 callersFunctionproc_obj_feat_final_train(cov_map, obj_scale, obj_cent, obj_feat, use_obj_scale_centroid, use_contact_feat)
Affordance-DrivenHOIDiffusion/lib/utils/proc.py:370
↓ 2 callersFunctionproc_results(
x_lhand, x_rhand, x_obj,
obj_verts, lhand_layer, rhand_layer,
is_lhand, is_rhand,
dataset
Affordance-DrivenHOIDiffusion/lib/utils/eval.py:26