Functions — github.com/YunzhuLi/DPI-Net

CodeHub MCPcopy Create free account

hub / github.com/YunzhuLi/DPI-Net / functions

Functions46 in github.com/YunzhuLi/DPI-Net

⨍ Functions 46 ◇ Types & classes 7

↓ 10 callersFunctionload_data

(data_names, path)

↓ 10 callersFunctionrand_float

↓ 9 callersFunctioninit_stat

↓ 5 callersFunctionstore_data

(data_names, data, path)

↓ 4 callersMethod__init__

(self, args, stat, phases_dict, residual=False, use_gpu=False)

↓ 3 callersFunctionnormalize

(data, stat, var=False)

↓ 3 callersFunctionvisualize_neighbors

(anchors, queries, idx, neighbors)

↓ 2 callersFunctioncalc_box_init_FluidShake

(dis_x, dis_z, height, border)

↓ 2 callersFunctioncalc_shape_states_FluidShake

(x_curr, x_last, box_dis, height, border)

↓ 2 callersFunctioncalc_shape_states_RiceGrip

(t, dt, shape_state_dim, gripper_config)

↓ 2 callersFunctioncombine_stat

(stat_0, stat_1)

↓ 2 callersFunctionprepare_input

(data, stat, args, phases_dict, verbose=0, var=False)

↓ 2 callersFunctionrand_int

↓ 1 callersMethodchamfer_distance

↓ 1 callersFunctioncount_parameters

↓ 1 callersFunctiondenormalize

(data, stat, var=False)

↓ 1 callersFunctionfind_relations_neighbor

(positions, query_idx, anchor_idx, radius, order, var=False)

↓ 1 callersMethodgen_data

↓ 1 callersMethodload_data

↓ 1 callersFunctionmake_hierarchy

(env, attr, positions, velocities, idx, st, ed, phases_dict, count_nodes, clusters, verbose=0, var=False)

↓ 1 callersFunctionquatFromAxisAngle

↓ 1 callersFunctionquatFromAxisAngle_var

↓ 1 callersMethodrotation_matrix_from_quaternion

↓ 1 callersFunctionsample_control_RiceGrip

(self, pred, label)

Method__getitem__

(self, input_size, hidden_size, output_size)

(self, input_size, hidden_size, output_size)

(self, input_size, output_size, residual=False)

(self, input_size, hidden_size, output_size)

(self, args, phase, phases_dict, verbose)

Functioncalc_shape_states_FluidShake_var

(x_curr, x_last, box_dis, height, border)

Functioncalc_shape_states_RiceGrip_var

(t, dt, gripper_config)

Functioncollate_fn

Args: x: [n_relations, input_size] Returns: [n_relations, output_size]

Args: x: [n_particles, input_size] Returns: [n_particles, output_size]

Args: x: [n_relations/n_particles, input_size] Returns: [n_relations/n_particles, output_size]

Args: x: [n_particles, input_size] Returns: [n_particles, output_size]

(self, attr, state, Rr, Rs, Ra, n_particles, node_r_idx, node_s_idx, pstep, instance_idx, phas

Functiongen_PyFleX

FunctionrotateByQuat

Functionsample_control_FluidShake

(x_box, time_step, dt)

Functionto_variable

(tensor, use_gpu, requires_grad=False)

Functionvisualize_point_clouds

(point_clouds, c=['b', 'r'], view=None, store=False, store_path='')