Functions — github.com/GasaiYU/PartRM

CodeHub MCPcopy Create free account

hub / github.com/GasaiYU/PartRM / functions

Functions389 in github.com/GasaiYU/PartRM

⨍ Functions 389◇ Types & classes 71

Method__getitem__

core/eval_dataset_objaverse_hq.py:246

Method__getitem__

preprocess/gen_propagate_drags.py:130

Method__getitem__

preprocess/zero123plus/pipeline.py:225

Method__getitem__

preprocess/src/data/objaverse_zero123plus.py:98

Method__getitem__

preprocess/src/data/objaverse.py:124

Method__getitem__

preprocess/src/data/objaverse.py:298

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

core/unet.py:14

( self, in_channels: int, out_channels: int, resample: Literal['default', 'up'

core/unet.py:54

( self, in_channels: int, out_channels: int, num_layers: int = 1, down

core/unet.py:108

( self, in_channels: int, num_layers: int = 1, attention: bool = True,

core/unet.py:153

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

core/unet.py:187

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

core/unet.py:334

(self, opt: Options)

core/eval_dataset_partdrag4d.py:29

( self, opt: Options, )

core/models.py:16

( self, dim: int, dim_q: int, dim_k: int, dim_v: int, num_head

core/attention.py:88

(self, num_drags, fourier_freqs=8, downsample_ratio=64)

core/drag_embedding.py:24

(self, fourier_freqs=8, scales=[256, 128, 64, 32, 16, 8], channels=[64, 64, 128, 256, 512, 1024], drag_layer_i

core/drag_embedding.py:77

(self, opt: Options)

(self, opt: Options)

core/train_dataset_objaverse_hq.py:31

(self, opt: Options)

core/train_dataset_partdrag4d.py:29

(self, opt: Options)

core/eval_dataset_objaverse_hq.py:31

(self, filelist, pcd_base, render_base)

preprocess/gen_propagate_drags.py:49

( self, chained_proc, enabled=False, name=None )

preprocess/zero123plus/pipeline.py:45

(self, unet: UNet2DConditionModel, train_sched: DDPMScheduler, val_sched: EulerAncestralDiscreteScheduler)

preprocess/zero123plus/pipeline.py:82

(self, unet: RefOnlyNoisedUNet, controlnet: Optional[diffusers.ControlNetModel] = None, conditioning_scale=1.0

preprocess/zero123plus/pipeline.py:179

(self, procs: dict)

preprocess/zero123plus/pipeline.py:220

( self, vae: AutoencoderKL, text_encoder: CLIPTextModel, tokenizer: CLIPTokeni

preprocess/zero123plus/pipeline.py:288

( self, stable_diffusion_config, drop_cond_prob=0.1, )

preprocess/zero123plus/model.py:45

( self, lrm_generator_config, input_size=256, render_size=512, init_ck

preprocess/src/model_mesh.py:27

( self, lrm_generator_config, lrm_path=None, input_size=256, render_si

preprocess/src/model.py:15

( self, batch_size=8, num_workers=4, train=None, validation=None,

preprocess/src/data/objaverse_zero123plus.py:16

( self, batch_size=8, num_workers=4, train=None, validation=None,

preprocess/src/data/objaverse.py:30

(self, root_dir='objaverse/', input_view_num=6, input_image_size=320, fov=30,

preprocess/src/data/objaverse.py:242

( self, encoder_freeze: bool = False, encoder_model_name: str = 'facebook/dino-vitb1

preprocess/src/models/lrm.py:32

( self, encoder_freeze: bool = False, encoder_model_name: str = 'facebook/dino-vitb1

preprocess/src/models/lrm_mesh.py:34

(self, triplane_dim: int, samples_per_ray: int)

preprocess/src/models/renderer/synthesizer_mesh.py:112

(self, triplane_dim: int, samples_per_ray: int)

preprocess/src/models/renderer/synthesizer.py:80

preprocess/src/models/renderer/utils/renderer.py:107

preprocess/src/models/renderer/utils/ray_sampler.py:87

(self, activation_factory)

preprocess/src/models/renderer/utils/ray_marcher.py:26

preprocess/src/models/geometry/camera/__init__.py:14

(self, fovy=49.0, device='cuda')

preprocess/src/models/geometry/camera/perspective_camera.py:25

preprocess/src/models/geometry/rep_3d/__init__.py:14

( self, grid_res=64, scale=2.0, device='cuda', renderer=None, render_type='neural_rend

preprocess/src/models/geometry/rep_3d/dmtet.py:397

(self, device="cuda", qef_reg_scale=1e-3, weight_scale=0.99)

preprocess/src/models/geometry/rep_3d/flexicubes.py:67

( self, grid_res=64, scale=2.0, device='cuda', renderer=None, render_type='neural_rend

preprocess/src/models/geometry/rep_3d/flexicubes_geometry.py:36

preprocess/src/models/geometry/render/__init__.py:4

(self, device='cuda', camera_model=None)

preprocess/src/models/geometry/render/neural_render.py:70

(self, model_name: str, freeze: bool = True)

preprocess/src/models/encoder/dino_wrapper.py:26

(self, config: ViTConfig, use_mask_token: bool = False)

preprocess/src/models/encoder/dino.py:39

preprocess/src/models/encoder/dino.py:121

(self, config: ViTConfig)

preprocess/src/models/encoder/dino.py:154

(self, config: ViTConfig)

preprocess/src/models/encoder/dino.py:219

(self, config: ViTConfig)

preprocess/src/models/encoder/dino.py:232

(self, config: ViTConfig)

preprocess/src/models/encoder/dino.py:271

(self, config: ViTConfig)

preprocess/src/models/encoder/dino.py:287

(self, config: ViTConfig)

preprocess/src/models/encoder/dino.py:358

(self, config: ViTConfig, add_pooling_layer: bool = True, use_mask_token: bool = False)

preprocess/src/models/encoder/dino.py:450

(self, config: ViTConfig)

preprocess/src/models/encoder/dino.py:539

( self, inner_dim: int, cond_dim: int, num_heads: int, eps: float,

preprocess/src/models/decoder/transformer.py:26

core/train_dataset_objaverse_hq.py:93

core/eval_dataset_objaverse_hq.py:92

preprocess/gen_propagate_drags.py:70

preprocess/src/data/objaverse_zero123plus.py:84

preprocess/src/data/objaverse.py:107

preprocess/src/data/objaverse.py:277

Method_init_weights

Initialize the weights

preprocess/src/models/encoder/dino.py:422

Method_prune_heads

Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer} See base class PreTrainedModel

preprocess/src/models/encoder/dino.py:466

Methodadd_controlnet

(self, controlnet: Optional[diffusers.ControlNetModel] = None, conditioning_scale=1.0)

preprocess/zero123plus/pipeline.py:319

Functionadjust_focus_to_head

PartDrag4D/rendering/blender_script.py:93

Functionbatch_subdivide_volume

(tet_pos_bxnx3, tet_bxfx4, grid_sdf)

preprocess/src/models/geometry/rep_3d/dmtet.py:227

Functioncompact_tets

(pos_nx3, sdf_n, tet_fx4)

preprocess/src/models/geometry/rep_3d/dmtet.py:207

Methodconfigure_optimizers

preprocess/zero123plus/model.py:273

Methodconfigure_optimizers

preprocess/src/model_mesh.py:318

Methodconfigure_optimizers

preprocess/src/model.py:293

Functioncount_params

(model, verbose=False)

preprocess/src/utils/train_util.py:4

Methodextract_mesh

Extract a 3D mesh from triplane nerf. Only support batch_size 1. :param planes: triplane features :param mesh_resolution: mar

preprocess/src/models/lrm.py:153

Methodextract_mesh

Extract a 3D mesh from FlexiCubes. Only support batch_size 1. :param planes: triplane features :param use_texture_map: use te

preprocess/src/models/lrm_mesh.py:340

core/unet.py:37

core/unet.py:88

core/unet.py:136

core/unet.py:177

core/unet.py:218

(self, x, drags_start=None, drags_end=None)

core/unet.py:302

(self, x, drags_start=None, drags_end=None)

core/unet.py:409

(self, data, step_ratio=1)

core/models.py:128

(self, x: Tensor, attn_bias=None)

core/attention.py:68

(self, q: Tensor, k: Tensor, v: Tensor)

core/attention.py:113

(self, q: Tensor, k: Tensor, v: Tensor, attn_bias=None)

core/attention.py:138

(self, drags_start, drags_end)

core/drag_embedding.py:43

(self, drags_start, drags_end)

core/drag_embedding.py:106

( self, sample, timestep, encoder_hidden_states, class_labels=None, *args, cross_attention_kwa

preprocess/zero123plus/pipeline.py:119

(self, sample, timestep, encoder_hidden_states, class_labels=None, *args, cross_attention_kwargs: dict, **kwar

preprocess/zero123plus/pipeline.py:198

(self, images, cameras, render_cameras, render_size: int)

preprocess/src/models/lrm.py:90

(self, images, cameras, render_cameras, render_size: int)

preprocess/src/models/lrm_mesh.py:325

(self, sampled_features, ray_directions)

preprocess/src/models/renderer/synthesizer.py:42

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

preprocess/src/models/renderer/synthesizer.py:98

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

preprocess/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

preprocess/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

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

← previous next →201–300 of 389, ranked by callers