MCPcopy Create free account

hub / github.com/amap-cvlab/MV-Painter / functions

Functions502 in github.com/amap-cvlab/MV-Painter

⨍ Functions 502◇ Types & classes 45
Method__getitem__
(self, index)
MVPainter/pbr/data/mv_dataset_arbobjaverse.py:494
Method__getitem__
(self, index)
MVPainter/pbr/data/custom_mv_dataset.py:120
Method__init__
(self, resume, logdir, ckptdir, cfgdir, config)
MVPainter/train.py:91
Method__init__
(self, d_model: int, n_head: int, attn_mask: torch.Tensor = None)
MVPainter/mvpainter/controlnet.py:57
Method__init__
( self, conditioning_embedding_channels: int, conditioning_channels: int = 3,
MVPainter/mvpainter/controlnet.py:107
Method__init__
(self, chained_proc, enabled=False, name=None)
MVPainter/mvpainter/mvpainter_pipeline.py:121
Method__init__
(self, unet, train_scheduler,val_scheduler = None)
MVPainter/mvpainter/mvpainter_pipeline.py:137
Method__init__
(self, unet: RefOnlyNoisedUNet, controlnet: Optional[diffusers.ControlNetModel] = None, condi
MVPainter/mvpainter/mvpainter_pipeline.py:258
Method__init__
( self, vae: AutoencoderKL, unet: UNet2DConditionModel, scheduler: KarrasDiffu
MVPainter/mvpainter/mvpainter_pipeline.py:324
Method__init__
( self, stable_diffusion_config, drop_cond_prob=0.1, )
MVPainter/mvpainter/model_unet.py:49
Method__init__
(self,offset,candidate_view_weights)
MVPainter/mvpainter/bake_pipeline_mvadapter.py:44
Method__init__
(self,offset = -90,candidate_view_weights = [1, 0.1, 0.1, 0.1, 0.05, 0.05])
MVPainter/mvpainter/bake_pipeline_mvadapter.py:66
Method__init__
( self, stable_diffusion_config, drop_cond_prob=0.1, )
MVPainter/mvpainter/model_controlnet.py:49
Method__init__
(self,offset,candidate_view_weights)
MVPainter/mvpainter/bake_pipeline.py:51
Method__init__
(self,offset = -90,candidate_view_weights = [1, 0.1, 0.1, 0.1, 0.0, 0.05])
MVPainter/mvpainter/bake_pipeline.py:71
Method__init__
( self, camera_distance=1.45, camera_type='orth', default_resolution=1024, texture_siz
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:131
Method__init__
( self, batch_size=8, num_workers=4, train=None, v
MVPainter/src/data/mvpainter_dataset.py:18
Method__init__
(self,offset,candidate_view_weights)
MVPainter/scripts/bake_texture.py:19
Method__init__
(self,offset = -90,candidate_view_weights = [1, 0.1, 0.1, 0.1, 0.05, 0])
MVPainter/scripts/bake_texture.py:37
Method__init__
(self)
MVPainter/evaluation/prompts.py:2
Method__init__
Runs at initialization of class object. @param base_rating - The rating a new player would have
MVPainter/evaluation/elopy.py:12
Method__init__
Runs at initialization of class object. @param name - TODO @param rating - TODO
MVPainter/evaluation/elopy.py:135
Method__init__
(self, metrics, device)
MVPainter/pbr/utils/metrics.py:155
Method__init__
(self, root_dir: str, object_list: list = None, )
MVPainter/pbr/data/custom_dataset.py:12
Method__init__
(self, root_dir: str, num_views: int = 4, object_list: list = None, use_cam: b
MVPainter/pbr/data/ortho_6_view_dataset.py:29
Method__init__
(self, img_wh: Tuple[int, int], object_list: str, data_root: dict, split: str
MVPainter/pbr/data/mv_dataset_arbobjaverse.py:89
Method__init__
(self, root_dir: str, num_views: int = 4, object_list: list = None, use_cam: b
MVPainter/pbr/data/custom_mv_dataset.py:12
Method__init__
( self, vae: AutoencoderKL, text_encoder: CLIPTextModel, tokenizer: CLIPTokeni
MVPainter/pbr/pipelines/pipeline_idarbdiffusion.py:71
Method__init__
( self, in_channels: int, temb_channels: int, num_domains: int, dropou
MVPainter/pbr/models/unet_dr2d_blocks.py:485
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, num_d
MVPainter/pbr/models/unet_dr2d_blocks.py:747
Method__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, n
MVPainter/pbr/models/transformer_dr2d.py:397
Method__init__
( self, sample_size: Optional[int] = None, in_channels: int = 4, out_channels:
MVPainter/pbr/models/unet_dr2d_condition.py:181
Method__len__
(self)
MVPainter/src/data/mvpainter_dataset.py:106
Method__len__
(self)
MVPainter/pbr/data/custom_dataset.py:72
Method__len__
(self)
MVPainter/pbr/data/ortho_6_view_dataset.py:126
Method__len__
(self)
MVPainter/pbr/data/mv_dataset_arbobjaverse.py:138
Method__len__
(self)
MVPainter/pbr/data/custom_mv_dataset.py:94
Method_find_mismatched_keys
( state_dict, model_state_dict, loaded_keys, ignore_mismatched
MVPainter/pbr/models/unet_dr2d_condition.py:1372
Method_set_gradient_checkpointing
(self, module, value=False)
MVPainter/mvpainter/controlnet.py:685
Method_set_gradient_checkpointing
(self, module, value=False)
MVPainter/pbr/models/unet_dr2d_condition.py:749
Methodattn_processors
r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with
MVPainter/mvpainter/controlnet.py:554
Methodattn_processors
r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with
MVPainter/pbr/models/unet_dr2d_condition.py:621
Methodbake_texture
(self, colors, elevs, azims, camera_distance=None, center=None, exp=6, weights=None)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:808
Methodblender2midas
Blender: rub midas: lub
MVPainter/pbr/data/mv_dataset_arbobjaverse.py:214
Methodblur_tensors
模糊操作：先将图像缩放，再缩放回来 :param tensors: Tensor，形状为(6, 3, h, w) :param scale: 缩放因子 :return: 模糊后的Tensor
MVPainter/src/data/mvpainter_dataset.py:344
Functionbuild_hierarchy
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp:311
Functionbuild_hierarchy_with_feat
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp:436
Functioncalculate_final_condition
(args)
data_process/get_texture_score.py:247
Functioncalculate_score_one
(args)
data_process/get_texture_score.py:224
Methodcall_bake
(self, mesh, image_paths)
MVPainter/scripts/bake_texture.py:132
Functioncheck_pbr_process_one
(glb_path)
MVPainter/scripts/blender_render_ortho.py:407
Functioncheck_pbr_process_one
(glb_path)
MVPainter/evaluation/blender_render_eval.py:433
Functioncheck_pbr_textures
(principled_bsdf)
MVPainter/scripts/blender_render_ortho.py:353
Functioncheck_pbr_textures
(principled_bsdf)
MVPainter/evaluation/blender_render_eval.py:365
Functionclean_imported_data
Remove all non-mesh objects from the scene. This includes Armatures and any other types that are not Mesh.
data_process/blender_script.py:582
Methodcolor_rgb_to_srgb
(self, image)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:358
Methodconfigure_optimizers
(self)
MVPainter/mvpainter/model_unet.py:354
Methodconfigure_optimizers
(self)
MVPainter/mvpainter/model_controlnet.py:371
Methodcontains
Returns true if this object contains a player with the given name. Otherwise returns false. @param name - name to check for.
MVPainter/evaluation/elopy.py:38
Methodconvert_normal_map
(self, image)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:542
Functioncount_params
(model, verbose=False)
MVPainter/src/utils/train_util.py:4
Methodcreate_custom_forward
(module, return_dict=None)
MVPainter/pbr/models/unet_dr2d_blocks.py:697
Methodcreate_custom_forward
(module, return_dict=None)
MVPainter/pbr/models/unet_dr2d_blocks.py:850
Methodcustom_forward
(*inputs)
MVPainter/pbr/models/unet_dr2d_blocks.py:698
Methodcustom_forward
(*inputs)
MVPainter/pbr/models/unet_dr2d_blocks.py:851
Methoddecode_latents
(self, latents)
MVPainter/pbr/pipelines/pipeline_idarbdiffusion.py:175
Functiondisable_all_lights
()
data_process/blender_script.py:666
Functiondisable_all_lights
()
MVPainter/scripts/blender_render_ortho.py:653
Functiondisable_all_lights
()
MVPainter/evaluation/blender_render_eval.py:703
Functiondisable_world_light
()
data_process/blender_script.py:673
Functiondisable_world_light
()
MVPainter/scripts/blender_render_ortho.py:661
Functiondisable_world_light
()
MVPainter/evaluation/blender_render_eval.py:711
Methoddo_classifier_free_guidance
(self)
MVPainter/mvpainter/mvpainter_pipeline.py:503
Functionerode_mask
(mask)
MVPainter/pbr/utils/metrics.py:133
Methodexpand_img_from_07_98
(self,img,ratio = 0.98 / 0.7)
MVPainter/src/data/mvpainter_dataset.py:259
Functionextract_depth
(directory)
data_process/blender_script.py:439
Functionextract_depth
(directory)
MVPainter/evaluation/blender_render_eval.py:545
Functionfetch_nearest_seq_layer
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp:97
Methodfn_recursive_add_processors
(name: str, module: torch.nn.Module, processors: Dict[str, AttentionProcessor])
MVPainter/mvpainter/controlnet.py:563
Methodfn_recursive_add_processors
(name: str, module: torch.nn.Module, processors: Dict[str, AttentionProcessor])
MVPainter/pbr/models/unet_dr2d_condition.py:630
Methodfn_recursive_attn_processor
(name: str, module: torch.nn.Module, processor)
MVPainter/mvpainter/controlnet.py:599
Methodfn_recursive_attn_processor
(name: str, module: torch.nn.Module, processor)
MVPainter/pbr/models/unet_dr2d_condition.py:665
Methodfn_recursive_retrieve_sliceable_dims
(module: torch.nn.Module)
MVPainter/mvpainter/controlnet.py:636
Methodfn_recursive_retrieve_sliceable_dims
(module: torch.nn.Module)
MVPainter/pbr/models/unet_dr2d_condition.py:700
Methodfn_recursive_set_attention_slice
(module: torch.nn.Module, slice_size: List[int])
MVPainter/mvpainter/controlnet.py:674
Methodfn_recursive_set_attention_slice
(module: torch.nn.Module, slice_size: List[int])
MVPainter/pbr/models/unet_dr2d_condition.py:738
Methodforward
(self, x: torch.Tensor)
MVPainter/mvpainter/controlnet.py:50
Methodforward
(self, x: torch.Tensor)
MVPainter/mvpainter/controlnet.py:72
Methodforward
(self, conditioning)
MVPainter/mvpainter/controlnet.py:129
Methodforward
The [`ControlNetModel`] forward method. Args: sample (`torch.FloatTensor`): The noisy input tensor.
MVPainter/mvpainter/controlnet.py:689
Methodforward
( self, sample: torch.FloatTensor, timestep: Union[torch.Tensor, float, int],
MVPainter/mvpainter/mvpainter_pipeline.py:160
Methodforward
(self, sample, timestep, encoder_hidden_states, class_labels=None, *args,
MVPainter/mvpainter/mvpainter_pipeline.py:280
Methodforward
( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None,
MVPainter/pbr/models/unet_dr2d_blocks.py:567
Methodforward
( self, hidden_states: torch.FloatTensor, res_hidden_states_tuple: Tuple[torch.FloatTe
MVPainter/pbr/models/unet_dr2d_blocks.py:676
Methodforward
( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None,
MVPainter/pbr/models/unet_dr2d_blocks.py:831
Methodforward
The [`Transformer2DModel`] forward method. Args: hidden_states (`torch.LongTensor` of shape `(batch size, num latent pix
MVPainter/pbr/models/transformer_dr2d.py:233
Methodforward
( self, hidden_states: torch.FloatTensor, attention_mask: Optional[torch.FloatTensor]
MVPainter/pbr/models/transformer_dr2d.py:519
Methodforward
r""" The [`UNet2DConditionModel`] forward method. Args: sample (`torch.FloatTensor`): The noisy input ten
MVPainter/pbr/models/unet_dr2d_condition.py:753
FunctiongenerateRandomSeeds
Function to generate n random seeds. @param n The number of seeds to generate.
MVPainter/mvpainter/differentiable_renderer/voronoi.py:79
MethodgetPlayerRating
Returns the rating of the player with the given name. @param name - name of the player. @return - the rating of the player wi
MVPainter/evaluation/elopy.py:111
← previousnext →301–400 of 502, ranked by callers