MCPcopy Create free account

hub / github.com/YuxuanSnow/Human3Diffusion / functions

Functions165 in github.com/YuxuanSnow/Human3Diffusion

⨍ Functions 165◇ Types & classes 33
Method__init__
(self, input_nc, output_nc, ngf=64, n_downsampling=3, n_blocks=9, norm_layer=nn.BatchNorm2d,
core/tsdf_mesh.py:390
Method__init__
(self, dim, padding_type, norm_layer, activation=nn.ReLU(True), use_dropout=False)
core/tsdf_mesh.py:420
Method__init__
( self, dim: int, dim_q: int, dim_k: int, dim_v: int, num_head
core/attention.py:88
Method__init__
( self, vae: AutoencoderKL, unet: MultiViewUNetModel, tokenizer: CLIPTokenizer
mvdream/pipeline_imagedream.py:35
Method__init__
(self, dim, dim_out=None, mult=4, glu=False, dropout=0.0)
mvdream/mv_unet.py:103
Method__init__
( self, query_dim, context_dim=None, heads=8,
mvdream/mv_unet.py:121
Method__init__
( self, dim, n_heads, d_head, context_dim, dropout=0.0,
mvdream/mv_unet.py:208
Method__init__
( self, in_channels, n_heads, d_head, context_dim, # cross attention i
mvdream/mv_unet.py:254
Method__init__
(self, *, dim, dim_head=64, heads=8)
mvdream/mv_unet.py:312
Method__init__
( self, dim=1024, depth=8, dim_head=64, heads=16, num_queries=
mvdream/mv_unet.py:355
Method__init__
(self, channels, use_conv, dims=2, out_channels=None, padding=1)
mvdream/mv_unet.py:414
Method__init__
(self, channels, use_conv, dims=2, out_channels=None, padding=1)
mvdream/mv_unet.py:440
Method__init__
( self, channels, emb_channels, dropout, out_channels=None, us
mvdream/mv_unet.py:467
Method__init__
( self, image_size, in_channels, model_channels, out_channels,
mvdream/mv_unet.py:554
Method__len__
(self)
core/dataset_human_imagedream.py:60
Method__len__
(self)
core/dataloader_inference.py:61
Method_execution_device
r""" Returns the device on which the pipeline's models will be executed. After calling `pipeline.enable_sequential_cpu_offload()` the
mvdream/pipeline_imagedream.py:177
Functionapply_depth_colormap
( depth_, accumulation_, near_plane = 0.5, far_plane = 2.5, cmap="turbo", )
core/utils.py:132
Functioncolormap
(images, cmap='jet')
core/utils.py:100
Functioncosine_lr_schedule
Decay the learning rate
train_MVDMVR_joint.py:399
Methoddisable_vae_slicing
r""" Disable sliced VAE decoding. If `enable_vae_slicing` was previously invoked, this method will go back to computing decoding in on
mvdream/pipeline_imagedream.py:101
Methoddisable_vae_tiling
r""" Disable tiled VAE decoding. If `enable_vae_tiling` was previously invoked, this method will go back to computing decoding in one
mvdream/pipeline_imagedream.py:117
Methodenable_model_cpu_offload
r""" Offloads all models to CPU using accelerate, reducing memory usage with a low impact on performance. Compared to `enable_sequenti
mvdream/pipeline_imagedream.py:148
Methodenable_sequential_cpu_offload
r""" Offloads all models to CPU using accelerate, significantly reducing memory usage. When called, unet, text_encoder, vae and safety
mvdream/pipeline_imagedream.py:124
Methodenable_vae_slicing
r""" Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to compute decoding in se
mvdream/pipeline_imagedream.py:92
Methodenable_vae_tiling
r""" Enable tiled VAE decoding. When this option is enabled, the VAE will split the input tensor into tiles to compute decoding and e
mvdream/pipeline_imagedream.py:108
Methodextract_mesh
(self, gaussians, cam_view, cam_view_proj, cam_pos)
core/gof.py:231
Methodextract_mesh
(self, gaussians, cam_view, cam_view_proj, cam_pos)
core/tsdf_mesh.py:238
Methodforward
(self, data, timestep, step_ratio=1)
core/diffusion3d_imagedream.py:94
Methodforward
(self, x)
core/unet_timeImage_cond.py:35
Methodforward
(self, x, t_emb=None)
core/unet_timeImage_cond.py:91
Methodforward
(self, x, t_emb=None)
core/unet_timeImage_cond.py:145
Methodforward
(self, x, t_emb=None)
core/unet_timeImage_cond.py:184
Methodforward
(self, x, xs, t_emb=None)
core/unet_timeImage_cond.py:225
Methodforward
(self, x, timestep)
core/unet_timeImage_cond.py:305
Methodforward
(self, data, timestep, step_ratio=1)
core/models_timeImage_cond_xt_gof.py:107
Methodforward
(self, input)
core/tsdf_mesh.py:416
Methodforward
(self, x: Tensor, attn_bias=None)
core/attention.py:68
Methodforward
(self, q: Tensor, k: Tensor, v: Tensor)
core/attention.py:113
Methodforward
(self, q: Tensor, k: Tensor, v: Tensor, attn_bias=None)
core/attention.py:134
Methodforward
(self, x)
mvdream/mv_unet.py:116
Methodforward
(self, x, context=None)
mvdream/mv_unet.py:155
Methodforward
(self, x, context=None, num_frames=1)
mvdream/mv_unet.py:243
Methodforward
(self, x, context=None, num_frames=1)
mvdream/mv_unet.py:294
Methodforward
(self, x, latents)
mvdream/mv_unet.py:326
Methodforward
(self, x)
mvdream/mv_unet.py:388
Methodforward
(self, x, emb, context=None, num_frames=1)
mvdream/mv_unet.py:401
Methodforward
(self, x)
mvdream/mv_unet.py:425
Methodforward
(self, x)
mvdream/mv_unet.py:460
Methodforward
(self, x, emb)
mvdream/mv_unet.py:529
Methodforward
( self, x, timesteps=None, context=None, y=None, camera=None,
mvdream/mv_unet.py:852
Functionget_3x4_RT_matrix_from_blender
(cam)
render_bproc_thuman2.py:119
Functionget_camera_objects
()
render_bproc_thuman2.py:219
Functionget_norm_layer
(norm_type='instance')
core/tsdf_mesh.py:460
Methodget_point_cloud
Extract a point cloud from the voxel volume.
core/tsdf_mesh.py:900
Methodload_ply
(self, path, compatible=True)
core/gof.py:342
Functionmove_bottom_left
(mask)
core/tsdf_mesh.py:512
Functionmove_bottom_right
(mask)
core/tsdf_mesh.py:513
Functionmove_top_left
(mask)
core/tsdf_mesh.py:510
Functionmove_top_right
(mask)
core/tsdf_mesh.py:511
Functionnormalize_camera
(camera_matrix)
render_bproc_thuman2.py:109
Methodprepare_default_rays
(self, device, elevation=0)
core/diffusion3d_imagedream.py:47
Methodprepare_default_rays
(self, device, elevation=0)
core/models_timeImage_cond_xt_gof.py:61
Methodsave_ply
(self, gaussians, path, compatible=True)
core/tsdf_mesh.py:302
Functionwarmup_lr_schedule
Warmup the learning rate
train_MVDMVR_joint.py:393
← previous101–165 of 165, ranked by callers