MCPcopy Create free account

hub / github.com/3DAnimals/3DAnimals / functions

Functions685 in github.com/3DAnimals/3DAnimals

⨍ Functions 685◇ Types & classes 122
↓ 133 callersMethoddetach
(self)
model/render/mesh.py:89
↓ 107 callersFunctionmake_cuda_tensor
model/render/renderutils/c_src/torch_bindings.cpp:116
↓ 83 callersMethodto
(self, device)
model/geometry/dmtet.py:52
↓ 79 callersMethodget_predictor
(self, name)
model/models/AnimalModel.py:121
↓ 75 callersMethodclone
(self)
model/render/mesh.py:69
↓ 54 callersMethodappend
(self, dataset, metric)
model/utils/meters.py:56
↓ 49 callersMethodget
(self)
model/utils/meters.py:23
↓ 42 callersFunctionrelative_loss
(name, ref, cuda)
model/render/renderutils/tests/test_bsdf.py:20
↓ 41 callersMethodrender
(self, render_modes, shape, texture, mvp, w2c, campos, resolution, background=None, im_features=None, light=No
model/models/AnimalModel.py:218
↓ 35 callersMethodupdate
(self, prediction=None, ground_truth=None)
model/utils/meters.py:90
↓ 32 callersMethoditems
(self)
model/predictors/InstancePredictorFauna4D.py:183
↓ 31 callersFunctionexpandBF
(x, b, f)
model/models/AnimalModel.py:754
↓ 28 callersMethodbackward
(ctx, dout)
model/render/renderutils/ops.py:511
↓ 27 callersFunction_get_plugin
()
model/render/renderutils/ops.py:23
↓ 27 callersFunctiongetLaunchGridSize
model/render/renderutils/c_src/common.cpp:65
↓ 25 callersFunctiongetLaunchBlockSize
model/render/renderutils/c_src/common.cpp:18
↓ 24 callersFunctionupdate_grid
model/render/renderutils/c_src/torch_bindings.cpp:100
↓ 23 callersFunctioncollapseBF
(x)
model/models/AnimalModel.py:750
↓ 22 callersMethodkeys
(self)
model/render/material.py:47
↓ 18 callersMethodadd_histogram
(self, tag, values, global_step=None)
model/utils/wandb_writer.py:45
↓ 15 callersMethodTensor
model/render/renderutils/c_src/tensor.h:29
↓ 14 callersFunction_dot
(x, y)
model/render/renderutils/bsdf.py:19
↓ 12 callersFunctionsave_images
(images, mask_pred, base_names, out_dir, suffix="", mode="transparent")
visualization/visualize_results.py:41
↓ 12 callersFunctionsave_images
(images, mask_pred, base_names, out_dir, suffix="", mode="transparent")
visualization/visualize_results_fauna.py:42
↓ 12 callersFunctionskinning
model/geometry/skinning.py:369
↓ 12 callersMethodto
(self, device)
model/models/Fauna.py:71
↓ 12 callersMethodto
(self, device)
model/models/AnimalModel.py:159
↓ 11 callersMethodget_camera_extrinsics_from_pose
(self, pose, znear=0.1, zfar=1000., offset_extra=None)
model/predictors/InstancePredictorBase.py:606
↓ 10 callersFunctioneuler_angles_to_matrix
[Borrowed from PyTorch3D] Convert rotations given as Euler angles in radians to rotation matrices. Args: euler_angles: Euler angles
model/geometry/skinning.py:315
↓ 10 callersFunctionin_range
(x, range)
model/models/AnimalModel.py:746
↓ 9 callersFunction_safe_normalize
(x)
model/render/renderutils/bsdf.py:25
↓ 9 callersMethodload
Load the metrics trace from the specified JSON file.
model/utils/meters.py:61
↓ 9 callersFunctionmake_mesh
Create new Mesh class with given verts, faces, uvs, and uv_idx. Args: verts: tensor of shape (B, V, 3) faces: tensor of shap
model/render/mesh.py:355
↓ 9 callersMethodsave
Save the metrics trace to the specified JSON file.
model/utils/meters.py:66
↓ 8 callersFunctionestimate_bones
Estimate the position and structure of bones given the mesh vertex positions. Args: seq_shape: a tensor of shape (B, F, V, 3), the b
model/geometry/skinning.py:50
↓ 7 callersMethod_load_ids
(self, path_patterns, loader, transform=None)
model/dataset/SequenceDataset.py:47
↓ 7 callersMethod_load_ids
(self, path_patterns, loader, transform=None)
model/dataset/SingleSequenceDataset.py:43
↓ 7 callersMethodapply_articulation_constraints
(self, articulation_angles, **kwargs)
model/predictors/InstancePredictorBase.py:435
↓ 7 callersMethodgetMips
(self)
model/render/texture.py:84
↓ 7 callersFunctioninterpolate
(attr, rast, attr_idx, rast_db=None)
model/render/render.py:23
↓ 6 callersMethod_load_ids
(self, path, loader, transform=None)
model/dataset/ImageDataset.py:61
↓ 6 callersMethod_load_ids
(self, path, loader, transform=None)
model/dataset/FaunaDataset.py:217
↓ 6 callersMethodload_model_state
(self, cp)
model/models/Fauna.py:51
↓ 6 callersMethodparse_dict_definition
The dict_config is a diction-based configuration with ascending order The key: value is the NUM_ITERATION_WEIGHT_BEGIN: WEIGHT
model/models/Fauna.py:87
↓ 6 callersMethodrender_rotation_frames
(self, render_mode, mesh, texture, light, resolution, background=None, im_features=None, prior_shape=None, num
model/models/AnimalModel.py:666
↓ 6 callersFunctionsave_image
(fn, x : np.ndarray)
model/render/util.py:423
↓ 6 callersFunctionsum
model/render/renderutils/c_src/vec3f.h:38
↓ 6 callersFunctiontest_loss
(loss, tonemapper)
model/render/renderutils/tests/test_loss.py:35
↓ 6 callersFunctionxmkdir
Create directory PATH recursively if it does not exist.
model/utils/misc.py:60
↓ 5 callersMethod__init__
(self, cin, cout, in_size=128, zdim=None, nf=64, activation=None)
model/networks/encoders.py:13
↓ 5 callersMethodadd_scalar
(self, tag, scalar_value, global_step)
model/utils/wandb_writer.py:31
↓ 5 callersMethodbackward
(ctx, dout)
model/render/light.py:33
↓ 5 callersMethoddeform
Create new Mesh class which is obtained by performing the deformation to the self. Args: deformation: tensor with shape
model/render/mesh.py:110
↓ 5 callersMethodforward
(self, input)
model/networks/MLPs.py:30
↓ 5 callersFunctionget_data_loaders
(cfg: DataLoaderConfig, dataset_split_num=-1)
model/dataloaders.py:34
↓ 5 callersMethodload_tets
(self, grid_res=None, scale=None)
model/geometry/dmtet.py:214
↓ 5 callersFunctionrelative_loss
(name, ref, cuda)
model/render/renderutils/tests/test_mesh.py:33
↓ 5 callersMethodrender_bones
(mvp, bones_pred, size=(256, 256), show_legend=False, overlay_img=None)
model/models/AnimalModel.py:705
↓ 5 callersFunctionsave_images
(out_fold, imgs, prefix='', suffix='', fnames=None, ext='.png')
model/utils/misc.py:132
↓ 4 callersMethod__init__
(self, cin, cout, num_layers, nf=256, dropout=0, activation=None)
model/networks/MLPs.py:10
↓ 4 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/networks/ArticulationNetwork.py:99
↓ 4 callersMethodadd_video
(self, tag, vid_tensor, global_step=None, fps=4)
model/utils/wandb_writer.py:40
↓ 4 callersFunctionbsdf_fresnel_shlick
(f0, f90, cosTheta)
model/render/renderutils/bsdf.py:96
↓ 4 callersFunctioncompute_distance_transform
(mask)
model/dataset/util.py:12
↓ 4 callersMethodcompute_reconstruction_losses
( self, image_pred, image_gt, mask_pred, mask_gt, mask_dt, mask_valid, flow_pred, flow_gt, dino_feat_i
model/models/Ponymation.py:87
↓ 4 callersFunctiondot
(x: torch.Tensor, y: torch.Tensor)
model/render/util.py:22
↓ 4 callersMethodextend
Create new Mesh class which contains each input mesh N times. Args: N: number of new copies of each mesh. Retur
model/render/mesh.py:92
↓ 4 callersFunctiongetWarpSize
model/render/renderutils/c_src/common.h:29
↓ 4 callersFunctionget_activation
(name, inplace=True, lrelu_param=0.2)
model/networks/util.py:4
↓ 4 callersMethodget_bones
Get estimated bones from prior shape and corresponding bone feature and encoding
model/predictors/InstancePredictorBase.py:316
↓ 4 callersMethodget_metric
(self)
evaluation/evaluate.py:451
↓ 4 callersMethodget_sdf
(self, pts=None, total_iter=0, feats=None)
model/geometry/dmtet.py:228
↓ 4 callersFunctionget_valid_mask
(boxs, image_size)
model/dataset/util.py:94
↓ 4 callersFunctionimages_to_video
(image_files, output_path, fps=10)
model/utils/misc.py:296
↓ 4 callersFunctionin_range
(x, range)
model/predictors/InstancePredictorBase.py:702
↓ 4 callersMethodplot
Plots and optionally save as PDF the metrics trace.
model/utils/meters.py:74
↓ 4 callersFunctionplot_points
(points, image, visible=None, correct=None)
evaluation/evaluate.py:66
↓ 4 callersFunctionsave_txts
(arrays, base_names, out_dir, suffix="")
visualization/visualize_results.py:34
↓ 4 callersFunctionsave_txts
(arrays, base_names, out_dir, suffix="")
visualization/visualize_results_fauna.py:35
↓ 4 callersMethodset_eval
(self)
model/models/Fauna.py:79
↓ 4 callersMethodset_train
(self)
model/models/Fauna.py:75
↓ 4 callersMethodtrain
Perform training.
model/Trainer.py:148
↓ 4 callersFunctionvalidate_all_to_device
(batch, device=None)
model/utils/misc.py:319
↓ 3 callersMethod__init__
( self, njoints, nfeats, latent_dim=256, ff_size=1024, num_layers=4, num_heads=4, dropout=0.1, activat
model/networks/MotionVAE.py:26
↓ 3 callersMethod_get_hook
generate a hook method for a specific block and facet.
model/networks/encoders.py:181
↓ 3 callersMethodadd_image
(self, tag, img_tensor, global_step=None, dataformats='CHW')
model/utils/wandb_writer.py:34
↓ 3 callersMethodapply_fauna_articulation_regularizer
(self, articulation_angles, total_iter)
model/predictors/InstancePredictorFauna.py:149
↓ 3 callersFunctionbsdf_lambda_ggx
(alphaSqr, cosTheta)
model/render/renderutils/bsdf.py:105
↓ 3 callersFunctionbuild_model
(cfg)
model/__init__.py:6
↓ 3 callersMethodconvert_keypoints
(self, name, keypoints, inverse=False)
evaluation/evaluate.py:360
↓ 3 callersFunctioncrop_image
(image, boxs, size)
model/dataset/util.py:21
↓ 3 callersMethoddecode
Maps the given latent codes onto the image space. :param z: (Tensor) [B x D] :return: (Tensor) [B x C x H x W]
model/networks/MotionVAE.py:163
↓ 3 callersMethodfirst_n
Create new Mesh class with only the first n meshes in the batch. Args: n: number of meshes to be contained. Ret
model/render/mesh.py:140
↓ 3 callersMethodforward_frozen_ViT
(self, images, dino_enc)
model/predictors/BasePredictorBank.py:106
↓ 3 callersMethodgetMesh
(self, material=None, total_iter=0, jitter_grid=True, feats=None)
model/geometry/dmtet.py:294
↓ 3 callersFunctionget_loader
(mode, data_dir)
model/dataloaders.py:48
↓ 3 callersFunctionget_optimizer
(model, lr=0.0001, betas=(0.9, 0.999), weight_decay=0, eps=1e-8)
model/models/AnimalModel.py:758
↓ 3 callersMethodget_repeat_num
(self, inputs)
model/utils/smooth_loss.py:52
↓ 3 callersFunctionimage_grid
(tensor, nrow=None)
model/utils/misc.py:218
↓ 3 callersFunctionrelative_loss
(name, ref, cuda)
model/render/renderutils/tests/test_loss.py:30
next →1–100 of 685, ranked by callers