MCPcopy Create free account

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

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

↓ 117 callersMethodto
(self, device)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:350
↓ 20 callersMethodfrom_pretrained
(cls, pretrained_model_name_or_path, *model_args, **kwargs)
MVPainter/mvpainter/mvpainter_pipeline.py:716
↓ 19 callersMethodto
(self, *args, **kwargs)
MVPainter/pbr/pipelines/pipeline_idarbdiffusion.py:254
↓ 12 callersFunctionsave_image
(out, name)
MVPainter/infer_pbr.py:52
↓ 9 callersMethodraster_interpolate
(self, uv, rast_out, uv_idx, rast_db=None, diff_attrs=None)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:190
↓ 7 callersFunctionfetch_seq
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp:55
↓ 6 callersFunctionmse
(inputs, target, mask, root=False)
MVPainter/pbr/utils/metrics.py:92
↓ 6 callersFunctionrank_zero_print
(*args)
MVPainter/train.py:19
↓ 6 callersMethodraster_rasterize
(self, pos, tri, resolution, ranges=None, grad_db=True)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:176
↓ 6 callersMethodrecordMatch
Should be called after a game is played. @param name1 - name of the first player. @param name2 - name of the second player.
MVPainter/evaluation/elopy.py:68
↓ 6 callersMethodrender
( self, elev, azim, camera_distance=None, center=None, resolut
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:426
↓ 6 callersFunctionzero_module
(module)
MVPainter/mvpainter/controlnet.py:922
↓ 5 callersFunctioninstantiate_from_config
(config)
MVPainter/src/utils/train_util.py:11
↓ 5 callersFunctionpos2key
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp:4
↓ 5 callersFunctiontransform_pos
(mtx, pos, keepdim=False)
MVPainter/mvpainter/differentiable_renderer/camera_utils.py:31
↓ 4 callersMethod__init__
(self, state_dict: Dict[str, torch.Tensor])
MVPainter/mvpainter/mvpainter_pipeline.py:219
↓ 4 callersMethodback_project
(self, image, elev, azim, camera_distance=None, center=None, method=None)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:702
↓ 4 callersFunctioncalculate_color_entropy
(image_path)
data_process/get_texture_score.py:16
↓ 4 callersFunctioncalculate_sobel_in_alpha_region
计算图像 alpha 区域内的 Sobel 梯度均值 :param image_path: 图像路径(RGBA 格式) :return: alpha 区域内的梯度幅值均值
data_process/get_texture_score.py:34
↓ 4 callersFunctionextract_into_tensor
(a, t, x_shape)
MVPainter/mvpainter/model_unet.py:42
↓ 4 callersFunctionextract_into_tensor
(a, t, x_shape)
MVPainter/mvpainter/model_controlnet.py:42
↓ 4 callersMethodfast_bake_texture
(self, textures, cos_maps)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:832
↓ 4 callersMethodgetPlayer
Returns the player in the implementation with the given name. @param name - name of the player to return. @return - the playe
MVPainter/evaluation/elopy.py:27
↓ 4 callersMethodget_added_cond_kwargs_train
(self,batch_size,is_drop,crops_coords_top_left: Tuple[int, int] = (0, 0),)
MVPainter/mvpainter/mvpainter_pipeline.py:402
↓ 4 callersFunctionget_attribute_data
(gltf, accessor_index)
MVPainter/mvpainter/custom_rasterizer/custom_rasterizer/io_glb.py:45
↓ 4 callersMethodget_prompt_embeds_train
(self,is_drop,cond_image)
MVPainter/mvpainter/mvpainter_pipeline.py:373
↓ 4 callersFunctionget_scene_root_objects
Returns all root objects in the scene. Yields: Generator[bpy.types.Object, None, None]: Generator of all root objects in the
MVPainter/scripts/blender_render_ortho.py:478
↓ 4 callersFunctionget_scene_root_objects
Returns all root objects in the scene. Yields: Generator[bpy.types.Object, None, None]: Generator of all root objects in the
MVPainter/evaluation/blender_render_eval.py:504
↓ 4 callersFunctionkey2pos
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp:11
↓ 4 callersMethodraster_antialias
(self, color, rast, pos, tri, topology_hash=None, pos_gradient_boost=1.0)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:214
↓ 4 callersFunctionsave_images
Saves rendered images of the object in the scene.
data_process/blender_script.py:746
↓ 4 callersFunctionscatter_add_nd_with_count
(input, count, indices, values, weights=None)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:49
↓ 4 callersMethodset_texture
(self, tex)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:312
↓ 3 callersMethod__getPlayerList
Returns this implementation's player list. @return - the list of all player objects in the implementation.
MVPainter/evaluation/elopy.py:20
↓ 3 callersMethodaddPlayer
Adds a new player to the implementation. @param name - The name to identify a specific player. @param rating - The player's r
MVPainter/evaluation/elopy.py:49
↓ 3 callersFunctioncalculateSignedArea2
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/rasterizer.h:12
↓ 3 callersFunctioncreate_image_grid
(image_folder, output_path, suffix='basecolor')
MVPainter/infer_pbr.py:29
↓ 3 callersMethodfrom_unet
r""" Instantiate a [`ControlNetModel`] from [`UNet2DConditionModel`]. Parameters: unet (`UNet2DConditionModel`):
MVPainter/mvpainter/controlnet.py:483
↓ 3 callersMethodgetRatingList
Returns a list of tuples in the form of ({name},{rating}) @return - the list of tuples
MVPainter/evaluation/elopy.py:120
↓ 3 callersFunctionget_mean_elo
(elo_metric)
MVPainter/evaluation/calculate_metric.py:46
↓ 3 callersFunctionget_mv_matrix
(elev, azim, camera_distance, center=None)
MVPainter/mvpainter/differentiable_renderer/camera_utils.py:47
↓ 3 callersMethodget_pos_from_mvp
(self, elev, azim, camera_distance, center)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:561
↓ 3 callersFunctionlinear_grid_put_2d
(H, W, coords, values, return_count=False)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:77
↓ 3 callersMethodload_im
(self, path, color, random_ratio=0.8)
MVPainter/src/data/mvpainter_dataset.py:232
↓ 3 callersMethodload_mesh
( self, mesh, scale_factor=1.15, auto_center=True, )
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:224
↓ 3 callersFunctionload_scheduler
(scheduler_type, cfg)
MVPainter/train_pbr.py:46
↓ 3 callersFunctionmesh_uv_wrap
(mesh)
MVPainter/mvpainter/uv_warp_utils.py:5
↓ 3 callersFunctionpos_from_seq
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp:50
↓ 3 callersMethodprepare
(self)
MVPainter/mvpainter/mvpainter_pipeline.py:359
↓ 3 callersMethodprepare_extra_step_kwargs
(self, generator, eta)
MVPainter/mvpainter/mvpainter_pipeline.py:479
↓ 3 callersMethodprepare_latents
(self, batch_size, num_channels_latents, height, width, dtype, device, generator, latents=None)
MVPainter/mvpainter/mvpainter_pipeline.py:436
↓ 3 callersMethodrender_normal
( self, elev, azim, camera_distance=None, center=None, resolut
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:473
↓ 3 callersMethodrender_position
(self, elev, azim, camera_distance=None, center=None, resolution=None, bg_color=[1, 1,
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:615
↓ 3 callersFunctionreset_scene
Resets the scene to a clean state. Returns: None
data_process/blender_script.py:242
↓ 3 callersMethodsave_mesh
(self)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:239
↓ 3 callersMethodsave_pretrained
(self, save_directory)
MVPainter/mvpainter/mvpainter_pipeline.py:709
↓ 3 callersFunctionscene_bbox
Returns the bounding box of the scene. Taken from Shap-E rendering script (https://github.com/openai/shap-e/blob/main/shap_e/rendering/blende
MVPainter/scripts/blender_render_ortho.py:440
↓ 3 callersFunctionscene_bbox
Returns the bounding box of the scene. Taken from Shap-E rendering script (https://github.com/openai/shap-e/blob/main/shap_e/rendering/blende
MVPainter/evaluation/blender_render_eval.py:466
↓ 3 callersMethodset_attn_processor
r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o
MVPainter/mvpainter/controlnet.py:578
↓ 3 callersFunctionto_rgb_image
convert a PIL.Image to rgb mode with white background maybe_rgba: PIL.Image return: PIL.Image
MVPainter/mvpainter/mvpainter_pipeline.py:46
↓ 3 callersMethoduv_inpaint
(self, texture, mask)
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:849
↓ 2 callersFunctionDownsampleGrid
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp:157
↓ 2 callersFunctionFOV_to_intrinsics
Creates a 3x3 camera intrinsics matrix from the camera field of view, specified in degrees. Note the intrinsics are returned as normalized by
MVPainter/src/utils/camera_util.py:84
↓ 2 callersFunctionFetchNeighbor
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp:116
↓ 2 callersFunctionNeighborGrid
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp:192
↓ 2 callersFunctionPadGrid
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp:264
↓ 2 callersMethod__fetch_mv
(self, index)
MVPainter/pbr/data/mv_dataset_arbobjaverse.py:464
↓ 2 callersMethod__init__
( self, in_channels: int = 4, conditioning_channels: int = 3, flip_sin_to_cos:
MVPainter/mvpainter/controlnet.py:211
↓ 2 callersMethod__init__
( self, in_channels: int, out_channels: int, prev_output_channel: int,
MVPainter/pbr/models/unet_dr2d_blocks.py:596
↓ 2 callersMethod_encode_image
(self, image_pil, device, num_images_per_prompt, do_classifier_free_guidance)
MVPainter/pbr/pipelines/pipeline_idarbdiffusion.py:141
↓ 2 callersFunctioncalculateBarycentricCoordinate
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/rasterizer.h:16
↓ 2 callersFunctioncape
(x, p)
MVPainter/pbr/models/transformer_dr2d.py:642
↓ 2 callersFunctioncenter_looking_at_camera_pose
Create OpenGL camera extrinsics from camera locations and look-at position. camera_position: (M, 3) or (3,) look_at: (3) up_world: (
MVPainter/src/utils/camera_util.py:16
↓ 2 callersMethodcompareRating
Compares the two ratings of the this player and the opponent. @param opponent - the player to compare against. @returns - The
MVPainter/evaluation/elopy.py:144
↓ 2 callersMethodencode_condition_image
(self, images)
MVPainter/mvpainter/model_unet.py:156
↓ 2 callersMethodencode_condition_image
(self, images)
MVPainter/mvpainter/model_controlnet.py:169
↓ 2 callersMethodencode_image
(self, image: torch.Tensor, scale_factor: bool = False)
MVPainter/mvpainter/mvpainter_pipeline.py:370
↓ 2 callersMethodforward_unet
(self, latents, t, prompt_embeds, cond_latents,depth_imgs,added_cond_kwargs,is_training,depth_imgs_2)
MVPainter/mvpainter/model_unet.py:179
↓ 2 callersMethodforward_unet
(self, latents, t, prompt_embeds, cond_latents,depth_imgs,added_cond_kwargs,is_training,depth_imgs_2)
MVPainter/mvpainter/model_controlnet.py:192
↓ 2 callersFunctionget_buffer_data
(gltf, buffer_view)
MVPainter/mvpainter/custom_rasterizer/custom_rasterizer/io_glb.py:36
↓ 2 callersFunctionget_down_block
( down_block_type, num_layers, in_channels, out_channels, temb_channels, add_downsampl
MVPainter/pbr/models/unet_dr2d_blocks.py:32
↓ 2 callersFunctionget_image_data
(gltf, image, folder)
MVPainter/mvpainter/custom_rasterizer/custom_rasterizer/io_glb.py:72
↓ 2 callersFunctionget_node_transform
(node)
MVPainter/mvpainter/custom_rasterizer/custom_rasterizer/io_glb.py:111
↓ 2 callersFunctionget_pair_win_rate_single
(pair_dir,target = 1)
MVPainter/evaluation/calculate_metric.py:7
↓ 2 callersFunctionget_scene_meshes
Returns all meshes in the scene. Yields: Generator[bpy.types.Object, None, None]: Generator of all meshes in the scene.
data_process/blender_script.py:410
↓ 2 callersFunctionget_scene_meshes
Returns all meshes in the scene. Yields: Generator[bpy.types.Object, None, None]: Generator of all meshes in the scene.
MVPainter/scripts/blender_render_ortho.py:490
↓ 2 callersMethodload_image
(self, img_path, bg_color, alpha, return_type='np', do_gamma=False)
MVPainter/pbr/data/mv_dataset_arbobjaverse.py:174
↓ 2 callersFunctionload_object
Loads a glb model into the scene.
data_process/blender_script.py:267
↓ 2 callersFunctionmse_to_psnr
Compute PSNR given an MSE (we assume the maximum pixel value is 1).
MVPainter/pbr/utils/metrics.py:11
↓ 2 callersMethodprepare_batch_data
(self, batch)
MVPainter/mvpainter/model_unet.py:122
↓ 2 callersMethodprepare_batch_data
(self, batch)
MVPainter/mvpainter/model_controlnet.py:135
↓ 2 callersMethodrandom_stretch_or_compress
(self,image, min_scale=0.5, max_scale=1.5)
MVPainter/src/data/mvpainter_dataset.py:123
↓ 2 callersMethodraster_texture
(self, tex, uv, uv_da=None, mip_level_bias=None, mip=None, filter_mode='auto', boundary
MVPainter/mvpainter/differentiable_renderer/mesh_render.py:204
↓ 2 callersFunctionrasterizeImagecoordsKernelCPU
MVPainter/mvpainter/custom_rasterizer/lib/custom_rasterizer_kernel/rasterizer.cpp:81
↓ 2 callersFunctionreset_scene
Resets the scene to a clean state. Returns: None
MVPainter/scripts/blender_render_ortho.py:239
↓ 2 callersFunctionreset_scene
Resets the scene to a clean state. Returns: None
MVPainter/scripts/remesh_reduce_blender_script.py:89
↓ 2 callersFunctionreset_scene
Resets the scene to a clean state. Returns: None
MVPainter/evaluation/blender_render_eval.py:251
↓ 2 callersFunctionrotate_every_two
(x)
MVPainter/pbr/models/transformer_dr2d.py:636
↓ 2 callersFunctionsimse
(img_pred, img_gt, mask_gt, divide_mask=False)
MVPainter/pbr/utils/metrics.py:103
↓ 2 callersFunctionsplit_data
(data_in)
MVPainter/train_pbr.py:69
next →1–100 of 502, ranked by callers