MCPcopy Create free account

hub / github.com/GasaiYU/PartRM / functions

Functions389 in github.com/GasaiYU/PartRM

↓ 1 callersFunctiongenerate_spherical_spiral_camera_views
(num_cameras, radius, num_spirals)
PartDrag4D/rendering/blender_script.py:336
↓ 1 callersFunctionget_center_boundary_index
(verts)
preprocess/src/models/geometry/rep_3d/dmtet_utils.py:12
↓ 1 callersFunctionget_center_boundary_index
(grid_res, device)
preprocess/src/models/geometry/rep_3d/flexicubes_geometry.py:17
↓ 1 callersFunctionget_circular_camera_poses
(M=120, radius=2.5, elevation=30.0)
preprocess/src/utils/camera_util.py:62
↓ 1 callersMethodget_gaussian_loss
(self, gt_gaussians, pred_gaussians)
core/models.py:120
↓ 1 callersMethodget_geometry_prediction
(self, sampled_features, flexicubes_indices)
preprocess/src/models/renderer/synthesizer_mesh.py:69
↓ 1 callersMethodget_mesh
(self, v_deformed_nx3, sdf_n, weight_n=None, with_uv=False, indices=None, is_training=False)
preprocess/src/models/geometry/rep_3d/flexicubes_geometry.py:63
↓ 1 callersMethodget_num_views
(self, path)
core/train_dataset_objaverse_hq.py:106
↓ 1 callersMethodget_num_views
(self, path)
core/eval_dataset_objaverse_hq.py:105
↓ 1 callersFunctionget_obj_from_str
(string, reload=False)
preprocess/src/utils/train_util.py:21
↓ 1 callersMethodget_other_action_ids
(self, image_dir, image_name, valid_image_list)
core/train_dataset_objaverse_hq.py:121
↓ 1 callersMethodget_other_action_ids
(self, image_dir, image_name, valid_image_list)
core/eval_dataset_objaverse_hq.py:120
↓ 1 callersMethodget_sdf_deformation_prediction
Predict SDF and deformation for tetrahedron vertices :param planes: triplane feature map for the geometry
preprocess/src/models/lrm_mesh.py:102
↓ 1 callersFunctionget_tet_bxfx4x3
(bxnxz, bxfx4)
preprocess/src/models/geometry/rep_3d/dmtet.py:328
↓ 1 callersMethodget_texture_prediction
Predict Texture given triplanes :param planes: the triplane feature map :param tex_pos: Position we want to query the texture
preprocess/src/models/lrm.py:107
↓ 1 callersMethodget_texture_prediction
(self, sampled_features)
preprocess/src/models/renderer/synthesizer_mesh.py:83
↓ 1 callersMethodget_v
(self, x, noise, t)
preprocess/zero123plus/model.py:176
↓ 1 callersMethodget_valid_image_list
(self, path)
core/train_dataset_objaverse_hq.py:96
↓ 1 callersMethodget_valid_image_list
(self, path)
core/eval_dataset_objaverse_hq.py:95
↓ 1 callersFunctiongrid_distortion
(images, strength=0.5)
core/utils.py:63
↓ 1 callersMethodinit_flexicubes_geometry
(self, device, fovy=50.0)
preprocess/src/models/lrm_mesh.py:77
↓ 1 callersFunctioninterpolate
(attr, rast, attr_idx, rast_db=None)
preprocess/src/utils/mesh_util.py:157
↓ 1 callersFunctioninterpolate
(attr, rast, attr_idx, rast_db=None)
preprocess/src/models/geometry/rep_3d/extract_texture_map.py:16
↓ 1 callersMethodinterpolate_pos_encoding
This method allows to interpolate the pre-trained position encodings, to be able to use the model on higher resolution images.
preprocess/src/models/encoder/dino.py:50
↓ 1 callersFunctionjudge
(center, radius, z_buffer, cur_depth, epsilon=0.005)
PartDrag4D/z_buffer_al.py:29
↓ 1 callersMethodjudge_action
(self, image_dir)
core/train_dataset_objaverse_hq.py:130
↓ 1 callersMethodjudge_action
(self, image_dir)
core/eval_dataset_objaverse_hq.py:129
↓ 1 callersMethodload_im
(self, path, color)
preprocess/src/data/objaverse_zero123plus.py:87
↓ 1 callersMethodload_im
replace background pixel with random color in rendering
preprocess/src/data/objaverse.py:280
↓ 1 callersFunctionload_object
(filepath: str)
PartDrag4D/rendering/blender_script.py:242
↓ 1 callersMethodload_ply
(self, path, compatible=True)
core/train_dataset_objaverse_hq.py:147
↓ 1 callersMethodload_ply
(self, path, compatible=True)
core/train_dataset_partdrag4d.py:219
↓ 1 callersFunctionlocal2world
(points, scale, world_matrix)
PartDrag4D/z_buffer_al.py:15
↓ 1 callersMethodlocal2world
(self, drags, scale, world_matrix)
core/eval_dataset_partdrag4d.py:172
↓ 1 callersMethodlocal2world
(self, drags, scale, world_matrix)
core/train_dataset_partdrag4d.py:167
↓ 1 callersMethodlocal2world
(self, drags, scale, world_matrix)
preprocess/gen_propagate_drags.py:78
↓ 1 callersFunctionmain
(dataset_name)
eval.py:13
↓ 1 callersFunctionmain
(dataset_name)
train.py:19
↓ 1 callersFunctionmain
(mesh_base)
PartDrag4D/z_buffer_al.py:105
↓ 1 callersFunctionmain
(val_filelist, mesh_base, render_base, save_dir, sample_num=10)
preprocess/gen_propagate_drags.py:195
↓ 1 callersFunctionmain
(src_filelist, save_dir)
preprocess/gen_mv_partdrag4d.py:20
↓ 1 callersFunctionmain
(src_filelist, save_dir)
preprocess/gen_mv_objaverse_hq.py:20
↓ 1 callersFunctionmarching_tets
(pos_nx3, sdf_n, tet_fx4, triangle_table, num_triangles_table, base_tet_edges, v_id)
preprocess/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
preprocess/src/models/geometry/rep_3d/dmtet.py:129
↓ 1 callersFunctionmerge_and_rename_materials
(src_mtl_path, dst_mtl_path, material_ranges, existing_material_count)
PartDrag4D/preprocess/process_data_textured_uv.py:51
↓ 1 callersFunctionmerge_meshes
(obj_path)
PartDrag4D/preprocess/animate_data.py:54
↓ 1 callersFunctionmerge_meshes
(save_folder, ids, mesh_dict)
PartDrag4D/preprocess/process_data_textured_uv.py:233
↓ 1 callersFunctionndc2Pix
(v, S)
PartDrag4D/z_buffer_al.py:25
↓ 1 callersMethodndc2Pix
(self, v, S)
core/eval_dataset_partdrag4d.py:169
↓ 1 callersMethodndc2Pix
(self, v, S)
core/train_dataset_partdrag4d.py:164
↓ 1 callersMethodndc2Pix
(self, v, S)
preprocess/gen_propagate_drags.py:74
↓ 1 callersFunctionorbit_camera_jitter
(poses, strength=0.1)
core/utils.py:45
↓ 1 callersFunctionpad_camera_extrinsics_4x4
(extrinsics)
preprocess/src/utils/camera_util.py:6
↓ 1 callersMethodpredict_start_from_z_and_v
(self, x_t, t, v)
preprocess/zero123plus/model.py:170
↓ 1 callersMethodprepare_batch_data
(self, batch)
preprocess/src/model_mesh.py:80
↓ 1 callersMethodprepare_batch_data
(self, batch)
preprocess/src/model.py:42
↓ 1 callersMethodprepare_validation_batch_data
(self, batch)
preprocess/src/model_mesh.py:139
↓ 1 callersMethodprepare_validation_batch_data
(self, batch)
preprocess/src/model.py:104
↓ 1 callersMethodproject
(self, points_bxnx4)
preprocess/src/models/geometry/camera/perspective_camera.py:31
↓ 1 callersMethodproject_drag
(self, drags, cam_view)
preprocess/gen_propagate_drags.py:88
↓ 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
preprocess/src/models/renderer/utils/renderer.py:46
↓ 1 callersFunctionproject_points
(points, cam_view)
PartDrag4D/z_buffer_al.py:51
↓ 1 callersFunctionprojection
(x=0.1, n=1.0, f=50.0, near_plane=None)
preprocess/src/models/geometry/camera/perspective_camera.py:14
↓ 1 callersFunctionprojection_matrix
(camd)
PartDrag4D/rendering/blender_script.py:189
↓ 1 callersMethodprune_heads
(self, heads: Set[int])
preprocess/src/models/encoder/dino.py:238
↓ 1 callersFunctionrefine_z_buffer_index
(z_buffer, z_buffer_index, pcd_points)
PartDrag4D/z_buffer_al.py:41
↓ 1 callersMethodregister_schedule
(self)
preprocess/zero123plus/model.py:75
↓ 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
preprocess/src/models/lrm_mesh.py:238
↓ 1 callersMethodrender_mesh
(self, mesh_v_nx3, mesh_f_fx3, camera_mv_bx4x4, resolution=256, hierarchical_mask=False)
preprocess/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)
preprocess/src/models/geometry/rep_3d/flexicubes_geometry.py:74
↓ 1 callersFunctionrotation_matrix
Return the rotation matrix associated with counterclockwise rotation about the given axis by theta radians.
PartDrag4D/preprocess/animate_data.py:27
↓ 1 callersMethodrun_model_activated
(self, planes, decoder, sample_coordinates, sample_directions, options)
preprocess/src/models/renderer/utils/renderer.py:206
↓ 1 callersFunctionsam_predict
(predictor: SamPredictor, image, query_point)
preprocess/gen_propagate_drags.py:30
↓ 1 callersMethodsample_importance
Return depths of importance sampled points along rays. See NeRF importance sampling for more.
preprocess/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
preprocess/src/models/renderer/utils/renderer.py:284
↓ 1 callersFunctionsave_images
(object_file: str, args)
PartDrag4D/rendering/blender_script.py:381
↓ 1 callersFunctionscale_latents
(latents)
preprocess/zero123plus/model.py:18
↓ 1 callersFunctionscene_meshes
()
PartDrag4D/rendering/blender_script.py:287
↓ 1 callersFunctionsdf_reg_loss_batch
(sdf, all_edges)
preprocess/src/model_mesh.py:15
↓ 1 callersFunctionset_camera_focus_on_head
(camera, model_object)
PartDrag4D/rendering/blender_script.py:211
↓ 1 callersFunctionset_spherical_spiral_camera
(camera, index, num_cameras, radius, num_spirals)
PartDrag4D/rendering/blender_script.py:346
↓ 1 callersFunctionspherical_camera_pose
(azimuths: np.ndarray, elevations: np.ndarray, radius=2.5)
preprocess/src/utils/camera_util.py:47
↓ 1 callersFunctiontet_to_tet_adj_sparse
(tet_tx4)
preprocess/src/models/geometry/rep_3d/dmtet.py:272
↓ 1 callersFunctiontraverse_tree
(current_node, mesh_dict)
PartDrag4D/preprocess/process_data_textured_uv.py:36
↓ 1 callersMethodunify_samples
(self, depths1, colors1, densities1, depths2, colors2, densities2, normals1=None, normals2=None)
preprocess/src/models/renderer/utils/renderer.py:218
↓ 1 callersFunctionunscale_image
(image)
preprocess/gen_mv_partdrag4d.py:16
↓ 1 callersFunctionunscale_image
(image)
preprocess/gen_mv_objaverse_hq.py:16
↓ 1 callersFunctionunscale_image
(image)
preprocess/zero123plus/pipeline.py:173
↓ 1 callersFunctionunscale_latents
(latents)
preprocess/zero123plus/pipeline.py:163
↓ 1 callersFunctionunscale_latents
(latents)
preprocess/zero123plus/model.py:23
↓ 1 callersFunctionview_plane
(camd, winx, winy, xasp, yasp)
PartDrag4D/rendering/blender_script.py:118
↓ 1 callersFunctionxfm_points
Transform points. Args: points: Tensor containing 3D points with shape [minibatch_size, num_vertices, 3] or [1, num_vertices, 3] m
preprocess/src/models/geometry/render/neural_render.py:23
Method__call__
x: arbitrary shape of tensor. dim: cat dim
core/drag_embedding.py:14
Method__call__
( self, attn: Attention, hidden_states, encoder_hidden_states=None, attention_mask=None, mode=
preprocess/zero123plus/pipeline.py:56
Method__call__
( self, image: Image.Image = None, prompt = "", *args, num_images_per_
preprocess/zero123plus/pipeline.py:329
Method__call__
r""" Main function for mesh extraction from scalar field using FlexiCubes. This function converts discrete signed distance fields, en
preprocess/src/models/geometry/rep_3d/flexicubes.py:133
Method__getattr__
(self, name: str)
preprocess/zero123plus/pipeline.py:101
Method__getitem__
(self, idx)
core/eval_dataset_partdrag4d.py:261
Method__getitem__
(self, idx)
core/train_dataset_objaverse_hq.py:243
Method__getitem__
(self, idx)
core/train_dataset_partdrag4d.py:257
← previousnext →101–200 of 389, ranked by callers