MCPcopy Index your code

hub / github.com/ShanghaiTech-IMPACT/TeethDreamer / functions

Functions848 in github.com/ShanghaiTech-IMPACT/TeethDreamer

↓ 2 callersMethodconstruct_view_frustum_volume
@param spatial_volume: B,C,V,V,V @param t_embed: B,t_dim @param v_embed: B,N,v_dim @param pose
ldm/models/diffusion/teeth_dreamer.py:203
↓ 2 callersMethodconstruct_view_frustum_volume
@param spatial_volume: B,C,V,V,V @param t_embed: B,t_dim @param v_embed: B,N,v_dim @param pose
ldm/models/diffusion/sync_dreamer.py:203
↓ 2 callersFunctioncount_flops_attn
A counter for the `thop` package to count the operations in an attention operation. Meant to be used like: macs, params = thop.pr
ldm/modules/diffusionmodules/openaimodel.py:328
↓ 2 callersFunctioncount_params
(model, verbose=False)
ldm/util.py:150
↓ 2 callersFunctioncreate_target_volume
(depth_size, volume_size, input_image_size, pose_target, K, near=None, far=None)
ldm/models/diffusion/sync_dreamer_utils.py:48
↓ 2 callersMethoddecode
(self, z)
ldm/models/autoencoder.py:330
↓ 2 callersMethoddecode_first_stage
(self, z)
ldm/models/diffusion/zero123.py:247
↓ 2 callersMethoddecode_first_stage
(self, z)
ldm/models/diffusion/teeth_dreamer.py:384
↓ 2 callersMethoddecode_first_stage
(self, z)
ldm/models/diffusion/sync_dreamer.py:449
↓ 2 callersMethoddrop
(self, cond, mask)
ldm/models/diffusion/zero123.py:46
↓ 2 callersMethodema_scope
(self, context=None)
ldm/models/autoencoder.py:64
↓ 2 callersMethodembed_time
(self, t)
ldm/models/diffusion/teeth_dreamer.py:422
↓ 2 callersMethodembed_time
(self, t)
ldm/models/diffusion/sync_dreamer.py:488
↓ 2 callersMethodencode_first_stage
(self, x, sample=True)
ldm/models/diffusion/zero123.py:239
↓ 2 callersMethodencode_first_stage
(self, x, sample=True)
ldm/models/diffusion/teeth_dreamer.py:376
↓ 2 callersMethodencode_first_stage
(self, x, sample=True)
ldm/models/diffusion/sync_dreamer.py:441
↓ 2 callersFunctionexists
(val)
ldm/modules/attention.py:11
↓ 2 callersFunctionextract_into_tensor
(a, t, x_shape)
ldm/modules/diffusionmodules/util.py:96
↓ 2 callersMethodfind_in_interval
(self, n)
ldm/lr_scheduler.py:52
↓ 2 callersFunctionget_3x4_RT_matrix_from_blender
(cam)
normal_render.py:281
↓ 2 callersFunctionget_3x4_RT_matrix_from_blender
(cam)
color_render.py:292
↓ 2 callersFunctionget_c2w
(R,t)
instant-nsr-pl/tools.py:59
↓ 2 callersFunctionget_calibration_matrix_K_from_blender
(camera)
color_render.py:146
↓ 2 callersFunctionget_encoding_with_network
(n_input_dims, n_output_dims, encoding_config, network_config)
instant-nsr-pl/models/network_utils.py:200
↓ 2 callersMethodget_grayscale_image_
(self, img, data_range, cmap)
instant-nsr-pl/utils/mixins.py:86
↓ 2 callersFunctionget_optional_dict
(name, config)
TeethDreamer.py:221
↓ 2 callersMethodget_uv_image_
(self, img, data_format, data_range, cmap)
instant-nsr-pl/utils/mixins.py:59
↓ 2 callersMethodget_viewpoint_embedding
@param dp {da,de,dz} @return: cam_pose_embedding
ldm/models/diffusion/zero123.py:169
↓ 2 callersMethodget_viewpoint_embedding
@param dp {da,de,dz} @return: cam_pose_embedding
ldm/models/diffusion/teeth_dreamer.py:301
↓ 2 callersMethodget_viewpoint_embedding
@param dp {da,de,dz} @return: cam_pose_embedding
ldm/models/diffusion/sync_dreamer.py:355
↓ 2 callersFunctionget_warp_coordinates
(volume_xyz, warp_size, input_size, Ks, warp_pose)
ldm/models/diffusion/sync_dreamer_utils.py:40
↓ 2 callersMethodgrid_vertices
(self)
instant-nsr-pl/models/geometry.py:46
↓ 2 callersFunctiongroup_dict_by_key
(cond, d)
ldm/modules/x_transformer.py:93
↓ 2 callersFunctiongroupby_prefix_and_trim
(prefix, d)
ldm/modules/x_transformer.py:110
↓ 2 callersMethodisosurface_
(self, vmin, vmax)
instant-nsr-pl/models/geometry.py:83
↓ 2 callersMethodload_im
(self, path, bg_color, normal=False, **kwargs)
ldm/data/teeth_dreamer.py:66
↓ 2 callersFunctionload_object
Loads a glb model into the scene.
normal_render.py:187
↓ 2 callersFunctionload_object
Loads a glb model into the scene.
color_render.py:198
↓ 2 callersMethodlog_img
(self, pl_module, batch, split="train")
TeethDreamer.py:108
↓ 2 callersMethodmake_activation
(self)
instant-nsr-pl/models/network_utils.py:135
↓ 2 callersFunctionparse_version
(ver)
instant-nsr-pl/utils/misc.py:53
↓ 2 callersMethodpredict_with_unconditional_scale
(self, x, t, clip_embed, x_concat, unconditional_scale, **kwargs)
ldm/models/diffusion/zero123.py:99
↓ 2 callersMethodpredict_with_unconditional_scale
(self, x, t, clip_embed, volume_feats, x_concat, unconditional_scale, **kwargs)
ldm/models/diffusion/teeth_dreamer.py:103
↓ 2 callersMethodprepare
(self, batch)
ldm/models/diffusion/zero123.py:252
↓ 2 callersMethodprepare
(self, batch)
ldm/models/diffusion/teeth_dreamer.py:389
↓ 2 callersMethodprepare
(self, batch)
ldm/models/diffusion/sync_dreamer.py:454
↓ 2 callersFunctionprepare_inputs
(image_paths, input_cam, target_cams, crop_size=-1, image_size=256)
ldm/util.py:284
↓ 2 callersMethodquantize
(self, x, *args, **kwargs)
ldm/models/autoencoder.py:437
↓ 2 callersFunctionread_cameras_binary
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCameras
instant-nsr-pl/datasets/colmap_utils.py:107
↓ 2 callersFunctionread_images_binary
see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBi
instant-nsr-pl/datasets/colmap_utils.py:167
↓ 2 callersFunctionread_pickle
(pkl_path)
instant-nsr-pl/datasets/blender.py:53
↓ 2 callersFunctionread_points3d_binary
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints
instant-nsr-pl/datasets/colmap_utils.py:229
↓ 2 callersFunctionrender
(azimuths, elevations, distances, output_dir, object_uid, num_images)
color_render.py:329
↓ 2 callersFunctionrepeat_to_batch
(tensor, B, VN)
ldm/models/diffusion/zero123.py:31
↓ 2 callersFunctionrepeat_to_batch
(tensor, B, VN)
ldm/models/diffusion/teeth_dreamer.py:31
↓ 2 callersFunctionreset_scene
Resets the scene to a clean state.
normal_render.py:169
↓ 2 callersFunctionreset_scene
Resets the scene to a clean state.
color_render.py:180
↓ 2 callersMethodsample
(self, sampler, batch, cfg_scale, batch_view_num, return_inter_results=False, inter_interval=50, inter_view_in
ldm/models/diffusion/sync_dreamer.py:778
↓ 2 callersMethodsample
@param input_info: x, elevation @param clip_embed: B,M,768 @param unconditional_scale: @param log_every_t:
ldm/models/diffusion/sync_dreamer.py:1100
↓ 2 callersMethodsave_img_sequence
(self, filename, img_dir, matcher, save_format='gif', fps=30)
instant-nsr-pl/utils/mixins.py:191
↓ 2 callersMethodsave_mesh
(self, filename, v_pos, t_pos_idx, v_tex=None, t_tex_idx=None, v_rgb=None)
instant-nsr-pl/utils/mixins.py:211
↓ 2 callersFunctionscene_bbox
(single_obj=None, ignore_matrix=False)
normal_render.py:251
↓ 2 callersFunctionscene_bbox
(single_obj=None, ignore_matrix=False)
color_render.py:262
↓ 2 callersFunctionscene_root_objects
()
normal_render.py:268
↓ 2 callersFunctionscene_root_objects
()
color_render.py:279
↓ 2 callersFunctionset_camera_location
(cam_pt)
normal_render.py:120
↓ 2 callersFunctionset_camera_location
(cam_pt)
color_render.py:131
↓ 2 callersFunctionset_object_rotation
(object_uid, rot_euler)
normal_render.py:129
↓ 2 callersFunctionset_object_rotation
(object_uid, rot_euler)
color_render.py:140
↓ 2 callersMethodsetup
(self, stage=None)
instant-nsr-pl/datasets/dtu.py:168
↓ 2 callersMethodsetup
(self, stage=None)
instant-nsr-pl/datasets/blender.py:193
↓ 2 callersMethodsetup
(self, stage=None)
instant-nsr-pl/datasets/colmap.py:299
↓ 2 callersMethodstate_dict
Returns the state of the scheduler as a :class:`dict`. It contains an entry for every variable in self.__dict__ which is not the opti
instant-nsr-pl/systems/utils.py:133
↓ 2 callersMethodto_rgb
(self, x)
ldm/models/autoencoder.py:417
↓ 2 callersMethodtrain
(self, mode=True)
instant-nsr-pl/models/base.py:21
↓ 1 callersMethod__init__
(self, batch_frequency, max_images, log_images_kwargs=None)
TeethDreamer.py:78
↓ 1 callersMethod__init__
(self, optimizer, start_factor=1.0 / 3, end_factor=1.0, total_iters=5, last_epoch=-1, verbose
instant-nsr-pl/systems/utils.py:255
↓ 1 callersMethod__init__
(self, config, split)
instant-nsr-pl/datasets/dtu.py:141
↓ 1 callersMethod__init__
(self, config, split)
instant-nsr-pl/datasets/blender.py:166
↓ 1 callersMethod__init__
(self, config, split)
instant-nsr-pl/datasets/colmap.py:272
↓ 1 callersMethod__init__
(self, resolution, use_torch=True)
instant-nsr-pl/models/geometry.py:33
↓ 1 callersMethod__init__
(self, config)
instant-nsr-pl/models/texture.py:41
↓ 1 callersMethod_gaussian_or_uniform_kernel
(self, kernel_size, sigma)
instant-nsr-pl/systems/criterions.py:91
↓ 1 callersMethod_get_next_version
(self)
instant-nsr-pl/utils/callbacks.py:33
↓ 1 callersMethod_init_clip_image_encoder
(self)
ldm/models/diffusion/zero123.py:211
↓ 1 callersMethod_init_clip_image_encoder
(self)
ldm/models/diffusion/teeth_dreamer.py:340
↓ 1 callersMethod_init_clip_image_encoder
(self)
ldm/models/diffusion/sync_dreamer.py:405
↓ 1 callersMethod_init_clip_projection
(self)
ldm/models/diffusion/zero123.py:160
↓ 1 callersMethod_init_clip_projection
(self)
ldm/models/diffusion/teeth_dreamer.py:292
↓ 1 callersMethod_init_first_stage
(self)
ldm/models/diffusion/zero123.py:186
↓ 1 callersMethod_init_first_stage
(self)
ldm/models/diffusion/teeth_dreamer.py:315
↓ 1 callersMethod_init_first_stage
(self)
ldm/models/diffusion/sync_dreamer.py:380
↓ 1 callersMethod_init_schedule
(self)
ldm/models/diffusion/zero123.py:215
↓ 1 callersMethod_init_schedule
(self)
ldm/models/diffusion/teeth_dreamer.py:344
↓ 1 callersMethod_init_schedule
(self)
ldm/models/diffusion/sync_dreamer.py:409
↓ 1 callersMethod_init_time_step_embedding
(self)
ldm/models/diffusion/teeth_dreamer.py:368
↓ 1 callersMethod_init_time_step_embedding
(self)
ldm/models/diffusion/sync_dreamer.py:433
↓ 1 callersMethod_make_schedule
(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True)
ldm/models/diffusion/zero123.py:484
↓ 1 callersMethod_make_schedule
(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True)
ldm/models/diffusion/teeth_dreamer.py:628
↓ 1 callersMethod_make_schedule
(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True)
ldm/models/diffusion/sync_dreamer.py:929
← previousnext →101–200 of 848, ranked by callers