Functions — github.com/TencentARC/InstantMesh

CodeHub MCPcopy Create free account

hub / github.com/TencentARC/InstantMesh / functions

Functions275 in github.com/TencentARC/InstantMesh

⨍ Functions 275◇ Types & classes 49

(self, planes, cameras, render_size=128, crop_params=None)

src/models/renderer/synthesizer.py:98

(self, planes, decoder, ray_origins, ray_directions, rendering_options)

src/models/renderer/utils/renderer.py:158

Create batches of rays and return origins and directions. cam2world_matrix: (N, 4, 4) intrinsics: (N, 3, 3) render_s

src/models/renderer/utils/ray_sampler.py:28

Create batches of rays and return origins and directions. cam2world_matrix: (N, 4, 4) ortho_scale: float render_size

src/models/renderer/utils/ray_sampler.py:92

(self, colors, densities, depths, rendering_options, normals=None)

src/models/renderer/utils/ray_marcher.py:66

src/models/geometry/rep_3d/__init__.py:17

src/models/geometry/render/__init__.py:7

(self, image, camera)

src/models/encoder/dino_wrapper.py:37

( self, pixel_values: torch.Tensor, bool_masked_pos: Optional[torch.BoolTensor] = None

src/models/encoder/dino.py:83

(self, pixel_values: torch.Tensor, interpolate_pos_encoding: bool = False)

src/models/encoder/dino.py:136

( self, hidden_states, head_mask: Optional[torch.Tensor] = None, output_attentions: bool = False )

src/models/encoder/dino.py:177

(self, hidden_states: torch.Tensor, input_tensor: torch.Tensor)

src/models/encoder/dino.py:224

( self, hidden_states: torch.Tensor, head_mask: Optional[torch.Tensor] = None,

src/models/encoder/dino.py:256

(self, hidden_states: torch.Tensor)

src/models/encoder/dino.py:279

(self, hidden_states: torch.Tensor, input_tensor: torch.Tensor)

src/models/encoder/dino.py:292

( self, hidden_states: torch.Tensor, adaln_input: torch.Tensor = None, head_ma

src/models/encoder/dino.py:325

( self, hidden_states: torch.Tensor, adaln_input: torch.Tensor = None, head_ma

src/models/encoder/dino.py:364

r""" bool_masked_pos (`torch.BoolTensor` of shape `(batch_size, num_patches)`, *optional*): Boolean masked positions. Indicates wh

src/models/encoder/dino.py:474

(self, hidden_states)

src/models/encoder/dino.py:544

(self, x, cond)

src/models/decoder/transformer.py:56

(self, image_feats)

src/models/decoder/transformer.py:102

Functiongenerate_mvs

(input_image, sample_steps, sample_seed)

src/models/geometry/rep_3d/dmtet.py:431

src/models/geometry/rep_3d/flexicubes_geometry.py:60

Methodget_geometry_prediction

(self, planes, sample_coordinates, flexicubes_indices)

src/models/renderer/synthesizer_mesh.py:127

Methodget_input_embeddings

src/models/encoder/dino.py:463

Functionget_ray_limits_box

Author: Petr Kellnhofer Intersects rays with the [-1, 1] NDC volume. Returns min and max distance of entry. Returns -1 for no interse

src/models/renderer/utils/math_utils.py:46

Methodget_tet_mesh

(self, v_deformed_nx3, sdf_n, with_uv=False, indices=None)

src/models/geometry/rep_3d/dmtet.py:446

Methodget_texture_prediction

(self, planes, sample_coordinates)

src/models/renderer/synthesizer_mesh.py:135

Functionimages_to_video

(images, output_path, fps=30)

Functionlinspace

Creates a tensor of shape [num, *start.shape] whose values are evenly spaced from start to end, inclusive. Replicates but the multi-dimension

src/models/renderer/utils/math_utils.py:101

Functionloadobj

src/utils/mesh_util.py:92

Functionloadobjtex

src/utils/mesh_util.py:114

(module, state_dict, *args, **kwargs)

zero123plus/pipeline.py:255

(module, state_dict, *args, **kwargs)

zero123plus/pipeline.py:240

Functionnormalize_vecs

Normalize vector lengths.

src/models/renderer/utils/math_utils.py:33

Methodon_fit_start

(self, trainer, pl_module)

Methodon_fit_start

(self, trainer, pl_module)

Methodon_fit_start

zero123plus/model.py:96

Methodon_fit_start

src/model_mesh.py:73

Methodon_fit_start

src/model.py:37

Methodon_validation_epoch_end

zero123plus/model.py:254

Methodon_validation_epoch_end

src/model_mesh.py:303

Methodon_validation_epoch_end

src/model.py:278

Functionpreprocess

(input_image, do_remove_background)

Methodremap_key

(key, state_dict)

zero123plus/pipeline.py:249

(self, v_deformed_bxnx3=None, sdf_bxn=None, camera_mv_bxnviewx4x4=None, resolution=256)

src/models/geometry/rep_3d/dmtet.py:484

(self, v_deformed_bxnx3=None, sdf_bxn=None, camera_mv_bxnviewx4x4=None, resolution=256)

src/models/geometry/rep_3d/flexicubes_geometry.py:100

Methodrender_mesh

( self, mesh_v_pos_bxnx3, mesh_t_pos_idx_fx3, camera_mv_bx4x4,

src/models/geometry/render/neural_render.py:77

Functionsample_from_3dgrid

Expects coordinates in shape (batch_size, num_points_per_batch, 3) Expects grid in shape (1, channels, H, W, D) (Also works if grid has b

src/models/renderer/utils/renderer.py:81

Functionscale_image

zero123plus/pipeline.py:166

Functionscale_image

zero123plus/model.py:27

Functionscale_latents

zero123plus/pipeline.py:156

Functionsdf_reg_loss

(sdf, all_edges)

src/models/geometry/rep_3d/dmtet.py:371

Functionsdf_reg_loss_batch

(sdf, all_edges)

src/models/geometry/rep_3d/dmtet.py:384

src/data/objaverse_zero123plus.py:38

Functionshrink_grid

(tet_pos_bxnx3, tet_bxfx4, grid_sdf)

src/models/geometry/rep_3d/dmtet.py:340

Methodsort_samples

(self, all_depths, all_colors, all_densities)

src/models/renderer/utils/renderer.py:211

Functionstr2bool

Methodtest_dataloader

src/data/objaverse_zero123plus.py:55

Methodtest_dataloader

src/data/objaverse.py:69

Functiontorch_dot

Dot product of two tensors.

src/models/renderer/utils/math_utils.py:39

Methodtrain_dataloader

src/data/objaverse_zero123plus.py:45

Methodtrain_dataloader

src/data/objaverse.py:59

Methodtraining_step

(self, batch, batch_idx)

zero123plus/model.py:180

Methodtraining_step

(self, batch, batch_idx)

src/model_mesh.py:196

Methodtraining_step

(self, batch, batch_idx)

src/model.py:195

Functiontransform_vectors

Left-multiplies MxM @ NxM. Returns NxM.

src/models/renderer/utils/math_utils.py:25

Methodval_dataloader

src/data/objaverse_zero123plus.py:50

Methodval_dataloader

src/data/objaverse.py:64

Methodvalidation_step

(self, batch, batch_idx)

zero123plus/model.py:236

Methodvalidation_step

(self, batch, batch_idx)

src/model_mesh.py:294

Methodvalidation_step

(self, batch, batch_idx)

src/model.py:269

Functionxatlas_uvmap

(ctx, mesh_v, mesh_pos_idx, resolution)

src/utils/mesh_util.py:161

← previous201–275 of 275, ranked by callers