MCPcopy Create free account

hub / github.com/alvinliu0/HumanGaussian / functions

Functions965 in github.com/alvinliu0/HumanGaussian

↓ 31 callersMethodinterpolate
( self, attr: Float[Tensor, "B Nv C"], rast: Float[Tensor, "B H W 4"], tri: In
threestudio/utils/rasterize.py:58
↓ 30 callersMethodfrom_pretrained
(cls, name="vgg_lpips")
threestudio/utils/perceptual/perceptual.py:36
↓ 20 callersMethodwrite
(self, msg: str)
threestudio/utils/callbacks.py:134
↓ 19 callersMethodconvert_data
(self, data)
threestudio/utils/saving.py:34
↓ 19 callersMethodget_save_path
(self, filename)
threestudio/utils/saving.py:51
↓ 17 callersFunctionget_activation
(name)
threestudio/utils/ops.py:77
↓ 15 callersMethod__init__
( self, *, ch, out_ch, ch_mult=(1, 2, 4, 8), num_res_blocks,
threestudio/utils/GAN/vae.py:205
↓ 14 callersFunctioncontract_to_unisphere
( x: Float[Tensor, "... 3"], bbox: Float[Tensor, "2 3"], unbounded: bool = False )
threestudio/models/geometry/base.py:20
↓ 13 callersFunctionC
(value: Any, epoch: int, global_step: int)
threestudio/utils/misc.py:65
↓ 13 callersFunctionassign_to_checkpoint
This does the final conversion step: take locally converted weights and apply a global renaming to them. It splits attention layers, and take
scripts/convert_zero123_to_diffusers.py:137
↓ 13 callersFunctionchunk_batch
(func: Callable, chunk_size: int, *args, **kwargs)
threestudio/utils/ops.py:112
↓ 13 callersFunctioncleanup
()
threestudio/utils/misc.py:89
↓ 13 callersFunctionget_mlp
(n_input_dims, n_output_dims, config)
threestudio/models/networks.py:272
↓ 13 callersFunctionread_next_bytes
Read and unpack the next bytes from a binary file. :param fid: :param num_bytes: Sum of combination of {2, 4, 8}, e.g. 2, 6, 16, 30, etc.
gaussiansplatting/scene/colmap_loader.py:72
↓ 13 callersMethodstep
Performs a single optimization step.
threestudio/systems/optimizers.py:104
↓ 12 callersFunctionparse_structured
(fields: Any, cfg: Optional[Union[dict, DictConfig]] = None)
threestudio/utils/config.py:121
↓ 10 callersFunctionconfig_to_primitive
(config, resolve: bool = True)
threestudio/utils/config.py:112
↓ 10 callersFunctionget_rank
()
threestudio/utils/misc.py:17
↓ 10 callersFunctionnonlinearity
(x)
threestudio/utils/GAN/vae.py:35
↓ 10 callersFunctionparse_version
(ver: str)
threestudio/utils/misc.py:13
↓ 9 callersFunctionNormalize
(in_channels, num_groups=32)
threestudio/utils/GAN/vae.py:40
↓ 9 callersMethod__init__
(self, dim_in: int, dim_out: int, config: dict)
threestudio/models/networks.py:151
↓ 9 callersMethodget_text_embeddings
( self, prompt: Union[str, List[str]], negative_prompt: Union[str, List[str]] )
threestudio/models/prompt_processors/base.py:444
↓ 9 callersMethodsave
(self, iteration)
gaussiansplatting/scene/__init__.py:85
↓ 9 callersFunctionshift_azimuth_deg
(azimuth: Float[Tensor, "..."])
threestudio/models/prompt_processors/base.py:173
↓ 9 callersMethodwrite
(self, x)
gaussiansplatting/utils/general_utils.py:118
↓ 8 callersMethoddecode_latents
( self, latents: Float[Tensor, "B 4 H W"], latent_height: int = 64, latent_wid
threestudio/models/guidance/dual_branch_guidance.py:248
↓ 8 callersMethoddo_update_step
( self, epoch: int, global_step: int, on_load_weights: bool = False )
threestudio/utils/base.py:22
↓ 8 callersMethodeval
(self)
threestudio/models/renderers/patch_renderer.py:105
↓ 8 callersFunctionget_ray_directions
Get ray directions for all pixels in camera coordinate. Reference: https://www.scratchapixel.com/lessons/3d-basic-rendering/ r
threestudio/utils/ops.py:179
↓ 7 callersMethodforward_density
(self, points: Float[Tensor, "*N Di"])
threestudio/models/geometry/volume_grid.py:148
↓ 7 callersFunctionget_encoding
(n_input_dims: int, config)
threestudio/models/networks.py:132
↓ 7 callersFunctionmake_attn
(in_channels, attn_type="vanilla")
threestudio/utils/GAN/vae.py:193
↓ 6 callersMethod__init__
(self, dim_in, dim_out)
threestudio/utils/GAN/attention.py:39
↓ 6 callersMethodencode
(self, c)
threestudio/utils/GAN/discriminator.py:114
↓ 6 callersMethodencode_images
( self, imgs: Float[Tensor, "B 3 512 512"] )
threestudio/models/guidance/dual_branch_guidance.py:238
↓ 6 callersMethodextract
(self, args)
gaussiansplatting/arguments/__init__.py:40
↓ 6 callersFunctionfocal2fov
(focal, pixels)
gaussiansplatting/utils/graphics_utils.py:98
↓ 6 callersMethodforward_unet
( self, latents: Float[Tensor, "..."], t: Float[Tensor, "..."], encoder_hidden
threestudio/models/guidance/stable_diffusion_guidance.py:146
↓ 6 callersMethodgeometry
(self)
threestudio/models/renderers/base.py:54
↓ 6 callersFunctionperpendicular_component
(x: Float[Tensor, "B C H W"], y: Float[Tensor, "B C H W"])
threestudio/utils/ops.py:431
↓ 6 callersFunctionrenew_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
scripts/convert_zero123_to_diffusers.py:216
↓ 6 callersFunctionscale_tensor
( dat: Num[Tensor, "... D"], inp_scale: ValidScale, tgt_scale: ValidScale )
threestudio/utils/ops.py:26
↓ 5 callersMethod__init__
(self, inplace=True)
threestudio/utils/GAN/mobilenet.py:34
↓ 5 callersMethodexport
(self, *args, **kwargs)
threestudio/models/geometry/base.py:50
↓ 5 callersMethodforward_sdf
(self, points: Float[Tensor, "*N Di"])
threestudio/models/geometry/implicit_sdf.py:335
↓ 5 callersMethodforward_unet
( self, control_image: Float[Tensor, "..."], latents: Float[Tensor, "..."], t:
threestudio/models/guidance/stable_diffusion_controlnet_guidance.py:161
↓ 5 callersFunctionget_device
()
threestudio/utils/misc.py:28
↓ 5 callersFunctionget_mvp_matrix
( c2w: Float[Tensor, "B 4 4"], proj_mtx: Float[Tensor, "B 4 4"] )
threestudio/utils/ops.py:281
↓ 5 callersFunctionget_rays
( directions: Float[Tensor, "... 3"], c2w: Float[Tensor, "... 4 4"], keepdim=False, noise_scal
threestudio/utils/ops.py:219
↓ 5 callersFunctionhash_prompt
(model: str, prompt: str)
threestudio/models/prompt_processors/base.py:19
↓ 5 callersMethodload_from_cache
(self, prompt)
threestudio/models/prompt_processors/base.py:412
↓ 5 callersMethodload_smplx
(self, path, betas=None, expression=None, gender="neutral")
animation.py:273
↓ 5 callersFunctionmake_divisible
(x, divisible_by=8)
threestudio/utils/GAN/mobilenet.py:78
↓ 5 callersMethodsave_image_grid
( self, filename, imgs, align=DEFAULT_GRID_KWARGS["align"], name: Opti
threestudio/utils/saving.py:301
↓ 5 callersFunctionshave_segments
Removes segments. Positive values shave the first segments, negative shave the last segments.
scripts/convert_zero123_to_diffusers.py:206
↓ 4 callersMethodC
(self, value: Any)
threestudio/systems/base.py:88
↓ 4 callersFunctionSH2RGB
(sh)
gaussiansplatting/utils/sh_utils.py:117
↓ 4 callersMethod__init__
( self, num_features, logdet=False, affine=True, allow_reverse_init=False )
threestudio/utils/GAN/discriminator.py:13
↓ 4 callersMethod__init__
(self)
gaussiansplatting/lpipsPyTorch/modules/networks.py:89
↓ 4 callersMethod_save_rgb_image
( self, filename, img, data_format, data_range, name: Optional
threestudio/utils/saving.py:111
↓ 4 callersMethodbackground
(self)
threestudio/models/renderers/base.py:62
↓ 4 callersMethoddecode_latents
( self, latents: Float[Tensor, "B 4 H W"], latent_height: int = 64, latent_wid
threestudio/models/guidance/stable_diffusion_guidance.py:170
↓ 4 callersMethoddecode_latents
( self, latents: Float[Tensor, "B 4 H W"], latent_height: int = 64, latent_wid
threestudio/models/guidance/stable_diffusion_controlnet_guidance.py:197
↓ 4 callersFunctiondot
(x, y)
threestudio/utils/ops.py:15
↓ 4 callersMethodenable_sequential_cpu_offload
r""" Offloads all models to CPU using accelerate, significantly reducing memory usage. When called, unet, text_encoder, vae and safety
threestudio/models/guidance/models/pipeline_rgbdepth.py:281
↓ 4 callersMethodforward_unet
( self, # control_image: Float[Tensor, "..."], noisy_latents_with_cond: Float[Tensor,
threestudio/models/guidance/dual_branch_guidance.py:202
↓ 4 callersMethodforward_unet
( self, latents: Float[Tensor, "..."], t: Float[Tensor, "..."], encoder_hidden
threestudio/models/guidance/deep_floyd_guidance.py:113
↓ 4 callersFunctionfov2focal
(fov, pixels)
gaussiansplatting/utils/graphics_utils.py:95
↓ 4 callersMethodgeneral_loader
(self, dataset, batch_size, collate_fn=None)
threestudio/data/image.py:320
↓ 4 callersMethodgeneral_loader
(self, dataset, batch_size, collate_fn=None)
threestudio/data/multiview.py:400
↓ 4 callersMethodgeneral_loader
(self, dataset, batch_size, collate_fn=None)
threestudio/data/uncond.py:662
↓ 4 callersMethodgeneral_loader
(self, dataset, batch_size, collate_fn=None)
threestudio/data/co3d.py:691
↓ 4 callersMethodgetTrainCameras
(self, scale=1.0)
gaussiansplatting/scene/__init__.py:89
↓ 4 callersMethodget_trilinear_feature
( self, points: Float[Tensor, "*N Di"], grid: Float[Tensor, "1 Df G1 G2 G3"] )
threestudio/models/geometry/volume_grid.py:65
↓ 4 callersMethodinterpolate_one
( self, attr: Float[Tensor, "Nv C"], rast: Float[Tensor, "B H W 4"], tri: Inte
threestudio/utils/rasterize.py:70
↓ 4 callersMethodmaterial
(self)
threestudio/models/renderers/base.py:58
↓ 4 callersMethodpreprocess_data
(self, batch, stage)
threestudio/systems/base.py:140
↓ 4 callersFunctionrender
Render the scene. Background tensor (bg_color) must be on GPU!
gaussiansplatting/gaussian_renderer/__init__.py:18
↓ 4 callersFunctionrenew_vae_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
scripts/convert_zero123_to_diffusers.py:809
↓ 4 callersMethodsample
(self)
threestudio/utils/GAN/distribution.py:17
↓ 4 callersFunctionshifted_expotional_decay
(a, b, c, r)
threestudio/utils/ops.py:423
↓ 4 callersMethodstep
(self)
animation.py:524
↓ 4 callersFunctionupdate_if_possible
(module: Any, epoch: int, global_step: int)
threestudio/utils/base.py:45
↓ 3 callersFunctionC_max
(value: Any)
threestudio/utils/config.py:31
↓ 3 callersMethod__init__
(self, requires_grad=False, pretrained=True)
threestudio/utils/perceptual/perceptual.py:100
↓ 3 callersMethod__init__
(self, parser: ArgumentParser, name : str, fill_none = False)
gaussiansplatting/arguments/__init__.py:20
↓ 3 callersMethod_isosurface
(self, bbox: Float[Tensor, "2 3"], fine_stage: bool = False)
threestudio/models/geometry/base.py:119
↓ 3 callersMethod_save_grayscale_image
( self, filename, img, data_range, cmap, name: Optional[str] =
threestudio/utils/saving.py:223
↓ 3 callersMethod_unwrap_uv
( self, xatlas_chart_options: dict = {}, xatlas_pack_options: dict = {} )
threestudio/models/mesh.py:207
↓ 3 callersMethodantialias
( self, color: Float[Tensor, "B H W C"], rast: Float[Tensor, "B H W 4"], pos:
threestudio/utils/rasterize.py:49
↓ 3 callersMethodcreate_from_pcd
(self, pcd: BasicPointCloud, spatial_lr_scale: float = 1)
gs_renderer.py:372
↓ 3 callersFunctioncrop_around_box
(tensor, bbox, impath="")
threestudio/data/co3d.py:100
↓ 3 callersFunctiondefault
(val, d)
threestudio/utils/GAN/attention.py:20
↓ 3 callersMethodforward_density
(self, points: Float[Tensor, "*N Di"])
threestudio/models/geometry/implicit_volume.py:195
↓ 3 callersFunctionget_projection_matrix
( fovy: Float[Tensor, "B"], aspect_wh: float, near: float, far: float )
threestudio/utils/ops.py:266
↓ 3 callersMethodget_rgb_image_
(self, img, data_format, data_range, rgba=False)
threestudio/utils/saving.py:77
↓ 3 callersMethodget_shifted_sdf
( self, points: Float[Tensor, "*N Di"], sdf: Float[Tensor, "*N 1"] )
threestudio/models/geometry/implicit_sdf.py:224
↓ 3 callersMethodget_text_embeddings_perp_neg
( self, elevation: Float[Tensor, "B"], azimuth: Float[Tensor, "B"], camera_dis
threestudio/models/prompt_processors/base.py:85
↓ 3 callersFunctioninstantiate_from_config
(config)
threestudio/utils/GAN/util.py:78
next →1–100 of 965, ranked by callers