MCPcopy Create free account

hub / github.com/TencentARC/InstantMesh / functions

Functions275 in github.com/TencentARC/InstantMesh

↓ 1 callersFunctionmarching_tets
(pos_nx3, sdf_n, tet_fx4, triangle_table, num_triangles_table, base_tet_edges, v_id)
src/models/geometry/rep_3d/dmtet.py:60
↓ 1 callersFunctionmarching_tets_tetmesh
( pos_nx3, sdf_n, tet_fx4, triangle_table, num_triangles_table, base_tet_edges, v_id, return_t
src/models/geometry/rep_3d/dmtet.py:129
↓ 1 callersFunctionpad_camera_extrinsics_4x4
(extrinsics)
src/utils/camera_util.py:6
↓ 1 callersMethodpredict_start_from_z_and_v
(self, x_t, t, v)
zero123plus/model.py:168
↓ 1 callersMethodprepare_batch_data
(self, batch)
src/model_mesh.py:80
↓ 1 callersMethodprepare_batch_data
(self, batch)
src/model.py:42
↓ 1 callersMethodprepare_validation_batch_data
(self, batch)
src/model_mesh.py:139
↓ 1 callersMethodprepare_validation_batch_data
(self, batch)
src/model.py:104
↓ 1 callersMethodproject
(self, points_bxnx4)
src/models/geometry/camera/perspective_camera.py:31
↓ 1 callersFunctionproject_onto_planes
Does a projection of a 3D point onto a batch of 2D planes, returning 2D plane coordinates. Takes plane axes of shape n_planes, 3, 3
src/models/renderer/utils/renderer.py:46
↓ 1 callersFunctionprojection
(x=0.1, n=1.0, f=50.0, near_plane=None)
src/models/geometry/camera/perspective_camera.py:14
↓ 1 callersMethodprune_heads
(self, heads: Set[int])
src/models/encoder/dino.py:238
↓ 1 callersMethodregister_schedule
(self)
zero123plus/model.py:73
↓ 1 callersFunctionrender_frames
Render frames from triplanes.
run.py:41
↓ 1 callersMethodrender_mesh
Function to render a generated mesh with nvdiffrast :param mesh_v: List of vertices for the mesh :param mesh_f: List of faces
src/models/lrm_mesh.py:238
↓ 1 callersMethodrender_mesh
(self, mesh_v_nx3, mesh_f_fx3, camera_mv_bx4x4, resolution=256, hierarchical_mask=False)
src/models/geometry/rep_3d/dmtet.py:459
↓ 1 callersMethodrender_mesh
(self, mesh_v_nx3, mesh_f_fx3, camera_mv_bx4x4, resolution=256, hierarchical_mask=False)
src/models/geometry/rep_3d/flexicubes_geometry.py:74
↓ 1 callersMethodrun_model_activated
(self, planes, decoder, sample_coordinates, sample_directions, options)
src/models/renderer/utils/renderer.py:206
↓ 1 callersMethodsample_importance
Return depths of importance sampled points along rays. See NeRF importance sampling for more.
src/models/renderer/utils/renderer.py:264
↓ 1 callersMethodsample_pdf
Sample @N_importance samples from @bins with distribution defined by @weights. Inputs: bins: (N_rays, N_samples_+1) where
src/models/renderer/utils/renderer.py:284
↓ 1 callersMethodsave_code_snapshot
(self)
train.py:133
↓ 1 callersFunctionsave_glb
(pointnp_px3, facenp_fx3, colornp_px3, fpath)
src/utils/mesh_util.py:31
↓ 1 callersFunctionsave_obj_with_mtl
(pointnp_px3, tcoords_px2, facenp_fx3, facetex_fx3, texmap_hxwx3, fname)
src/utils/mesh_util.py:43
↓ 1 callersFunctionsave_video
( frames: torch.Tensor, output_path: str, fps: int = 30, )
src/utils/infer_util.py:87
↓ 1 callersFunctionscale_latents
(latents)
zero123plus/model.py:17
↓ 1 callersFunctionsdf_reg_loss_batch
(sdf, all_edges)
src/model_mesh.py:15
↓ 1 callersMethodsetup
(self, stage)
src/data/objaverse.py:52
↓ 1 callersFunctionspherical_camera_pose
(azimuths: np.ndarray, elevations: np.ndarray, radius=2.5)
src/utils/camera_util.py:47
↓ 1 callersFunctiontet_to_tet_adj_sparse
(tet_tx4)
src/models/geometry/rep_3d/dmtet.py:272
↓ 1 callersMethodunify_samples
(self, depths1, colors1, densities1, depths2, colors2, densities2, normals1=None, normals2=None)
src/models/renderer/utils/renderer.py:218
↓ 1 callersFunctionunscale_image
(image)
zero123plus/pipeline.py:171
↓ 1 callersFunctionunscale_latents
(latents)
zero123plus/pipeline.py:161
↓ 1 callersFunctionunscale_latents
(latents)
zero123plus/model.py:22
↓ 1 callersFunctionxfm_points
Transform points. Args: points: Tensor containing 3D points with shape [minibatch_size, num_vertices, 3] or [1, num_vertices, 3] m
src/models/geometry/render/neural_render.py:23
Method__call__
( self, attn: Attention, hidden_states, encoder_hidden_states=None, attention_mask=None, mode=
zero123plus/pipeline.py:55
Method__call__
( self, image: Image.Image = None, prompt = "", *args, num_images_per_
zero123plus/pipeline.py:327
Method__call__
r""" Main function for mesh extraction from scalar field using FlexiCubes. This function converts discrete signed distance fields, en
src/models/geometry/rep_3d/flexicubes.py:133
Method__getattr__
(self, name: str)
zero123plus/pipeline.py:100
Method__getitem__
(self, key)
zero123plus/pipeline.py:223
Method__getitem__
(self, index)
src/data/objaverse_zero123plus.py:98
Method__getitem__
(self, index)
src/data/objaverse.py:124
Method__getitem__
(self, index)
src/data/objaverse.py:298
Method__init__
(self, resume, logdir, ckptdir, cfgdir, config)
train.py:89
Method__init__
( self, chained_proc, enabled=False, name=None )
zero123plus/pipeline.py:44
Method__init__
(self, unet: UNet2DConditionModel, train_sched: DDPMScheduler, val_sched: EulerAncestralDiscreteScheduler)
zero123plus/pipeline.py:81
Method__init__
(self, unet: RefOnlyNoisedUNet, controlnet: Optional[diffusers.ControlNetModel] = None, conditioning_scale=1.0
zero123plus/pipeline.py:177
Method__init__
(self, procs: dict)
zero123plus/pipeline.py:218
Method__init__
( self, vae: AutoencoderKL, text_encoder: CLIPTextModel, tokenizer: CLIPTokeni
zero123plus/pipeline.py:286
Method__init__
( self, stable_diffusion_config, drop_cond_prob=0.1, )
zero123plus/model.py:44
Method__init__
( self, lrm_generator_config, input_size=256, render_size=512, init_ck
src/model_mesh.py:27
Method__init__
( self, lrm_generator_config, lrm_path=None, input_size=256, render_si
src/model.py:15
Method__init__
( self, batch_size=8, num_workers=4, train=None, validation=None,
src/data/objaverse_zero123plus.py:16
Method__init__
( self, batch_size=8, num_workers=4, train=None, validation=None,
src/data/objaverse.py:30
Method__init__
(self, root_dir='objaverse/', input_view_num=6, input_image_size=320, fov=30,
src/data/objaverse.py:242
Method__init__
( self, encoder_freeze: bool = False, encoder_model_name: str = 'facebook/dino-vitb1
src/models/lrm.py:32
Method__init__
( self, encoder_freeze: bool = False, encoder_model_name: str = 'facebook/dino-vitb1
src/models/lrm_mesh.py:34
Method__init__
(self, triplane_dim: int, samples_per_ray: int)
src/models/renderer/synthesizer_mesh.py:112
Method__init__
(self, triplane_dim: int, samples_per_ray: int)
src/models/renderer/synthesizer.py:80
Method__init__
(self)
src/models/renderer/utils/renderer.py:107
Method__init__
(self)
src/models/renderer/utils/ray_sampler.py:87
Method__init__
(self, activation_factory)
src/models/renderer/utils/ray_marcher.py:26
Method__init__
(self)
src/models/geometry/camera/__init__.py:14
Method__init__
(self, fovy=49.0, device='cuda')
src/models/geometry/camera/perspective_camera.py:25
Method__init__
(self)
src/models/geometry/rep_3d/__init__.py:14
Method__init__
( self, grid_res=64, scale=2.0, device='cuda', renderer=None, render_type='neural_rend
src/models/geometry/rep_3d/dmtet.py:397
Method__init__
(self, device="cuda", qef_reg_scale=1e-3, weight_scale=0.99)
src/models/geometry/rep_3d/flexicubes.py:67
Method__init__
( self, grid_res=64, scale=2.0, device='cuda', renderer=None, render_type='neural_rend
src/models/geometry/rep_3d/flexicubes_geometry.py:36
Method__init__
(self)
src/models/geometry/render/__init__.py:4
Method__init__
(self, device='cuda', camera_model=None)
src/models/geometry/render/neural_render.py:70
Method__init__
(self, model_name: str, freeze: bool = True)
src/models/encoder/dino_wrapper.py:26
Method__init__
(self, config: ViTConfig, use_mask_token: bool = False)
src/models/encoder/dino.py:39
Method__init__
(self, config)
src/models/encoder/dino.py:121
Method__init__
(self, config: ViTConfig)
src/models/encoder/dino.py:154
Method__init__
(self, config: ViTConfig)
src/models/encoder/dino.py:219
Method__init__
(self, config: ViTConfig)
src/models/encoder/dino.py:232
Method__init__
(self, config: ViTConfig)
src/models/encoder/dino.py:271
Method__init__
(self, config: ViTConfig)
src/models/encoder/dino.py:287
Method__init__
(self, config: ViTConfig)
src/models/encoder/dino.py:358
Method__init__
(self, config: ViTConfig, add_pooling_layer: bool = True, use_mask_token: bool = False)
src/models/encoder/dino.py:450
Method__init__
(self, config: ViTConfig)
src/models/encoder/dino.py:539
Method__init__
( self, inner_dim: int, cond_dim: int, num_heads: int, eps: float,
src/models/decoder/transformer.py:26
Method__len__
(self)
src/data/objaverse_zero123plus.py:84
Method__len__
(self)
src/data/objaverse.py:107
Method__len__
(self)
src/data/objaverse.py:277
Method_init_weights
Initialize the weights
src/models/encoder/dino.py:422
Method_prune_heads
Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer} See base class PreTrainedModel
src/models/encoder/dino.py:466
Methodadd_controlnet
(self, controlnet: Optional[diffusers.ControlNetModel] = None, conditioning_scale=1.0)
zero123plus/pipeline.py:317
Functionbatch_subdivide_volume
(tet_pos_bxnx3, tet_bxfx4, grid_sdf)
src/models/geometry/rep_3d/dmtet.py:227
Functioncheck_input_image
(input_image)
app.py:119
Functioncompact_tets
(pos_nx3, sdf_n, tet_fx4)
src/models/geometry/rep_3d/dmtet.py:207
Methodconfigure_optimizers
(self)
zero123plus/model.py:266
Methodconfigure_optimizers
(self)
src/model_mesh.py:318
Methodconfigure_optimizers
(self)
src/model.py:293
Functioncount_params
(model, verbose=False)
src/utils/train_util.py:4
Methodextract_mesh
Extract a 3D mesh from FlexiCubes. Only support batch_size 1. :param planes: triplane features :param use_texture_map: use te
src/models/lrm_mesh.py:340
Methodforward
( self, sample, timestep, encoder_hidden_states, class_labels=None, *args, cross_attention_kwa
zero123plus/pipeline.py:118
Methodforward
(self, sample, timestep, encoder_hidden_states, class_labels=None, *args, cross_attention_kwargs: dict, **kwar
zero123plus/pipeline.py:196
Methodforward
(self, images, cameras, render_cameras, render_size: int)
src/models/lrm.py:90
Methodforward
(self, images, cameras, render_cameras, render_size: int)
src/models/lrm_mesh.py:325
Methodforward
(self, sampled_features, ray_directions)
src/models/renderer/synthesizer.py:42
← previousnext →101–200 of 275, ranked by callers