MCPcopy Create free account

hub / github.com/NeuralCarver/Michelangelo / functions

Functions318 in github.com/NeuralCarver/Michelangelo

⨍ Functions 318◇ Types & classes 68
↓ 23 callersFunctioninstantiate_from_config
(config, **kwargs)
michelangelo/utils/misc.py:45
↓ 9 callersFunctioncheckpoint
Evaluate a function without caching intermediate activations, allowing for reduced memory at the expense of extra compute in the backward pas
michelangelo/models/modules/checkpoint.py:10
↓ 8 callersFunctioninit_linear
(l, stddev)
michelangelo/models/modules/transformer_blocks.py:12
↓ 7 callersMethod__init__
(self, *, device: Optional[torch.device], dtype: Optional[torch.dtype],
michelangelo/models/modules/transformer_blocks.py:230
↓ 7 callersMethod__init__
(self, *, device: Optional[torch.device], dtype: Optional[torch.dtype],
michelangelo/models/modules/transformer_vit.py:257
↓ 6 callersMethod__init__
(self, embed_dim, n_classes=1000, key="class")
michelangelo/models/conditional_encoders/encoder_factory.py:24
↓ 5 callersMethod__add_object
(self, obj, parent=None)
michelangelo/utils/visualizers/pythreejs_viewer.py:58
↓ 5 callersMethod__get_shading
(self, shading)
michelangelo/utils/visualizers/pythreejs_viewer.py:39
↓ 5 callersFunctionextract_geometry
Args: geometric_func: device: bounds: octree_depth: batch_size: num_chunks: disable:
michelangelo/models/tsal/inference_utils.py:14
↓ 4 callersMethod__update_view
(self)
michelangelo/utils/visualizers/pythreejs_viewer.py:90
↓ 4 callersFunctionddim_sample
(ddim_scheduler: DDIMScheduler, diffusion_model: torch.nn.Module, shape: Union
michelangelo/models/asl_diffusion/inference_utils.py:12
↓ 4 callersMethoddecode
(self, *args, **kwargs)
michelangelo/models/tsal/tsal_base.py:60
↓ 4 callersMethodencode_kl_embed
(self, latents: torch.FloatTensor, sample_posterior: bool = True)
michelangelo/models/tsal/sal_perceiver.py:383
↓ 4 callersFunctionget_colors
(inp, colormap="viridis", normalize=True, vmin=None, vmax=None)
michelangelo/utils/visualizers/color_util.py:6
↓ 3 callersMethod__init__
(self, transforms)
michelangelo/data/transforms.py:368
↓ 3 callersMethod__init__
( self, *, device: Optional[torch.device], dtype: Optional[tor
michelangelo/models/modules/diffusion_transformer.py:93
↓ 3 callersMethod__init__
(self)
michelangelo/models/tsal/tsal_base.py:9
↓ 3 callersMethod__init__
(self, *, device: Optional[torch.device], dtype: Optional[torch.dtype],
michelangelo/models/tsal/sal_perceiver.py:22
↓ 3 callersMethodencode
(self, *args, **kwargs)
michelangelo/models/tsal/tsal_base.py:57
↓ 3 callersMethodkl
(self, other=None, dims=(1, 2, 3))
michelangelo/models/modules/distributions.py:47
↓ 2 callersMethod__add_line_geometry
(self, lines, shading, obj=None)
michelangelo/utils/visualizers/pythreejs_viewer.py:73
↓ 2 callersMethod__get_colors
(self, v, f, c, sh)
michelangelo/utils/visualizers/pythreejs_viewer.py:123
↓ 2 callersMethod__init__
The initialization
michelangelo/models/modules/embedder.py:48
↓ 2 callersMethod__init__
(self, near_weight: float = 0.1, kl_weight: float = 1.0, nu
michelangelo/models/tsal/loss.py:14
↓ 2 callersFunction_add_to_buf_dict
(buf_dict, sample)
michelangelo/data/transforms.py:34
↓ 2 callersFunction_uid_buffer_pick
(buf_dict, rng)
michelangelo/data/transforms.py:18
↓ 2 callersMethodcompute_loss
Args: model_outputs (dict): - x_0: - noise: - noise_prior: - noi
michelangelo/models/asl_diffusion/clip_asl_diffuser_pl_module.py:188
↓ 2 callersMethodcompute_loss
Args: model_outputs (dict): - x_0: - noise: - noise_prior: - noi
michelangelo/models/asl_diffusion/asl_diffuser_pl_module.py:261
↓ 2 callersMethoddecode
(self, latents: torch.FloatTensor)
michelangelo/models/tsal/sal_perceiver.py:273
↓ 2 callersMethoddecode
(self, z_q, bounds: Union[Tuple[float], List[float], float] = 1.1,
michelangelo/models/tsal/asl_pl_module.py:146
↓ 2 callersMethoddecode_first_stage
(self, z_q: torch.FloatTensor, **kwargs)
michelangelo/models/asl_diffusion/clip_asl_diffuser_pl_module.py:164
↓ 2 callersMethoddecode_first_stage
(self, z_q: torch.FloatTensor, **kwargs)
michelangelo/models/asl_diffusion/asl_diffuser_pl_module.py:237
↓ 2 callersMethodencode_first_stage
(self, surface: torch.FloatTensor, sample_posterior=True)
michelangelo/models/asl_diffusion/clip_asl_diffuser_pl_module.py:156
↓ 2 callersMethodencode_first_stage
(self, surface: torch.FloatTensor, sample_posterior=True)
michelangelo/models/asl_diffusion/asl_diffuser_pl_module.py:229
↓ 2 callersMethodencode_image
(self, images: Iterable[Optional[ImageType]])
michelangelo/models/conditional_encoders/clip.py:39
↓ 2 callersMethodencode_latents
(self, pc: torch.FloatTensor, feats: Optional[torch.FloatTensor]
michelangelo/models/tsal/sal_perceiver.py:372
↓ 2 callersMethodencode_shape_embed
(self, *args, **kwargs)
michelangelo/models/tsal/tsal_base.py:102
↓ 2 callersMethodexport
(self, fname)
michelangelo/graphics/primitives/mesh.py:90
↓ 2 callersFunctionget_obj_from_str
(string, reload=False)
michelangelo/utils/misc.py:30
↓ 2 callersMethodinstantiate_first_stage
(self, config)
michelangelo/models/asl_diffusion/asl_diffuser_pl_module.py:104
↓ 2 callersMethodinstantiate_non_trainable_model
(self, config)
michelangelo/models/asl_diffusion/clip_asl_diffuser_pl_module.py:89
↓ 2 callersFunctionis_dist_avail_and_initialized
()
michelangelo/utils/misc.py:60
↓ 2 callersMethodmode
(self)
michelangelo/models/modules/distributions.py:69
↓ 2 callersFunctionnorm_cdf
(x)
michelangelo/models/modules/transformer_vit.py:15
↓ 2 callersMethodquery_geometry
(self, queries: torch.FloatTensor, latents: torch.FloatTensor)
michelangelo/models/tsal/sal_perceiver.py:277
↓ 2 callersMethodsample
(self)
michelangelo/models/modules/distributions.py:18
↓ 2 callersMethodsample
(self)
michelangelo/models/modules/distributions.py:43
↓ 2 callersMethodsave
(self, filename="")
michelangelo/utils/visualizers/pythreejs_viewer.py:526
↓ 2 callersFunctionsave_output
(args, mesh_outputs)
inference.py:64
↓ 1 callersMethod__get_bbox
(self, v)
michelangelo/utils/visualizers/pythreejs_viewer.py:111
↓ 1 callersMethod__get_point_colors
(self, v, c, sh)
michelangelo/utils/visualizers/pythreejs_viewer.py:171
↓ 1 callersMethod__update_settings
(self, settings={})
michelangelo/utils/visualizers/pythreejs_viewer.py:51
↓ 1 callersFunction_trunc_normal_
(tensor, mean, std, a, b)
michelangelo/models/modules/transformer_vit.py:12
↓ 1 callersMethodadd_edges
(self, vertices, edges, shading={}, obj=None, **kwargs)
michelangelo/utils/visualizers/pythreejs_viewer.py:329
↓ 1 callersMethodclip
(self)
michelangelo/models/conditional_encoders/encoder_factory.py:68
↓ 1 callersMethodclip
(self)
michelangelo/models/conditional_encoders/encoder_factory.py:141
↓ 1 callersFunctioncompute_psnr
(x, y, data_range: float = 2, eps: float = 1e-7)
michelangelo/utils/eval.py:6
↓ 1 callersMethodcontain_uv_texture
(self)
michelangelo/graphics/primitives/mesh.py:84
↓ 1 callersMethodcontain_vertex_colors
(self)
michelangelo/graphics/primitives/mesh.py:87
↓ 1 callersMethoddecode
(self, z_q, bounds: Union[Tuple[float], List[float], float] = 1.1,
michelangelo/models/tsal/sal_pl_module.py:119
↓ 1 callersMethodencode
Args: pc (torch.FloatTensor): [B, N, 3] feats (torch.FloatTensor or None): [B, N, C] sample_posterior (b
michelangelo/models/tsal/sal_perceiver.py:242
↓ 1 callersMethodencode
Args: pc (torch.FloatTensor): [B, N, 3] feats (torch.FloatTensor or None): [B, N, c] sample_posterior (b
michelangelo/models/tsal/sal_perceiver.py:350
↓ 1 callersMethodencode
(self, surface: torch.FloatTensor, sample_posterior=True)
michelangelo/models/tsal/sal_pl_module.py:108
↓ 1 callersMethodencode_image_embed
Args: image (torch.FloatTensor): [bs, 3, h, w] Returns: x (torch.FloatTensor): [bs, projection_dim]
michelangelo/models/tsal/clip_asl_module.py:53
↓ 1 callersMethodencode_image_embed
(self, *args, **kwargs)
michelangelo/models/tsal/tsal_base.py:96
↓ 1 callersMethodencode_shape_embed
Args: surface (torch.FloatTensor): [bs, n, 3 + c] return_latents (bool): Returns: x (torch.Floa
michelangelo/models/tsal/clip_asl_module.py:30
↓ 1 callersMethodencode_text
(self, texts: List[str])
michelangelo/models/conditional_encoders/clip.py:56
↓ 1 callersMethodencode_text_embed
(self, text)
michelangelo/models/tsal/clip_asl_module.py:67
↓ 1 callersMethodencode_text_embed
(self, *args, **kwargs)
michelangelo/models/tsal/tsal_base.py:99
↓ 1 callersMethodforward
(self, x, t, context)
michelangelo/models/asl_diffusion/base.py:12
↓ 1 callersMethodforward
(self, text)
michelangelo/models/conditional_encoders/encoder_factory.py:83
↓ 1 callersMethodforward
(self, text)
michelangelo/models/conditional_encoders/encoder_factory.py:156
↓ 1 callersFunctiongen_checkers
(n_checkers_x, n_checkers_y, width=256, height=256)
michelangelo/utils/visualizers/color_util.py:16
↓ 1 callersFunctiongen_circle
(width=256, height=256)
michelangelo/utils/visualizers/color_util.py:34
↓ 1 callersFunctiongenerate_dense_grid_points
(bbox_min: np.ndarray, bbox_max: np.ndarray, oct
michelangelo/graphics/primitives/volume.py:6
↓ 1 callersFunctionget_config_from_file
(config_file: str)
michelangelo/utils/misc.py:11
↓ 1 callersMethodget_dims
(self, input_dim)
michelangelo/models/modules/embedder.py:81
↓ 1 callersFunctionget_world_size
()
michelangelo/utils/misc.py:74
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=())
michelangelo/models/tsal/asl_pl_module.py:68
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=())
michelangelo/models/tsal/sal_pl_module.py:57
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=())
michelangelo/models/asl_diffusion/clip_asl_diffuser_pl_module.py:105
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=())
michelangelo/models/asl_diffusion/asl_diffuser_pl_module.py:134
↓ 1 callersMethodinstantiate_cond_stage
(self, cond_stage_config)
michelangelo/models/asl_diffusion/clip_asl_diffuser_pl_module.py:102
↓ 1 callersMethodinstantiate_first_stage
(self, first_stage_config)
michelangelo/models/asl_diffusion/clip_asl_diffuser_pl_module.py:98
↓ 1 callersMethodlatent2mesh
Args: latents: [bs, num_latents, dim] bounds: octree_depth: num_chunks: Returns:
michelangelo/models/tsal/asl_pl_module.py:306
↓ 1 callersMethodlatent2mesh
Args: latents: [bs, num_latents, dim] bounds: octree_depth: num_chunks: Returns:
michelangelo/models/tsal/sal_pl_module.py:243
↓ 1 callersFunctionload_model
(args)
inference.py:24
↓ 1 callersFunctionload_surface
(fp)
inference.py:36
↓ 1 callersMethodmove
(self)
michelangelo/models/conditional_encoders/encoder_factory.py:71
↓ 1 callersMethodmove
(self)
michelangelo/models/conditional_encoders/encoder_factory.py:144
↓ 1 callersMethodmove
(self)
michelangelo/models/conditional_encoders/encoder_factory.py:287
↓ 1 callersMethodmove
(self)
michelangelo/models/conditional_encoders/encoder_factory.py:341
↓ 1 callersMethodmove
(self)
michelangelo/models/conditional_encoders/encoder_factory.py:477
↓ 1 callersFunctionprepare_image
(args, number_samples=2)
inference.py:51
↓ 1 callersMethodquery_geometry
(self, *args, **kwargs)
michelangelo/models/tsal/tsal_base.py:63
↓ 1 callersMethodrandom_resize_params
(self)
michelangelo/data/transforms.py:313
↓ 1 callersFunctionsave_obj
(pointnp_px3, facenp_fx3, fname)
michelangelo/graphics/primitives/mesh.py:12
↓ 1 callersFunctionsavemeshtes2
(pointnp_px3, tcoords_px2, facenp_fx3, facetex_fx3, tex_map, fname)
michelangelo/graphics/primitives/mesh.py:27
↓ 1 callersMethodset_shape_model_only
(self)
michelangelo/models/tsal/tsal_base.py:93
↓ 1 callersMethodto_html
(self, imports=True, html_frame=True)
michelangelo/utils/visualizers/pythreejs_viewer.py:471
next →1–100 of 318, ranked by callers