MCPcopy Create free account

hub / github.com/AiuniAI/Unique3D / functions

Functions239 in github.com/AiuniAI/Unique3D

⨍ Functions 239◇ Types & classes 26
↓ 1 callersFunctionget_8view_cameras
(device, focal=2/1.35)
scripts/project_mesh.py:322
↓ 1 callersFunctionget_camera
(world_to_cam, fov_in_degrees=60, focal_length=1 / (2**0.5), cam_type='fov')
scripts/project_mesh.py:16
↓ 1 callersMethodget_forward_args
(self)
custum_3d_diffusion/trainings/image2mvimage_trainer.py:108
↓ 1 callersMethodget_forward_args
(self)
custum_3d_diffusion/trainings/image2image_trainer.py:58
↓ 1 callersFunctionget_ortho_ray_directions_origins
(W, H, use_pixel_centers=True, device="cuda")
scripts/mesh_init.py:12
↓ 1 callersMethodget_refunet
(self)
custum_3d_diffusion/custum_modules/unifield_processor.py:181
↓ 1 callersFunctionget_visible_faces
(meshes: Meshes, cameras: CamerasBase, resolution=1024)
scripts/project_mesh.py:100
↓ 1 callersMethodinit_shared_modules
(self, shared_modules: dict)
custum_3d_diffusion/trainings/base.py:102
↓ 1 callersFunctioninit_trainers
(cfg_path: str, weight_dtype: torch.dtype, extras: dict)
app/custom_models/utils.py:17
↓ 1 callersFunctionintegrate_vector_field
( vector_field: np.ndarray, mask: np.ndarray, target_iteration_count: int, thread_count: int,
scripts/normal_to_height_map.py:113
↓ 1 callersFunctionlaunch
( port, listen=False, share=False, gradio_root="", )
app/gradio_local.py:34
↓ 1 callersFunctionload_base_model_components
(base_model=DEFAULT_BASE_MODEL, torch_dtype=torch.float16)
scripts/sd_model_zoo.py:40
↓ 1 callersFunctionload_common_sd15_pipe
(base_model=DEFAULT_BASE_MODEL, device="auto", controlnet=None, ip_adapter=False, plus_model=True, torch_dtype
scripts/sd_model_zoo.py:68
↓ 1 callersFunctionload_image_encoder
()
scripts/sd_model_zoo.py:60
↓ 1 callersFunctionload_onnx
(file_path: str)
scripts/load_onnx.py:19
↓ 1 callersFunctionload_onnx_caller
(file_path: str, single_output=False)
scripts/load_onnx.py:26
↓ 1 callersFunctionmake_star_cameras
(az_count,pol_count,distance:float=10.,r=None,image_size=[512,512],device='cuda')
mesh_reconstruction/func.py:80
↓ 1 callersFunctionmake_star_cameras_orthographic_py3d
(azim_list, device, focal=2/1.35, dist=1.1)
mesh_reconstruction/func.py:130
↓ 1 callersFunctionnew_multiview_to_mesh_ui
(concurrency_id="wkl")
app/gradio_3dgen_steps.py:55
↓ 1 callersFunctionnormalmap_to_depthmap
(normal_np)
scripts/mesh_init.py:55
↓ 1 callersFunctionpack
removes unused elements in vertices and faces
mesh_reconstruction/remesh.py:121
↓ 1 callersFunctionparse_structured
(fields, cfg)
custum_3d_diffusion/trainings/utils.py:4
↓ 1 callersFunctionpredict
(img_list: List[Image.Image], guidance_scale=2., **kwargs)
app/custom_models/mvimg_prediction.py:17
↓ 1 callersFunctionpredict_normals
(image: List[Image.Image], guidance_scale=2., do_rotate=True, num_inference_steps=30, **kwargs)
app/custom_models/normal_prediction.py:13
↓ 1 callersFunctionprepend_dummies
prepend dummy elements to vertices and faces to enable "masked" scatter operations
mesh_reconstruction/remesh.py:7
↓ 1 callersFunctionproject_color
Projects color from a given image onto a 3D mesh. Args: meshes (pytorch3d.structures.Meshes): The 3D mesh object. cameras (p
scripts/project_mesh.py:108
↓ 1 callersFunctionpy3dmesh_to_meshlab_mesh
(meshes: Meshes)
scripts/utils.py:66
↓ 1 callersFunctionrecursive_change_processors
(name: str, module: torch.nn.Module)
custum_3d_diffusion/custum_modules/attention_processors.py:336
↓ 1 callersFunctionrefine_lr_with_sd
(pil_image_list, concept_img_list, control_image_list, prompt_list, pipe=None, strength=0.35, neg_prompt_list=
scripts/refine_lr_to_sr.py:11
↓ 1 callersFunctionrefine_rgb
(rgb_pils, front_pil)
scripts/multiview_inference.py:36
↓ 1 callersFunctionremesh
( vertices_etc:torch.Tensor, #V,D faces:torch.Tensor, #F,3 long min_edgelen:torch.Tens
mesh_reconstruction/opt.py:9
↓ 1 callersFunctionremove_color
(arr)
app/utils.py:16
↓ 1 callersFunctionremove_dummies
remove dummy elements added with prepend_dummies()
mesh_reconstruction/remesh.py:17
↓ 1 callersFunctionrender_mesh_vertex_color
(mesh, cameras, H, W, blur_radius=0.0, faces_per_pixel=1, bkgd=(0., 0., 0.), dtype=torch.float32, device="cuda
mesh_reconstruction/render.py:76
↓ 1 callersMethodrender_pix2faces_nvdiff
(self, meshes: Meshes, cameras: CamerasBase, H=512, W=512)
scripts/project_mesh.py:89
↓ 1 callersFunctionrotate_normalmap_by_angle_torch
rotate along y-axis normal_map: torch.Tensor, shape=(H, W, 3) in [-1, 1], device='cuda' angle: float, in degree
scripts/utils.py:130
↓ 1 callersFunctionrotate_normals_torch
(normal_pils, return_types='np', rotate_direction=1)
scripts/utils.py:150
↓ 1 callersFunctionrotate_vector_field_normals
(normals: np.ndarray, angle: float)
scripts/normal_to_height_map.py:84
↓ 1 callersFunctionrun_mesh_refine
(vertices, faces, pils: List[Image.Image], steps=100, start_edge_len=0.02, end_edge_len=0.005, decay=0.99, upd
mesh_reconstruction/refine.py:12
↓ 1 callersFunctionsave_py3dmesh_with_trimesh_fast
(meshes: Meshes, save_glb_path, apply_sRGB_to_LinearRGB=True)
scripts/utils.py:220
↓ 1 callersMethodset_config
(self, attn_config: AttnConfig)
custum_3d_diffusion/custum_modules/unifield_processor.py:82
↓ 1 callersMethodset_config
(self, attn_config: AttnConfig)
custum_3d_diffusion/custum_modules/unifield_processor.py:187
↓ 1 callersFunctionsimple_clean_mesh
(pyml_mesh: ml.Mesh, apply_smooth=True, stepsmoothnum=1, apply_sub_divide=False, sub_divide_threshold=0.25)
scripts/utils.py:255
↓ 1 callersFunctionsimple_preprocess
(input_image, rembg_session=session, background_color=255)
scripts/utils.py:282
↓ 1 callersFunctionsplit_edges
( vertices:torch.Tensor, #V,3 first unused faces:torch.Tensor, #F,3 long, 0 for unused
mesh_reconstruction/remesh.py:148
↓ 1 callersFunctionsrgb_to_linear
(c_srgb)
scripts/utils.py:215
↓ 1 callersFunctionto_pyml_mesh
(vertices,faces)
scripts/utils.py:76
↓ 1 callersFunctiontransform_back_normal_to_front
(normal_pil)
scripts/mesh_init.py:60
↓ 1 callersMethodtransform_vertices
(self, meshes: Meshes, cameras: CamerasBase)
scripts/project_mesh.py:74
↓ 1 callersMethodunload_ip_adapter
(self)
custum_3d_diffusion/custum_modules/unifield_processor.py:176
↓ 1 callersMethodupdate_config
(self, attn_config: AttnConfig, force_update=False)
custum_3d_diffusion/custum_modules/unifield_processor.py:368
↓ 1 callersMethodvalidate_trainer_config
(self)
custum_3d_diffusion/trainings/base.py:81
Method__call__
( self, attn: Attention, hidden_states, encoder_hidden_states=None, attention_mask=None, ref_d
custum_3d_diffusion/custum_modules/attention_processors.py:75
Method__call__
( self, attn: Attention, hidden_states: torch.FloatTensor, encoder_hidden_stat
custum_3d_diffusion/custum_modules/attention_processors.py:216
Method__call__
( self, *args, **kwargs )
custum_3d_diffusion/custum_modules/attention_processors.py:302
Method__call__
r""" The call function to the pipeline for generation. Args: image (`Image.Image` or `List[Image.Image]` or `torch.FloatT
custum_3d_diffusion/custum_pipeline/unifield_pipeline_img2mvimg.py:100
Method__call__
r""" The call function to the pipeline for generation. Args: image (`Image.Image` or `List[Image.Image]` or `torch.FloatT
custum_3d_diffusion/custum_pipeline/unifield_pipeline_img2img.py:100
Method__getattr__
(self, name: str)
custum_3d_diffusion/custum_modules/unifield_processor.py:160
Method__getattr__
(self, name: str)
custum_3d_diffusion/custum_modules/attention_processors.py:277
Method__init__
(self, base_model=None)
app/all_models.py:11
Method__init__
(self, scale, onnx_path, tile=0, tile_pad=
scripts/upsampler.py:26
Method__init__
(self, device="cuda")
scripts/project_mesh.py:69
Method__init__
( self, accelerator: Accelerator, logger: MultiProcessAdapter, unet: Configu
custum_3d_diffusion/trainings/base.py:48
Method__init__
(self, init_config: AttnConfig, weight_dtype)
custum_3d_diffusion/custum_modules/unifield_processor.py:139
Method__init__
( self, chained_proc, enabled=False, name=None, hidden_states_dim=None
custum_3d_diffusion/custum_modules/attention_processors.py:194
Method__init__
(self, module)
custum_3d_diffusion/custum_modules/attention_processors.py:270
Method__init__
( self, proc_dict: dict, enabled_proc="default", name=None, switch_nam
custum_3d_diffusion/custum_modules/attention_processors.py:284
Method__init__
( self, vae: AutoencoderKL, image_encoder: CLIPVisionModelWithProjection, unet
custum_3d_diffusion/custum_pipeline/unifield_pipeline_img2mvimg.py:34
Method__init__
( self, vae: AutoencoderKL, image_encoder: CLIPVisionModelWithProjection, unet
custum_3d_diffusion/custum_pipeline/unifield_pipeline_img2img.py:34
Method__init__
(self, vertices:torch.Tensor, #V,3 faces:torch.Tensor, #F,3 lr=0.3, #lear
mesh_reconstruction/opt.py:64
Method__init__
( self, mv: torch.Tensor, #C,4,4 proj: torch.Tensor, #C,4,4 im
mesh_reconstruction/render.py:21
Method__init__
(self, cameras, image_size, device)
mesh_reconstruction/render.py:117
Methodbatched_validation_forward
(self, pipeline, **pipeline_call_kwargs)
custum_3d_diffusion/trainings/image2mvimage_trainer.py:138
Methodbatched_validation_forward
(self, pipeline, **pipeline_call_kwargs)
custum_3d_diffusion/trainings/image2image_trainer.py:85
Functioncache_model
(func)
scripts/sd_model_zoo.py:10
Functioncaller
(*args)
scripts/load_onnx.py:28
Functionchange_bkgd
(img_pils, new_bkgd=(0., 0., 0.))
scripts/utils.py:165
Functionchange_bkgd_to_normal
(normal_pils)
scripts/utils.py:181
Functionclean_up
()
app/utils.py:12
Functionconcept_to_multiview
(preview_img, input_processing, seed, guidance=1.)
app/gradio_3dgen_steps.py:8
Methodconfigure
(self)
custum_3d_diffusion/trainings/image2mvimage_trainer.py:50
Methodconstruct_pipeline
(self, shared_modules, unet, old_version=False)
custum_3d_diffusion/trainings/image2mvimage_trainer.py:84
Methodconstruct_pipeline
(self, shared_modules, unet, old_version=False)
custum_3d_diffusion/trainings/image2image_trainer.py:35
Functioncopied_cache_model
(func)
scripts/sd_model_zoo.py:21
Functioncreate_step_ui
(concurrency_id="wkl")
app/gradio_3dgen_steps.py:83
Functioncreate_ui
(concurrency_id="wkl")
app/gradio_3dgen.py:31
Functiondecorator
(cls)
custum_3d_diffusion/modules.py:4
Methoddo_paritial_crosspond
(hidden_states, ref_state)
custum_3d_diffusion/custum_modules/attention_processors.py:123
Methoddo_set_config
(self, attn_config: AttnConfig)
custum_3d_diffusion/custum_modules/unifield_processor.py:88
Methoddo_update_config
(self, attn_config: AttnConfig)
custum_3d_diffusion/custum_modules/unifield_processor.py:98
Methodfaces
(self)
mesh_reconstruction/opt.py:109
Functionfind
(name)
custum_3d_diffusion/modules.py:11
Functionforward
r""" The forward method of the `Attention` class. Args: hidden_states (`torch.Tensor`): The hidden states of the query.
custum_3d_diffusion/custum_modules/attention_processors.py:351
Methodforward
(self, *args, **kwargs)
custum_3d_diffusion/custum_modules/unifield_processor.py:112
Methodforward
(self, *args, **kwargs)
custum_3d_diffusion/custum_modules/attention_processors.py:274
Methodforward
(self, fragments, meshes, **kwargs)
mesh_reconstruction/render.py:71
Methodforward_step
input a batch return a loss
custum_3d_diffusion/trainings/base.py:119
Methodforward_step
(self, batch, unet, shared_modules, noise_scheduler: DDPMScheduler, global_step)
custum_3d_diffusion/trainings/image2mvimage_trainer.py:81
Methodforward_step
(self, batch, unet, shared_modules, noise_scheduler: DDPMScheduler, global_step)
custum_3d_diffusion/trainings/image2image_trainer.py:32
Functiongenerate3dv2
(preview_img, input_processing, seed, render_video=True, do_refine=True, expansion_weight=0.1, init_type="std"
app/gradio_3dgen.py:12
← previousnext →101–200 of 239, ranked by callers