MCPcopy Create free account

hub / github.com/GasaiYU/PartRM / functions

Functions389 in github.com/GasaiYU/PartRM

⨍ Functions 389◇ Types & classes 71
Methodforward
(self, colors, densities, depths, rendering_options, normals=None)
preprocess/src/models/renderer/utils/ray_marcher.py:66
Methodforward
(self)
preprocess/src/models/geometry/rep_3d/__init__.py:17
Methodforward
(self)
preprocess/src/models/geometry/render/__init__.py:7
Methodforward
(self, image, camera)
preprocess/src/models/encoder/dino_wrapper.py:37
Methodforward
( self, pixel_values: torch.Tensor, bool_masked_pos: Optional[torch.BoolTensor] = None
preprocess/src/models/encoder/dino.py:83
Methodforward
(self, pixel_values: torch.Tensor, interpolate_pos_encoding: bool = False)
preprocess/src/models/encoder/dino.py:136
Methodforward
( self, hidden_states, head_mask: Optional[torch.Tensor] = None, output_attentions: bool = False )
preprocess/src/models/encoder/dino.py:177
Methodforward
(self, hidden_states: torch.Tensor, input_tensor: torch.Tensor)
preprocess/src/models/encoder/dino.py:224
Methodforward
( self, hidden_states: torch.Tensor, head_mask: Optional[torch.Tensor] = None,
preprocess/src/models/encoder/dino.py:256
Methodforward
(self, hidden_states: torch.Tensor)
preprocess/src/models/encoder/dino.py:279
Methodforward
(self, hidden_states: torch.Tensor, input_tensor: torch.Tensor)
preprocess/src/models/encoder/dino.py:292
Methodforward
( self, hidden_states: torch.Tensor, adaln_input: torch.Tensor = None, head_ma
preprocess/src/models/encoder/dino.py:325
Methodforward
( self, hidden_states: torch.Tensor, adaln_input: torch.Tensor = None, head_ma
preprocess/src/models/encoder/dino.py:364
Methodforward
r""" bool_masked_pos (`torch.BoolTensor` of shape `(batch_size, num_patches)`, *optional*): Boolean masked positions. Indicates wh
preprocess/src/models/encoder/dino.py:474
Methodforward
(self, hidden_states)
preprocess/src/models/encoder/dino.py:544
Methodforward
(self, x, cond)
preprocess/src/models/decoder/transformer.py:56
Methodforward
(self, image_feats)
preprocess/src/models/decoder/transformer.py:102
Methodforward_synthesizer
(self, planes, render_cameras, render_size: int)
preprocess/src/models/lrm.py:82
MethodgetAABB
(self)
preprocess/src/models/geometry/rep_3d/dmtet.py:431
MethodgetAABB
(self)
preprocess/src/models/geometry/rep_3d/flexicubes_geometry.py:60
Functionget_3x4_RT_matrix_from_blender
(cam)
PartDrag4D/rendering/blender_script.py:292
Functionget_action_embeddings
(action)
core/utils.py:111
Methodget_geometry_prediction
(self, planes, sample_coordinates, flexicubes_indices)
preprocess/src/models/renderer/synthesizer_mesh.py:127
Methodget_input_embeddings
(self)
preprocess/src/models/encoder/dino.py:463
Functionget_local2world_mat
Returns the pose of the object in the form of a local2world matrix. :return: The 4x4 local2world matrix.
PartDrag4D/rendering/blender_script.py:363
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
preprocess/src/models/renderer/utils/math_utils.py:46
Methodget_tet_mesh
(self, v_deformed_nx3, sdf_n, with_uv=False, indices=None)
preprocess/src/models/geometry/rep_3d/dmtet.py:446
Methodget_texture_prediction
(self, planes, sample_coordinates)
preprocess/src/models/renderer/synthesizer_mesh.py:135
Functionget_zero123plus_input_cameras
Get the input camera parameters.
preprocess/src/utils/camera_util.py:95
Functionimages_to_video
( images: torch.Tensor, output_path: str, fps: int = 30, )
preprocess/src/utils/infer_util.py:66
Functionlinspace
Creates a tensor of shape [num, *start.shape] whose values are evenly spaced from start to end, inclusive. Replicates but the multi-dimension
preprocess/src/models/renderer/utils/math_utils.py:101
Functionload_obj_with_material
(obj_path)
PartDrag4D/preprocess/process_data_textured_uv.py:156
Methodload_ply
(self, path, compatible=True)
core/eval_dataset_partdrag4d.py:223
Methodload_ply
(self, path, compatible=True)
core/gs.py:154
Methodload_ply
(self, path, compatible=True)
core/eval_dataset_objaverse_hq.py:146
Functionloadobj
(meshfile)
preprocess/src/utils/mesh_util.py:92
Functionloadobjtex
(meshfile)
preprocess/src/utils/mesh_util.py:114
Methodmap_from
(module, state_dict, *args, **kwargs)
preprocess/zero123plus/pipeline.py:257
Methodmap_to
(module, state_dict, *args, **kwargs)
preprocess/zero123plus/pipeline.py:242
Functionnormalize_scene
()
PartDrag4D/rendering/blender_script.py:308
Functionnormalize_vecs
Normalize vector lengths.
preprocess/src/models/renderer/utils/math_utils.py:33
Methodon_fit_start
(self)
preprocess/zero123plus/model.py:98
Methodon_fit_start
(self)
preprocess/src/model_mesh.py:73
Methodon_fit_start
(self)
preprocess/src/model.py:37
Methodon_validation_epoch_end
(self)
preprocess/zero123plus/model.py:261
Methodon_validation_epoch_end
(self)
preprocess/src/model_mesh.py:303
Methodon_validation_epoch_end
(self)
preprocess/src/model.py:278
Methodprepare_default_rays
(self, device, elevation=0)
core/models.py:74
Methodremap_key
(key, state_dict)
preprocess/zero123plus/pipeline.py:251
Functionremove_background
(image: PIL.Image.Image, rembg_session: Any = None, force: bool = False, **rembg_kwargs, )
preprocess/src/utils/infer_util.py:11
Methodrender
(self, v_deformed_bxnx3=None, sdf_bxn=None, camera_mv_bxnviewx4x4=None, resolution=256)
preprocess/src/models/geometry/rep_3d/dmtet.py:484
Methodrender
(self, v_deformed_bxnx3=None, sdf_bxn=None, camera_mv_bxnviewx4x4=None, resolution=256)
preprocess/src/models/geometry/rep_3d/flexicubes_geometry.py:100
Methodrender_mesh
( self, mesh_v_pos_bxnx3, mesh_t_pos_idx_fx3, camera_mv_bx4x4,
preprocess/src/models/geometry/render/neural_render.py:77
Functionreset_scene
Resets the scene to a clean state.
PartDrag4D/rendering/blender_script.py:226
Functionresize_foreground
( image: PIL.Image.Image, ratio: float, )
preprocess/src/utils/infer_util.py:25
Functionrotation_matrix
Return the rotation matrix associated with counterclockwise rotation about the given axis by theta radians.
PartDrag4D/preprocess/process_data_textured_uv.py:19
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
preprocess/src/models/renderer/utils/renderer.py:81
Functionsave_combined_obj
(verts, normals, faces, mtl_files, mtl_dict, output_obj)
PartDrag4D/preprocess/process_data_textured_uv.py:191
Functionsave_glb
(pointnp_px3, facenp_fx3, colornp_px3, fpath)
preprocess/src/utils/mesh_util.py:31
Functionsave_obj
(pointnp_px3, facenp_fx3, colornp_px3, fpath)
preprocess/src/utils/mesh_util.py:18
Functionsave_obj_with_mtl
(pointnp_px3, tcoords_px2, facenp_fx3, facetex_fx3, texmap_hxwx3, fname)
preprocess/src/utils/mesh_util.py:43
Methodsave_ply
(self, gaussians, path, compatible=True)
core/gs.py:101
Functionsave_video
( frames: torch.Tensor, output_path: str, fps: int = 30, )
preprocess/src/utils/infer_util.py:87
Functionscale_image
(image)
preprocess/zero123plus/pipeline.py:168
Functionscale_image
(image)
preprocess/zero123plus/model.py:28
Functionscale_latents
(latents)
preprocess/zero123plus/pipeline.py:158
Functionsdf_reg_loss
(sdf, all_edges)
preprocess/src/models/geometry/rep_3d/dmtet.py:371
Functionsdf_reg_loss_batch
(sdf, all_edges)
preprocess/src/models/geometry/rep_3d/dmtet.py:384
Methodsetup
(self, stage)
preprocess/src/data/objaverse_zero123plus.py:38
Methodsetup
(self, stage)
preprocess/src/data/objaverse.py:52
Functionshrink_grid
(tet_pos_bxnx3, tet_bxfx4, grid_sdf)
preprocess/src/models/geometry/rep_3d/dmtet.py:340
Methodsort_samples
(self, all_depths, all_colors, all_densities)
preprocess/src/models/renderer/utils/renderer.py:211
Methodstate_dict
(self, **kwargs)
core/models.py:65
Methodtest_dataloader
(self)
preprocess/src/data/objaverse_zero123plus.py:55
Methodtest_dataloader
(self)
preprocess/src/data/objaverse.py:69
Functiontorch_dot
Dot product of two tensors.
preprocess/src/models/renderer/utils/math_utils.py:39
Methodtrain_dataloader
(self)
preprocess/src/data/objaverse_zero123plus.py:45
Methodtrain_dataloader
(self)
preprocess/src/data/objaverse.py:59
Methodtraining_step
(self, batch, batch_idx)
preprocess/zero123plus/model.py:182
Methodtraining_step
(self, batch, batch_idx)
preprocess/src/model_mesh.py:196
Methodtraining_step
(self, batch, batch_idx)
preprocess/src/model.py:195
Functiontransform_vectors
Left-multiplies MxM @ NxM. Returns NxM.
preprocess/src/models/renderer/utils/math_utils.py:25
Methodval_dataloader
(self)
preprocess/src/data/objaverse_zero123plus.py:50
Methodval_dataloader
(self)
preprocess/src/data/objaverse.py:64
Methodvalidation_step
(self, batch, batch_idx)
preprocess/zero123plus/model.py:243
Methodvalidation_step
(self, batch, batch_idx)
preprocess/src/model_mesh.py:294
Methodvalidation_step
(self, batch, batch_idx)
preprocess/src/model.py:269
Functionworker
( queue: multiprocessing.JoinableQueue, count: multiprocessing.Value, gpu: int, s3: Optional[b
PartDrag4D/rendering/distributed.py:37
Functionxatlas_uvmap
(ctx, mesh_v, mesh_pos_idx, resolution)
preprocess/src/utils/mesh_util.py:161
← previous301–389 of 389, ranked by callers