Functions — github.com/YuxuanSnow/Human3Diffusion

hub / github.com/YuxuanSnow/Human3Diffusion / functions

Functions165 in github.com/YuxuanSnow/Human3Diffusion

⨍ Functions 165◇ Types & classes 33

↓ 10 callersMethod__init__

(self, dim_in, dim_out)

mvdream/mv_unet.py:93

↓ 9 callersFunctionconv_nd

(dims, *args, **kwargs)

mvdream/mv_unet.py:66

↓ 6 callersFunctionget_rays

(pose, h, w, fovy, opengl=True)

core/utils.py:14

↓ 5 callersMethod__init__

( self, dim: int, num_heads: int = 8, qkv_bias: bool = False, proj_bi

core/unet_timeImage_cond.py:12

↓ 5 callersMethodrender

(self, gaussians, cam_view, cam_view_proj, cam_pos, bg_color=None, sub_pixel_offset=None, scale_modifier=1.0)

core/gof.py:31

↓ 4 callersMethodforward

(self, x: Tensor)

core/attention.py:51

↓ 4 callersFunctionprint_model_info

(model)

train_MVDMVR_joint.py:336

↓ 3 callersFunctiondefault

(val, d)

mvdream/mv_unet.py:86

↓ 3 callersFunctiondepth_to_normal

(c2w, opt, depth)

core/utils.py:87

↓ 3 callersMethodforward

(self, x)

mvdream/mv_unet.py:97

↓ 3 callersMethodforward_gaussians

(self, images, timestep)

core/diffusion3d_imagedream.py:71

↓ 3 callersFunctionprint_model_info

(model)

train_MultiviewDiffusion_diffusion.py:278

↓ 3 callersFunctionscene_root_objects

()

render_bproc_thuman2.py:75

↓ 3 callersFunctionzero_module

(module)

mvdream/mv_unet.py:60

↓ 2 callersMethod__init__

(self, vol_bnds, voxel_size, use_gpu=True)

core/tsdf_mesh.py:657

↓ 2 callersMethod_encode_prompt

r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):

mvdream/pipeline_imagedream.py:194

↓ 2 callersMethodalpha_to_sdf

(alpha)

core/gof.py:225

↓ 2 callersMethodalpha_to_sdf

(alpha)

core/tsdf_mesh.py:232

↓ 2 callersMethoddecode_latents

(self, latents)

mvdream/pipeline_imagedream.py:340

↓ 2 callersMethodencode_image

(self, image, device, num_images_per_prompt)

mvdream/pipeline_imagedream.py:430

↓ 2 callersMethodencode_image_latents

(self, image, device, num_images_per_prompt)

mvdream/pipeline_imagedream.py:470

↓ 2 callersMethodevaluate_alpha

(self, points, gaussians, cam_view, cam_view_proj, cam_pos)

core/gof.py:210

↓ 2 callersMethodevaluate_alpha

(self, points, gaussians, cam_view, cam_view_proj, cam_pos)

core/tsdf_mesh.py:217

↓ 2 callersMethodget_volume

(self)

core/tsdf_mesh.py:894

↓ 2 callersMethodintegrate

(self, color_im, depth_im, cam_intr, cam_pose, obs_weight=1.)

core/tsdf_mesh.py:821

↓ 2 callersFunctionmove_bottom

(mask)

core/tsdf_mesh.py:509

↓ 2 callersFunctionmove_left

(mask)

core/tsdf_mesh.py:506

↓ 2 callersFunctionmove_right

(mask)

core/tsdf_mesh.py:507

↓ 2 callersFunctionmove_top

(mask)

core/tsdf_mesh.py:508

↓ 2 callersMethodprepare_latents

( self, batch_size, num_channels_latents, height, width, dtype

mvdream/pipeline_imagedream.py:385

↓ 2 callersFunctionscene_bbox

(single_obj=None, ignore_matrix=False)

render_bproc_thuman2.py:59

↓ 2 callersFunctionscene_meshes

()

render_bproc_thuman2.py:80

↓ 2 callersFunctionsigmoid

(x, k=1)

core/tsdf_mesh.py:553

↓ 1 callersMethodCLIP_preprocess

(self, x)

mvdream/pipeline_imagedream.py:418

↓ 1 callersMethod__init__

( self, dim: int, num_heads: int = 8, qkv_bias: bool = False, proj_bia

core/attention.py:32

↓ 1 callersFunction_encode_text_prompt

( tokenizer, text_encoder, prompt, device, batch_size )

train_MVDMVR_joint.py:88

↓ 1 callersFunction_encode_text_prompt

( tokenizer, text_encoder, prompt, device, batch_size )

train_MultiviewDiffusion_diffusion.py:81

↓ 1 callersFunctionapply_colormap

(image, cmap="viridis")

core/utils.py:121

↓ 1 callersFunctionavg_pool_nd

(dims, *args, **kwargs)

mvdream/mv_unet.py:76

↓ 1 callersFunctionbilateral_normal_integration_depth

(normal_map, depth_map, mask,

core/tsdf_mesh.py:556

↓ 1 callersMethodbuild_conv_block

(self, dim, padding_type, norm_layer, activation, use_dropout)

core/tsdf_mesh.py:424

↓ 1 callersMethodcam2pix

(cam_pts, intr)

core/tsdf_mesh.py:801

↓ 1 callersMethodcheck_inputs

(self, image, height, width, callback_steps)

mvdream/pipeline_imagedream.py:347

↓ 1 callersFunctionconvert_opengl_to_blender

(camera_matrix)

render_bproc_thuman2.py:53

↓ 1 callersFunctionconvert_opengl_to_blender

(camera_matrix)

core/tsdf_mesh.py:977

↓ 1 callersFunctioncreate_camera_to_world_matrix

(elevation, azimuth, radius=1.0)

render_bproc_thuman2.py:31

↓ 1 callersFunctioncreate_camera_to_world_matrix

(elevation, azimuth, radius=1.0)

core/tsdf_mesh.py:955

↓ 1 callersFunctiondepths_to_points

(c2w, opt, depthmap)

core/utils.py:61

↓ 1 callersMethodextract_xxx_part

(s)

core/dataset_human_imagedream.py:64

↓ 1 callersMethodforward

(self, x)

core/tsdf_mesh.py:456

↓ 1 callersMethodforward_gaussians

(self, images, timestep)

core/models_timeImage_cond_xt_gof.py:85

↓ 1 callersFunctiongenerate_dx_dy

(mask, nz_horizontal, nz_vertical, step_size=1)

core/tsdf_mesh.py:515

↓ 1 callersFunctiongenerate_tsdf_mesh

(subj_base_path, normal_checkpoint_path, quality='high')

core/tsdf_mesh.py:983

↓ 1 callersFunctionget_2ddiffusion_model

(unet2d_dict_path, device, org_imgdream_path="ashawkey/imagedream-ipmv-diffusers")

core/diffusion3d_pipeline.py:33

↓ 1 callersFunctionget_3ddiffusion_model

(unet3d_dict_path, device, opt)

core/diffusion3d_pipeline.py:79

↓ 1 callersFunctionget_a_camera_location

(loc)

render_bproc_thuman2.py:102

↓ 1 callersFunctionget_calibration_matrix_K_from_blender

(mode='simple')

render_bproc_thuman2.py:130

↓ 1 callersFunctionget_camera

( num_frames, elevation=0, azimuth_start=0, azimuth_span=360, blender_coord=True, extra_view=False, )

mvdream/mv_unet.py:23

↓ 1 callersFunctionget_edge_aware_distortion_map

(gt_image, distortion_map)

core/utils.py:49

↓ 1 callersMethodget_mesh

(self)

core/tsdf_mesh.py:919

↓ 1 callersFunctionget_normal_estimator

(state_dict_path)

core/tsdf_mesh.py:469

↓ 1 callersMethodintegrate

(self, points3D, gaussians, cam_view, cam_view_proj, cam_pos, bg_color=None, sub_pixel_offset=None, scale_modi

core/gof.py:115

↓ 1 callersMethodintegrate

(self, points3D, gaussians, cam_view, cam_view_proj, cam_pos, bg_color=None, sub_pixel_offset=None, scale_modi

core/tsdf_mesh.py:120

↓ 1 callersMethodintegrate_tsdf

(tsdf_vol, dist, w_old, obs_weight)

core/tsdf_mesh.py:813

↓ 1 callersFunctionjoint_2d_3d_diffusion

(batch, device, diffusion_2d_pipe, diffusion_3d_pipe, weight_dtype=torch.float32)

core/diffusion3d_pipeline.py:103

↓ 1 callersFunctionload_human_obj

(human_obj_file, human_tex_file, smpl_z_component)

render_bproc_thuman2.py:175

↓ 1 callersMethodload_ply

(self, path, compatible=True)

core/tsdf_mesh.py:350

↓ 1 callersFunctionmain

(args)

train_MVDMVR_joint.py:146

↓ 1 callersFunctionmain

()

infer.py:5

↓ 1 callersFunctionmain

()

infer_mesh.py:4

↓ 1 callersFunctionmain

(args)

train_MultiviewDiffusion_diffusion.py:137

↓ 1 callersFunctionmain

()

train_MultiviewReconstructor_diffusion.py:31

↓ 1 callersFunctionnormalize_scene_human

()

render_bproc_thuman2.py:85

↓ 1 callersFunctionparse_args

(input_args=None)

train_MVDMVR_joint.py:52

↓ 1 callersFunctionparse_args

(input_args=None)

train_MultiviewDiffusion_diffusion.py:48

↓ 1 callersMethodprepare_extra_step_kwargs

(self, generator, eta)

mvdream/pipeline_imagedream.py:369

↓ 1 callersMethodpreprocess

(input_image_path)

core/dataloader_inference.py:65

↓ 1 callersFunctionprint_model_info

(model)

train_MultiviewReconstructor_diffusion.py:83

↓ 1 callersMethodrender

(self, gaussians, cam_view, cam_view_proj, cam_pos, bg_color=None, sub_pixel_offset=None, scale_modifier=1.0)

core/tsdf_mesh.py:38

↓ 1 callersFunctionreset_scene

()

render_bproc_thuman2.py:198

↓ 1 callersFunctionrigid_transform

(xyz, transform)

core/tsdf_mesh.py:935

↓ 1 callersFunctionsave_generation_results

(subject_save_folder, batch, device, gaussians, diffusion_3d_pipe, weight_dtype)

core/diffusion3d_pipeline.py:308

↓ 1 callersFunctionsave_images

(save_folder, scan_subj_path)

render_bproc_thuman2.py:227

↓ 1 callersMethodsave_ply

(self, gaussians, path, compatible=True)

core/gof.py:295

↓ 1 callersFunctiontimestep_embedding

(timesteps, dim, max_period=10000, repeat_only=False)

mvdream/mv_unet.py:41

↓ 1 callersMethodvox2world

(vol_origin, vox_coords, vox_size)

core/tsdf_mesh.py:790

Method__call__

( self, prompt: str = "", image: Union[np.ndarray, torch.FloatTensor] = None,

mvdream/pipeline_imagedream.py:512

Method__getitem__

(self, idx)

core/dataset_human_imagedream.py:72

Method__getitem__

(self, idx)

core/dataloader_inference.py:96

Method__init__

(self, render_path, opt, training=True, front_input=False, all_data=False, white_bg=False)

core/dataset_human_imagedream.py:18

Method__init__

( self, opt: Options, )

core/diffusion3d_imagedream.py:17

(self, opt, rgb_path_list, white_bg=False)

core/dataloader_inference.py:18

Method__init__

( self, in_channels: int, out_channels: int, time_embed_dim: int, resa

core/unet_timeImage_cond.py:51

Method__init__

( self, in_channels: int, out_channels: int, time_embed_dim: int, num_

core/unet_timeImage_cond.py:116

Method__init__

( self, in_channels: int, time_embed_dim: int, num_layers: int = 1, at

core/unet_timeImage_cond.py:161

Method__init__

( self, in_channels: int, prev_out_channels: int, out_channels: int, t

core/unet_timeImage_cond.py:193

Method__init__

( self, in_channels: int = 3, out_channels: int = 3, down_channels: Tuple[int,

core/unet_timeImage_cond.py:242

Method__init__

( self, opt: Options, )

core/models_timeImage_cond_xt_gof.py:17

Method__init__

(self)

core/tsdf_mesh.py:21

next →1–100 of 165, ranked by callers