Functions — github.com/NVlabs/ACID

CodeHub MCPcopy Create free account

hub / github.com/NVlabs/ACID / functions

Functions410 in github.com/NVlabs/ACID

⨍ Functions 410◇ Types & classes 36

↓ 1 callersFunctiontransform_points_cam_to_world

(cam_pts, camera_pose)

ACID/src/utils/mentalsim_util.py:38

↓ 1 callersMethodungrasp

PlushSim/scripts/utils.py:282

↓ 1 callersFunctionupconv2x2

(in_channels, out_channels, mode='transpose')

ACID/src/encoder/unet.py:25

↓ 1 callersFunctionupdate_recursive

Update two config dictionaries recursively. Args: dict1 (dict): first dictionary to be updated dict2 (dict): second dictionary w

ACID/src/config.py:43

↓ 1 callersFunctionupdate_reso

Update the defined resolution so that UNet can process. Args: reso (int): defined resolution depth (int): U-Net number of layers

ACID/src/common.py:270

↓ 1 callersFunctionusing_complex

ACID/preprocess/gen_data_contrastive_pairs_flow.py:33

↓ 1 callersFunctionvisualize_pointcloud

r''' Visualizes point cloud data. Args: points (tensor): point data normals (tensor): normal data (if existing) out_file

ACID/src/utils/visualize.py:51

↓ 1 callersFunctionvisualize_voxels

r''' Visualizes voxel data. Args: voxels (tensor): voxel data out_file (string): output file show (bool): whether the plo

ACID/src/utils/visualize.py:25

↓ 1 callersMethodweight_init

ACID/src/encoder/unet.py:214

↓ 1 callersFunctionworld_coord_view_augmentation

ACID/src/utils/mentalsim_util.py:29

MethodPyArrayToCFunc

ACID/src/utils/libmcubes/pywrapper.cpp:82

FunctionPyArray_SafeGet

ACID/src/utils/libmcubes/pyarraymodule.h:42

FunctionPyArray_SafeSet

ACID/src/utils/libmcubes/pyarraymodule.h:83

MethodPythonToCFunc

ACID/src/utils/libmcubes/pywrapper.cpp:11

MethodSymetricMatrix

ACID/src/utils/libsimplify/Simplify.h:254

ACID/src/common.py:339

ACID/src/common.py:359

Calls the transformation. Args: data (dictionary): data dictionary

ACID/src/data/transforms.py:17

Calls the transformation. Args: data (dict): data dictionary

ACID/src/data/transforms.py:42

Calls the transformation. Args: data (dictionary): data dictionary

ACID/src/data/transforms.py:64

PlushSim/scripts/python_app.py:150

Method__getitem__

Returns an item of the dataset. Args: idx (int): ID of data point

ACID/src/data/core.py:87

Method__getitem__

Returns an item of the dataset. Args: idx (int): ID of data point

ACID/src/data/core.py:310

Method__getitem__

Returns an item of the dataset. Args: idx (int): ID of data point

ACID/src/data/core.py:548

(self, data_dir, num_worker_threads, max_queue_size=500, sensor_settings=None)

PlushSim/scripts/writer.py:25

PlushSim/scripts/syntheticdata.py:34

(self, end_effector, stage, eef_default_loc=None, default_speed=1, fingerL=None, fingerR=None)

PlushSim/scripts/utils.py:221

The config variable is a dictionary containing the following entries Args: width (int): Width of the viewport and generated image

PlushSim/scripts/python_app.py:54

(self, SCENE_PATH, PLUSH_ANIMAL_PATH, PLUSH_SCALE=4,

PlushSim/scripts/attic_scene.py:37

(self, size_in, size_out=None, size_h=None)

ACID/src/layers.py:15

(self, basis_function='sin_cos')

ACID/src/common.py:351

(self, checkpoint_dir='./chkpts', **kwargs)

ACID/src/checkpoints.py:15

(self, in_channels, out_channels, pooling=True)

ACID/src/encoder/unet.py:53

(self, in_channels, out_channels, merge_mode='concat', up_mode='transpose')

ACID/src/encoder/unet.py:80

(self, c_dim=128, dim=3, f_dim=9, hidden_dim=128, scatter_type='max', unet_kwargs=None, plan

ACID/src/encoder/pointnet.py:29

(self, flow_root, pair_root, num_points, split="train", transform={}, pos_ratio=2)

ACID/src/data/core.py:63

(self, geom_root, pair_root, num_points, split="train", transform={}, pos_ratio=2)

ACID/src/data/core.py:286

(self, plan_root, category="teddy",transform={})

ACID/src/data/core.py:532

ACID/src/data/transforms.py:14

ACID/src/data/transforms.py:39

ACID/src/data/transforms.py:61

(self, model, points_batch_size=100000, threshold=0.5, refinement_step=0, device=None,

ACID/src/conv_onet/generation.py:38

(self, model, optimizer, cfg, device=None, vis_dir=None, )

ACID/src/conv_onet/training.py:30

(self, dim=3, c_per_dim=128, c_act_dim=128, corr_dim=0, corr_head=True, hid

ACID/src/conv_onet/models/decoder.py:102

(self, obj_encoder, env_encoder, decoder, device=None, env_scale_factor=2.)

ACID/src/conv_onet/models/__init__.py:330

Returns the length of the dataset.

ACID/src/data/core.py:82

Returns the length of the dataset.

ACID/src/data/core.py:305

Returns the length of the dataset.

ACID/src/data/core.py:543

Functionadd_key

Add new keys to the given input Args: base (tensor): inputs new (tensor): new info for the inputs base_name (str): name

ACID/src/common.py:309

Methodadd_physics_material_to_prim

(stage, prim, materialPath)

PlushSim/scripts/attic_scene.py:309

ACID/src/utils/libsimplify/Simplify.h:96

ACID/src/utils/libsimplify/Simplify.h:108

Performs batch matrix inversion. Arguments: b_mat: the batch of matrices that should be inverted

ACID/src/common.py:115

Functionchamfer_distance

Returns the chamfer distance for the sets of points. Args: points1 (numpy array): first point set points2 (numpy array): second

ACID/src/common.py:38

Functionchamfer_distance

(x, y, metric='l2', direction='bi')

ACID/src/utils/common_util.py:317

Functionchamfer_distance

(x, y, metric='l2', direction='bi')

ACID/src/utils/mentalsim_util.py:400

Functionchamfer_distance

(x, y, metric='l2', direction='bi')

ACID/src/utils/plushsim_util.py:335

Functioncollate_pair_fn

ACID/src/data/core.py:46

Functioncollate_remove_none

Collater that puts each data field into a tensor with outer dimension batch size. Args: batch: batch

ACID/src/data/core.py:17

Functioncoord2index

Normalize coordinate to [0, 1] for sliding-window experiments. Corresponds to our 3D model Args: p (tensor): points vol_

ACID/src/common.py:243

Methodcreate_prim

Create a prim, apply specified transforms, apply semantic label and set specified attributes. args: path (str): The path

PlushSim/scripts/python_app.py:325

Functiondecide_total_volume_range

Update the defined resolution so that UNet can process. Args: query_vol_metric (numpy array): query volume size recep_field (int

ACID/src/common.py:285

Methoddock_window

(space, name, location)

PlushSim/scripts/python_app.py:112

Functionembed_tsne

N x D np.array data

ACID/src/utils/common_util.py:19

Methodeval_points

ACID/src/conv_onet/models/__init__.py:281

Methodeval_points

ACID/src/conv_onet/models/__init__.py:380

Performs an evaluation. Args: val_loader (dataloader): pytorch dataloader

ACID/src/conv_onet/training.py:72

Functionexport_mesh

Exports a mesh in the COLLADA (.dae) format. Needs PyCollada (https://github.com/pycollada/pycollada).

ACID/src/utils/libmcubes/exporter.py:35

Functionexport_obj

Exports a mesh in the (.obj) format.

ACID/src/utils/libmcubes/exporter.py:5

Functionexport_off

Exports a mesh in the (.off) format.

ACID/src/utils/libmcubes/exporter.py:19

Functionexport_pointcloud

(vertices, out_file, as_text=True)

ACID/src/utils/io.py:6

Functionexport_quad_obj

(verts, faces, export_path)

PlushSim/scripts/utils.py:389

Functionexport_simulation_voxels

(prim, export_path, binarize=True, export_faces=False)

PlushSim/scripts/utils.py:111

Functionexport_train_data

ACID/preprocess/gen_data_flow_plush.py:38

Functionexport_visual_mesh

(prim, export_path, loc=None, rot=None, binarize=True)

PlushSim/scripts/utils.py:70

Functionextent_to_cube

PlushSim/scripts/utils.py:364

Functionextent_to_cube

ACID/src/utils/common_util.py:175

Functionf1_score

(x, y, metric='l2', th=0.01)

ACID/src/utils/common_util.py:324

Functionf1_score

(x, y, metric='l2', th=0.01)

ACID/src/utils/mentalsim_util.py:407

Functionf1_score

(x, y, metric='l2', th=0.01)

ACID/src/utils/plushsim_util.py:342

Functionfind_nn_cpu

(feat0, feat1, return_distance=False)

ACID/src/utils/plushsim_util.py:361

Functionfind_nn_cpu_symmetry_consistent

(feat0, feat1, pts0, pts1, n_neighbor=10, local_radis=0.05, return_distance=False)

ACID/src/utils/plushsim_util.py:378

Functionfix_Rt_camera

Fixes Rt camera matrix. Args: Rt (tensor): Rt camera matrix loc (tensor): location scale (float): scale

ACID/src/common.py:138

ACID/src/layers.py:39

ACID/src/encoder/unet.py:66

Forward pass Arguments: from_down: tensor from the encoder pathway from_up: upconv'd tensor from the decoder pathway

ACID/src/encoder/unet.py:101

ACID/src/encoder/unet.py:225

ACID/src/encoder/pointnet.py:88

(self, p, c_plane, **kwargs)

ACID/src/conv_onet/models/decoder.py:61

(self, p, c_per_plane, c_act_plane)

ACID/src/conv_onet/models/decoder.py:222

Performs a forward pass through the network. Args: p (tensor): sampled points inputs (tensor): conditioning input

ACID/src/conv_onet/models/__init__.py:29

Performs a forward pass through the network. Args: p (tensor): sampled points inputs (tensor): conditioning input

ACID/src/conv_onet/models/__init__.py:343

Methodforward_perception

(self, inputs, filter=True,)

ACID/src/conv_onet/models/__init__.py:41

ACID/src/utils/libsimplify/Simplify.h:157

Methodgenerate_occ_grid

Generates mesh from latent. Works for shapes normalized to a unit cube Args: c (tensor): latent conditioned code c

ACID/src/conv_onet/generation.py:360

Functionget_action_info

(int_group, int_id, data_root)

ACID/src/utils/mentalsim_util.py:229

Functionget_action_info

(model_category, model_name, split_id, reset_id, interaction_id, data_root)

ACID/src/utils/plushsim_util.py:188

Methodget_camera_params

Get active camera intrinsic and extrinsic parameters. Returns: A dict of the active camera's parameters. pose (numpy

PlushSim/scripts/syntheticdata.py:103

Functionget_color_map

ACID/src/utils/common_util.py:15

Functionget_end_frame

(int_group, int_id, data_root)

ACID/src/utils/mentalsim_util.py:245

← previous next →201–300 of 410, ranked by callers