SYMBOL INDEX (260 symbols across 29 files)

FILE: data_processing/prepare_cache.py
  function parse_args (line 15) | def parse_args():
  function prepare_dataset (line 45) | def prepare_dataset(video_dir, output_dir, prefix, max_total_videos=20000):

FILE: data_processing/visualize_samples.py
  function init_smplx_model (line 11) | def init_smplx_model():

FILE: dataset/visualize_samples.py
  function init_smplx_model (line 11) | def init_smplx_model():

FILE: lib/datasets/avatar_dataset.py
  function load_pose (line 20) | def load_pose(path):
  function load_npy (line 30) | def load_npy(file_path):
  function load_smpl (line 33) | def load_smpl(path, smpl_type='smpl'):
  class AvatarDataset (line 63) | class AvatarDataset(Dataset):
    method __init__ (line 64) | def __init__(self,
    method load_scenes (line 149) | def load_scenes(self):
    method parse_scene (line 180) | def parse_scene(self, scene_id):
    method __len__ (line 454) | def __len__(self):
    method __getitem__ (line 457) | def __getitem__(self, scene_id):
    method parse_scene_test (line 466) | def parse_scene_test(self, scene_id):
  function gather_imgs (line 583) | def gather_imgs(img_ids, poses, image_paths_or_video, smpl_params, load_...
  function calculate_angle (line 644) | def calculate_angle(vector1, vector2):
  function axis_angle_to_rotation_matrix (line 650) | def axis_angle_to_rotation_matrix(axis_angle):
  function find_front_camera_by_rotation (line 676) | def find_front_camera_by_rotation(poses, global_orient_human):
  function read_frames (line 693) | def read_frames(video_path):
  function prepare_camera (line 711) | def prepare_camera( resolution_x, resolution_y, num_views=24, stides=1):
  function from_video_to_get_ref_smplx (line 769) | def from_video_to_get_ref_smplx(video_path):
  function random_scale_and_crop (line 789) | def random_scale_and_crop(image: torch.Tensor, scale_range=(0.8, 1.2)) -...

FILE: lib/datasets/dataloader.py
  class DataModuleFromConfig (line 19) | class DataModuleFromConfig(pl.LightningDataModule):
    method __init__ (line 20) | def __init__(
    method setup (line 42) | def setup(self, stage):
    method train_dataloader (line 45) | def train_dataloader(self):
    method val_dataloader (line 57) | def val_dataloader(self):
    method test_dataloader (line 68) | def test_dataloader(self):

FILE: lib/humanlrm_wrapper_sa_v1.py
  function get_1d_rotary_pos_embed (line 28) | def get_1d_rotary_pos_embed(
  class FluxPosEmbed (line 81) | class FluxPosEmbed(torch.nn.Module):
    method __init__ (line 83) | def __init__(self, theta: int, axes_dim: [int]):
    method forward (line 88) | def forward(self, ids: torch.Tensor) -> torch.Tensor:
  class SapiensGS_SA_v1 (line 105) | class SapiensGS_SA_v1(pl.LightningModule):
    method __init__ (line 107) | def __init__(
    method get_default_smplx_params (line 276) | def get_default_smplx_params(self):
    method forward_decoder (line 302) | def forward_decoder(self, decoder, code, target_rgbs, cameras,
    method on_fit_start (line 311) | def on_fit_start(self):
    method forward (line 319) | def forward(self, data):
    method forward_image_to_uv (line 376) | def forward_image_to_uv(self, inputs_img, is_training=True):
    method compute_loss (line 410) | def compute_loss(self, render_out):
    method compute_metrics (line 481) | def compute_metrics(self, render_out):
    method new_on_before_optimizer_step (line 513) | def new_on_before_optimizer_step(self):
    method validation_step (line 519) | def validation_step(self, batch, batch_idx):
    method forward_nvPose (line 537) | def forward_nvPose(self, batch, smplx_given):
    method on_validation_epoch_end (line 563) | def on_validation_epoch_end(self): #
    method on_test_start (line 608) | def on_test_start(self):
    method on_test_epoch_end (line 612) | def on_test_epoch_end(self):
    method configure_optimizers (line 643) | def configure_optimizers(self):
    method training_step (line 665) | def training_step(self, batch, batch_idx):
    method test_step (line 701) | def test_step(self, batch, batch_idx):
    method on_test_start (line 732) | def on_test_start(self):
    method on_test_epoch_end (line 736) | def on_test_epoch_end(self):
  function weighted_mse_loss (line 769) | def weighted_mse_loss(render_images, target_images, weights):

FILE: lib/mmutils/initialize.py
  function constant_init (line 3) | def constant_init(module: nn.Module, val: float, bias: float = 0) -> None:
  function xavier_init (line 10) | def xavier_init(module: nn.Module,

FILE: lib/models/decoders/uvmaps_decoder_gender.py
  function ensure_dtype (line 25) | def ensure_dtype(input_tensor, target_dtype=torch.float32):
  class UVNDecoder_gender (line 35) | class UVNDecoder_gender(nn.Module):
    method __init__ (line 44) | def __init__(self,
    method init_weights (line 213) | def init_weights(self):
    method extract_pcd (line 224) | def extract_pcd(self, code, smpl_params, init=False, zeros_hands_off=F...
    method deform_pcd (line 253) | def deform_pcd(self, code, smpl_params, init=False, zeros_hands_off=Fa...
    method _sample_feature (line 280) | def _sample_feature(self,results,):
    method _decode_feature (line 303) | def _decode_feature(self, point_code, init=False):
    method _decode (line 334) | def _decode(self, point_code, init=False):
    method gaussian_render (line 371) | def gaussian_render(self, pcd, sigmas, rgbs, normals, rot, num_scenes,...
    method visualize (line 424) | def visualize(self, code, scene_name, viz_dir, code_range=[-1, 1]):
    method forward (line 434) | def forward(self, code, smpl_params, cameras, num_imgs,
    method forward_render (line 527) | def forward_render(self, code, cameras, num_imgs,
    method forward_testing_time (line 613) | def forward_testing_time(self, code, smpl_params, cameras, num_imgs,

FILE: lib/models/decoders/vit_head.py
  class VitHead (line 6) | class VitHead(nn.Module):
    method __init__ (line 7) | def __init__(self,
    method _make_conv_layers (line 52) | def _make_conv_layers(self, in_channels: int,
    method _make_deconv_layers (line 66) | def _make_deconv_layers(self, in_channels: int,
    method forward (line 91) | def forward(self, inputs):

FILE: lib/models/deformers/fast_snarf/cuda/filter/filter.cpp
  function filter (line 13) | void filter(const torch::Tensor &x, const torch::Tensor &mask, torch::Te...
  function PYBIND11_MODULE (line 18) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: lib/models/deformers/fast_snarf/cuda/fuse_kernel/fuse_cuda.cpp
  function fuse_broyden (line 25) | void fuse_broyden(torch::Tensor &x,
  function PYBIND11_MODULE (line 47) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: lib/models/deformers/fast_snarf/cuda/precompute/precompute.cpp
  function precompute (line 12) | void precompute(const torch::Tensor &voxel_w, const torch::Tensor &tfs, ...
  function PYBIND11_MODULE (line 18) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: lib/models/deformers/fast_snarf/lib/model/deformer_smpl.py
  class ForwardDeformer (line 14) | class ForwardDeformer(torch.nn.Module):
    method __init__ (line 24) | def __init__(self,  **kwargs):
    method forward (line 36) | def forward(self, xd, cond, mask, tfs, eval_mode=False):
    method precompute (line 54) | def precompute(self, tfs):
    method search (line 63) | def search(self, xd, cond, mask, tfs, eval_mode=False):
    method broyden_cuda (line 85) | def broyden_cuda(self,
    method forward_skinning (line 116) | def forward_skinning(self, xc, cond, tfs, mask=None):
    method switch_to_explicit (line 135) | def switch_to_explicit(self,resolution=32,smpl_verts=None, smpl_faces=...
    method update_lbs_voxel (line 189) | def update_lbs_voxel(self):
    method query_sdf_smpl (line 199) | def query_sdf_smpl(self, x, smpl_verts, smpl_faces, smpl_weights):
    method skinning_normal (line 221) | def skinning_normal(self, xc, normal, tfs, cond=None, mask=None, inver...
  function skinning_mask (line 242) | def skinning_mask(x, w, tfs, inverse=False):
  function skinning (line 271) | def skinning(x, w, tfs, inverse=False):
  function fast_inverse (line 300) | def fast_inverse(T):
  function bmv (line 317) | def bmv(m, v):
  function create_voxel_grid (line 320) | def create_voxel_grid(d, h, w, device='cuda'):
  function query_weights_smpl (line 329) | def query_weights_smpl(x, smpl_verts, smpl_weights):

FILE: lib/models/deformers/fast_snarf/lib/model/deformer_smplx.py
  class ForwardDeformer (line 14) | class ForwardDeformer(torch.nn.Module):
    method __init__ (line 24) | def __init__(self,  **kwargs):
    method forward_skinning (line 35) | def forward_skinning(self, xc, shape_offset, pose_offset, cond, tfs, t...
    method switch_to_explicit (line 78) | def switch_to_explicit(self,resolution=32,smpl_verts=None, smpl_faces=...
    method update_lbs_voxel (line 132) | def update_lbs_voxel(self):
    method query_sdf_smpl (line 142) | def query_sdf_smpl(self, x, smpl_verts, smpl_faces, smpl_weights):
    method skinning_normal (line 164) | def skinning_normal(self, xc, normal, tfs, cond=None, mask=None, inver...
  function skinning_mask (line 185) | def skinning_mask(x, w, tfs, inverse=False):
  function skinning (line 214) | def skinning(x, w, tfs, inverse=False):
  function fast_inverse (line 243) | def fast_inverse(T):
  function bmv (line 260) | def bmv(m, v):
  function create_voxel_grid (line 264) | def create_voxel_grid(d, h, w, device='cuda'):
  function query_weights_smpl (line 273) | def query_weights_smpl(x, smpl_verts, smpl_weights):

FILE: lib/models/deformers/smplx/body_models.py
  class SMPL (line 39) | class SMPL(nn.Module):
    method __init__ (line 44) | def __init__(
    method num_betas (line 260) | def num_betas(self):
    method num_expression_coeffs (line 264) | def num_expression_coeffs(self):
    method create_mean_pose (line 267) | def create_mean_pose(self, data_struct) -> Tensor:
    method name (line 270) | def name(self) -> str:
    method reset_params (line 274) | def reset_params(self, **params_dict) -> None:
    method get_num_verts (line 281) | def get_num_verts(self) -> int:
    method get_num_faces (line 284) | def get_num_faces(self) -> int:
    method extra_repr (line 287) | def extra_repr(self) -> str:
    method forward_shape (line 295) | def forward_shape(
    method forward (line 303) | def forward(
  class SMPLH (line 405) | class SMPLH(SMPL):
    method __init__ (line 412) | def __init__(
    method create_mean_pose (line 570) | def create_mean_pose(self, data_struct, flat_hand_mean=False):
    method name (line 582) | def name(self) -> str:
    method extra_repr (line 585) | def extra_repr(self):
    method forward (line 593) | def forward(
  class SMPLX (line 661) | class SMPLX(SMPLH):
    method __init__ (line 677) | def __init__(
    method name (line 859) | def name(self) -> str:
    method num_expression_coeffs (line 863) | def num_expression_coeffs(self):
    method create_mean_pose (line 866) | def create_mean_pose(self, data_struct, flat_hand_mean=False):
    method extra_repr (line 884) | def extra_repr(self):
    method forward (line 892) | def forward(

FILE: lib/models/deformers/smplx/lbs.py
  function find_dynamic_lmk_idx_and_bcoords (line 30) | def find_dynamic_lmk_idx_and_bcoords(
  function vertices2landmarks (line 108) | def vertices2landmarks(
  function lbs (line 152) | def lbs(
  function vertices2joints (line 251) | def vertices2joints(J_regressor: Tensor, vertices: Tensor) -> Tensor:
  function blend_shapes (line 271) | def blend_shapes(betas: Tensor, shape_disps: Tensor) -> Tensor:
  function batch_rodrigues (line 295) | def batch_rodrigues(
  function transform_mat (line 332) | def transform_mat(R: Tensor, t: Tensor) -> Tensor:
  function batch_rigid_transform (line 345) | def batch_rigid_transform(

FILE: lib/models/deformers/smplx/utils.py
  class ModelOutput (line 28) | class ModelOutput:
    method __getitem__ (line 36) | def __getitem__(self, key):
    method get (line 39) | def get(self, key, default=None):
    method __iter__ (line 42) | def __iter__(self):
    method keys (line 45) | def keys(self):
    method values (line 49) | def values(self):
    method items (line 53) | def items(self):
  class SMPLOutput (line 59) | class SMPLOutput(ModelOutput):
  class SMPLHOutput (line 69) | class SMPLHOutput(SMPLOutput):
  class SMPLXOutput (line 76) | class SMPLXOutput(SMPLHOutput):
  function find_joint_kin_chain (line 81) | def find_joint_kin_chain(joint_id, kinematic_tree):
  function to_tensor (line 90) | def to_tensor(
  class Struct (line 99) | class Struct(object):
    method __init__ (line 100) | def __init__(self, **kwargs):
  function to_np (line 105) | def to_np(array, dtype=np.float32):
  function rot_mat_to_euler (line 111) | def rot_mat_to_euler(rot_mats):

FILE: lib/models/deformers/smplx/vertex_joint_selector.py
  class VertexJointSelector (line 29) | class VertexJointSelector(nn.Module):
    method __init__ (line 31) | def __init__(self, vertex_ids=None,
    method forward (line 73) | def forward(self, vertices, joints):

FILE: lib/models/deformers/smplx_deformer_gender.py
  class SMPLXDeformer_gender (line 12) | class SMPLXDeformer_gender(torch.nn.Module):
    method __init__ (line 14) | def __init__(self, gender, is_sub2=False) -> None:
    method initialize (line 68) | def initialize(self):
    method forword_body_model (line 125) | def forword_body_model(self, smpl_params, point_pool=4):
    method prepare_deformer (line 150) | def prepare_deformer(self, smpl_params=None, num_scenes=1, device=None):
    method __call__ (line 215) | def __call__(self, pts_in, rot_in, mask=None, cano=True, offset_gs=Non...

FILE: lib/models/renderers/gau_renderer.py
  function batch_rodrigues (line 9) | def batch_rodrigues(rot_vecs, epsilon = 1e-8):
  function build_scaling_rotation (line 42) | def build_scaling_rotation(s, r, tfs):
  function strip_symmetric (line 54) | def strip_symmetric(sym):
  function strip_lowerdiag (line 57) | def strip_lowerdiag(L):
  function build_covariance_from_scaling_rotation (line 68) | def build_covariance_from_scaling_rotation(scaling, scaling_modifier, ro...
  function build_rotation (line 74) | def build_rotation(r):
  function get_covariance (line 97) | def get_covariance(scaling, rotation, scaling_modifier = 1):
  class GRenderer (line 105) | class GRenderer(nn.Module):
    method __init__ (line 106) | def __init__(self, image_size=256, anti_alias=False, f=5000, near=0.01...
    method prepare (line 126) | def prepare(self, cameras):
    method render_gaussian (line 165) | def render_gaussian(self, means3D, colors_precomp, rotations, opacitie...
  function get_view_matrix (line 195) | def get_view_matrix(R, t):
  function get_proj_yy (line 200) | def get_proj_yy(f, image_size, far, near):
  function get_proj_matrix (line 206) | def get_proj_matrix(fovY,fovX, z_near, z_far, z_sign):
  function get_fov (line 226) | def get_fov(focal, princpt, img_shape):

FILE: lib/models/sapiens/sapiens_wrapper_torchscipt.py
  function pretrain_forward (line 23) | def pretrain_forward(sp_lite, inputs: Tensor, layer_num: int, return_hid...
  class SapiensWrapper_ts (line 63) | class SapiensWrapper_ts(nn.Module):
    method __init__ (line 68) | def __init__(self,
    method forward (line 94) | def forward(self, image, use_my_proces=False, requires_grad=False, out...
    method _freeze (line 139) | def _freeze(self):

FILE: lib/models/transformer_sa/mae_decoder_v3_skip.py
  function get_2d_sincos_pos_embed (line 9) | def get_2d_sincos_pos_embed(embed_dim, grid_size, add_cls_token=False):
  function get_2d_sincos_pos_embed_from_grid (line 37) | def get_2d_sincos_pos_embed_from_grid(embed_dim, grid):
  function get_1d_sincos_pos_embed_from_grid (line 49) | def get_1d_sincos_pos_embed_from_grid(embed_dim, pos):
  class neck_SA_v3_skip (line 71) | class neck_SA_v3_skip(nn.Module):
    method __init__ (line 72) | def __init__(self, patch_size=4, in_chans=32, num_patches=196, embed_d...
    method initialize_weights (line 106) | def initialize_weights(self):
    method _init_weights (line 116) | def _init_weights(self, m):
    method forward_decoder (line 126) | def forward_decoder(self, in_features, ids_restore):
    method forward (line 160) | def forward(self, encoded_latent, ids_restore):

FILE: lib/ops/activation.py
  class _trunc_exp (line 8) | class _trunc_exp(Function):
    method forward (line 11) | def forward(ctx, x):
    method backward (line 18) | def backward(ctx, g):
  class TruncExp (line 26) | class TruncExp(nn.Module):
    method forward (line 29) | def forward(x):

FILE: lib/utils/infer_util.py
  function reset_first_frame_rotation (line 26) | def reset_first_frame_rotation(root_orient, trans):
  function rotation_matrix_to_rodrigues (line 73) | def rotation_matrix_to_rodrigues(rotation_matrices):
  function get_hand_pose_mean (line 82) | def get_hand_pose_mean():
  function load_smplify_json (line 105) | def load_smplify_json(smplx_smplify_path):
  function load_image (line 144) | def load_image(input_path, output_folder, image_frame_ratio=None):
  function prepare_camera (line 191) | def prepare_camera( resolution_x = 640, resolution_y = 640, focal_length...
  function construct_camera (line 239) | def construct_camera(K, cam_list, device='cuda'):
  function get_name_str (line 266) | def get_name_str(name):
  function load_smplx_from_npy (line 272) | def load_smplx_from_npy(smplx_path, device='cuda'):
  function add_root_rotate_to_smplx (line 303) | def add_root_rotate_to_smplx(smpl_tmp, frames_num=180, device='cuda'):
  function load_smplx_from_json (line 328) | def load_smplx_from_json(smplx_path, device='cuda'):
  function get_image_dimensions (line 354) | def get_image_dimensions(input_path):
  function construct_camera_from_motionx (line 358) | def construct_camera_from_motionx(smplx_path, device='cuda'):
  function remove_background (line 385) | def remove_background(image: PIL.Image.Image,
  function resize_foreground (line 399) | def resize_foreground(
  function images_to_video (line 440) | def images_to_video(
  function save_video (line 461) | def save_video(

FILE: lib/utils/mesh.py
  function dot (line 7) | def dot(x, y):
  function length (line 11) | def length(x, eps=1e-20):
  function safe_normalize (line 15) | def safe_normalize(x, eps=1e-20):
  class Mesh (line 18) | class Mesh:
    method __init__ (line 19) | def __init__(
    method load (line 47) | def load(cls, path=None, resize=True, renormal=True, retex=False, fron...
    method load_obj (line 100) | def load_obj(cls, path, albedo_path=None, device=None):
    method load_trimesh (line 238) | def load_trimesh(cls, path, device=None):
    method aabb (line 325) | def aabb(self):
    method auto_size (line 330) | def auto_size(self):
    method auto_normal (line 336) | def auto_normal(self):
    method auto_uv (line 359) | def auto_uv(self, cache_path=None, vmap=True):
    method align_v_to_vt (line 392) | def align_v_to_vt(self, vmapping=None):
    method to (line 407) | def to(self, device):
    method write (line 415) | def write(self, path):
    method write_ply (line 426) | def write_ply(self, path):
    method write_glb (line 435) | def write_glb(self, path):
    method write_obj (line 568) | def write_obj(self, path):

FILE: lib/utils/mesh_utils.py
  function gaussian_3d_coeff (line 6) | def gaussian_3d_coeff(xyzs, covs):
  function poisson_mesh_reconstruction (line 27) | def poisson_mesh_reconstruction(points, normals=None):
  function decimate_mesh (line 67) | def decimate_mesh(
  function clean_mesh (line 111) | def clean_mesh(

FILE: lib/utils/train_util.py
  function main_print (line 7) | def main_print(*args):
  function count_params (line 12) | def count_params(model, verbose=False):
  function instantiate_from_config (line 19) | def instantiate_from_config(config):
  function get_obj_from_str (line 29) | def get_obj_from_str(string, reload=False):

FILE: run_demo.py
  function parse_args (line 20) | def parse_args():
  function process_data_on_gpu (line 44) | def process_data_on_gpu(args, model, gpu_id, img_paths_list, smplx_ref_p...
  function setup_directories (line 237) | def setup_directories(base_path, config_name):
  function main (line 248) | def main():

FILE: train.py
  function get_parser (line 34) | def get_parser(**parser_kwargs):
  class SetupCallback (line 134) | class SetupCallback(Callback):
    method __init__ (line 135) | def __init__(self, resume, logdir, ckptdir, cfgdir, config):
    method on_fit_start (line 143) | def on_fit_start(self, trainer, pl_module):
  class CodeSnapshot (line 156) | class CodeSnapshot(Callback):
    method __init__ (line 160) | def __init__(self, savedir, exclude_patterns=None):
    method get_file_list (line 167) | def get_file_list(self):
    method _match_pattern (line 194) | def _match_pattern(self, file_path, pattern):
    method save_code_snapshot (line 210) | def save_code_snapshot(self):
    method on_fit_start (line 218) | def on_fit_start(self, trainer, pl_module):
  function load_folder_ckpt (line 428) | def load_folder_ckpt(checkpoint_dir):