MCPcopy Index your code

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

Functions848 in github.com/ShanghaiTech-IMPACT/TeethDreamer

⨍ Functions 848◇ Types & classes 168↳ Endpoints 1
↓ 1 callersMethodadd_noise
@param x_start: B,* @param t: B, @return:
ldm/models/diffusion/zero123.py:353
↓ 1 callersMethodadd_noise
@param x_start: B,* @param t: B, @return:
ldm/models/diffusion/teeth_dreamer.py:502
↓ 1 callersMethodadd_noise
@param x_start: B,* @param t: B, @return:
ldm/models/diffusion/sync_dreamer.py:762
↓ 1 callersFunctionavg_pool_nd
Create a 1D, 2D, or 3D average pooling module.
ldm/modules/diffusionmodules/util.py:238
↓ 1 callersFunctioncamNormal2worldNormal
(rot_c2w, camNormal)
ldm/base_utils.py:7
↓ 1 callersMethodcheck_frequency
(self, check_idx)
TeethDreamer.py:131
↓ 1 callersFunctionclick_seg
(image_path, mask_path)
seg_foreground.py:65
↓ 1 callersFunctioncompute_dR_dt
(Rt0, Rt1)
ldm/base_utils.py:87
↓ 1 callersFunctionconcat_images
(img0,img1,vert=False)
ldm/base_utils.py:94
↓ 1 callersFunctioncreate_spheric_poses
(cameras, n_steps=120)
instant-nsr-pl/datasets/dtu.py:36
↓ 1 callersFunctioncreate_spheric_poses
(cameras, n_steps=120)
instant-nsr-pl/datasets/colmap.py:112
↓ 1 callersFunctioncreate_sun_light
(loc=[0, 0, 0], rot=[0, 0, 0])
normal_render.py:225
↓ 1 callersFunctioncreate_sun_light
(loc=[0, 0, 0], rot=[0, 0, 0])
color_render.py:236
↓ 1 callersMethoddecode
(self, text)
ldm/modules/encoders/modules.py:146
↓ 1 callersFunctiondefault
(val, d)
ldm/util.py:136
↓ 1 callersMethoddenoise_apply
@param x_target_noisy: B,N,4,H,W @param input_info: @param clip_embed: B,M,768 @param time_steps: B,
ldm/models/diffusion/zero123.py:528
↓ 1 callersMethoddenoise_apply
@param x_target_noisy: B,N,4,H,W @param input_info: @param clip_embed: B,M,768 @param time_steps: B,
ldm/models/diffusion/teeth_dreamer.py:672
↓ 1 callersMethoddenoise_apply
@param x_target_noisy: B,N,4,H,W @param input_info: @param clip_embed: B,M,768 @param time_steps: B,
ldm/models/diffusion/sync_dreamer.py:973
↓ 1 callersMethoddenoise_apply_impl
@param x_target_noisy: B,N,4,H,W @param index: index @param noise_pred: B,N,4,H,W @param is_step0:
ldm/models/diffusion/zero123.py:502
↓ 1 callersMethoddenoise_apply_impl
@param x_target_noisy: B,N,4,H,W @param index: index @param noise_pred: B,N,4,H,W @param is_step0:
ldm/models/diffusion/teeth_dreamer.py:646
↓ 1 callersMethoddenoise_apply_impl
@param x_target_noisy: B,N,4,H,W @param index: index @param noise_pred: B,N,4,H,W @param is_step0:
ldm/models/diffusion/sync_dreamer.py:947
↓ 1 callersFunctiondot
(x, y)
instant-nsr-pl/models/utils.py:100
↓ 1 callersFunctiondraw_epipolar_line
(F, img0, img1, pt0, color)
ldm/base_utils.py:45
↓ 1 callersFunctiondump_config
(path, config)
instant-nsr-pl/utils/misc.py:38
↓ 1 callersMethodencode
(self, x)
ldm/models/autoencoder.py:324
↓ 1 callersMethodencode_with_pretrained
(self,x)
ldm/modules/diffusionmodules/model.py:816
↓ 1 callersFunctionequals
(val)
ldm/modules/x_transformer.py:76
↓ 1 callersMethodexport
(self)
instant-nsr-pl/systems/nerf.py:213
↓ 1 callersMethodexport
(self)
instant-nsr-pl/systems/neus.py:425
↓ 1 callersMethodforward
(self, x)
ldm/modules/diffusionmodules/util.py:210
↓ 1 callersMethodforward_
(self, rays)
instant-nsr-pl/models/nerf.py:61
↓ 1 callersMethodforward_
(self, rays)
instant-nsr-pl/models/neus.py:205
↓ 1 callersMethodforward_bg_
(self, rays)
instant-nsr-pl/models/neus.py:141
↓ 1 callersMethodforward_level
(self, points)
instant-nsr-pl/models/geometry.py:80
↓ 1 callersMethodforward_level
(self, points)
instant-nsr-pl/models/geometry.py:132
↓ 1 callersMethodfreeze
(self)
ldm/modules/encoders/modules.py:194
↓ 1 callersMethodfreeze
(self)
ldm/modules/encoders/modules.py:264
↓ 1 callersMethodfunc
(self)
instant-nsr-pl/systems/criterions.py:8
↓ 1 callersFunctiongen_poses
(basedir, match_type)
instant-nsr-pl/scripts/imgs2poses.py:58
↓ 1 callersMethodget_alpha
(self, sdf, normal, dirs, dists)
instant-nsr-pl/models/neus.py:117
↓ 1 callersMethodget_bg_color
(self)
ldm/data/teeth_dreamer.py:46
↓ 1 callersFunctionget_blocks
(num_layers)
ldm/thirdp/psp/helpers.py:31
↓ 1 callersFunctionget_calibration_matrix_K_from_blender
(camera)
normal_render.py:135
↓ 1 callersFunctionget_center
(pts)
instant-nsr-pl/datasets/colmap.py:20
↓ 1 callersMethodget_data_for_index
(self, index)
ldm/data/teeth_dreamer.py:96
↓ 1 callersMethodget_data_for_index
(self, index)
ldm/data/teeth_dreamer.py:185
↓ 1 callersMethodget_data_for_index
(self, index)
ldm/data/teeth_dreamer.py:237
↓ 1 callersMethodget_drop_scheme
(self, B, device)
ldm/models/diffusion/zero123.py:55
↓ 1 callersMethodget_drop_scheme
(self, B, device)
ldm/models/diffusion/teeth_dreamer.py:55
↓ 1 callersMethodget_drop_scheme
(self, B, device)
ldm/models/diffusion/sync_dreamer.py:55
↓ 1 callersMethodget_file_list
(self)
instant-nsr-pl/utils/callbacks.py:56
↓ 1 callersMethodget_image_grid_
(self, imgs)
instant-nsr-pl/utils/mixins.py:123
↓ 1 callersMethodget_null_cond
(self, version, max_length)
ldm/modules/encoders/modules.py:301
↓ 1 callersFunctionget_obj_from_str
(string, reload=False)
ldm/util.py:167
↓ 1 callersFunctionget_parameters
(model, name)
instant-nsr-pl/systems/utils.py:305
↓ 1 callersFunctionget_parser
(**parser_kwargs)
TeethDreamer.py:25
↓ 1 callersFunctionget_scheduler
(name)
instant-nsr-pl/systems/utils.py:290
↓ 1 callersFunctionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: From Fairseq. Build sinusoidal embeddings. This matches
ldm/modules/diffusionmodules/model.py:12
↓ 1 callersMethodget_trainable_parameters
(self)
ldm/models/diffusion/zero123.py:52
↓ 1 callersMethodget_trainable_parameters
(self)
ldm/models/diffusion/teeth_dreamer.py:52
↓ 1 callersMethodget_trainable_parameters
(self)
ldm/models/diffusion/sync_dreamer.py:52
↓ 1 callersFunctiongetattr_recursive
(m, attr)
instant-nsr-pl/systems/utils.py:299
↓ 1 callersMethodinit_
(self)
ldm/modules/x_transformer.py:31
↓ 1 callersMethodinit_
(self)
ldm/modules/x_transformer.py:595
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
ldm/models/autoencoder.py:78
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
ldm/models/autoencoder.py:313
↓ 1 callersMethodinstantiate_pretrained
(self, config)
ldm/modules/diffusionmodules/model.py:807
↓ 1 callersMethodisosurface
(self)
instant-nsr-pl/models/nerf.py:57
↓ 1 callersMethodisosurface
(self)
instant-nsr-pl/models/neus.py:113
↓ 1 callersMethodisosurface_
(self, vmin, vmax)
instant-nsr-pl/models/geometry.py:171
↓ 1 callersFunctionl2_norm
(input, axis=1)
ldm/thirdp/psp/helpers.py:17
↓ 1 callersFunctionload_K_Rt_from_P
(P=None)
instant-nsr-pl/datasets/dtu.py:20
↓ 1 callersFunctionload_config
(*yaml_files, cli_args=[])
instant-nsr-pl/utils/misc.py:26
↓ 1 callersFunctionload_config
(*yaml_files, cli_args=[])
ldm/util.py:20
↓ 1 callersFunctionload_pretrain_stable_diffusion
(new_model, finetune_from)
TeethDreamer.py:188
↓ 1 callersMethodlog_image
(self, x_sample, batch, step, output_dir)
ldm/models/diffusion/zero123.py:394
↓ 1 callersMethodlog_image
(self, x_sample, batch, step, output_dir)
ldm/models/diffusion/teeth_dreamer.py:543
↓ 1 callersMethodlog_image
(self, x_sample, batch, step, output_dir)
ldm/models/diffusion/sync_dreamer.py:803
↓ 1 callersMethodlog_images
(self, batch, only_inputs=False, plot_ema=False, **kwargs)
ldm/models/autoencoder.py:233
↓ 1 callersMethodlog_to_file
(self, save_dir, split, images, global_step, current_epoch)
TeethDreamer.py:94
↓ 1 callersFunctionmain
()
instant-nsr-pl/launch.py:9
↓ 1 callersFunctionmake_beta_schedule
(schedule, n_timestep, linear_start=1e-4, linear_end=2e-2, cosine_s=8e-3)
ldm/modules/diffusionmodules/util.py:21
↓ 1 callersMethodmatch_log_keys
(self, s)
instant-nsr-pl/utils/loggers.py:15
↓ 1 callersFunctionmax_neg_value
(tensor)
ldm/modules/x_transformer.py:82
↓ 1 callersFunctionnear_far_from_unit_sphere_using_camera_poses
@param camera_poses: b 3 4 @return: near: b,1 far: b,1
ldm/models/diffusion/sync_dreamer_utils.py:86
↓ 1 callersFunctionnormal2img
(normal)
instant-nsr-pl/datasets/blender.py:50
↓ 1 callersFunctionnormalize_poses
(poses, pts, up_est_method, center_est_method)
instant-nsr-pl/datasets/colmap.py:29
↓ 1 callersFunctionnot_equals
(val)
ldm/modules/x_transformer.py:70
↓ 1 callersFunctionparse_optimizer
(config, model)
instant-nsr-pl/systems/utils.py:314
↓ 1 callersFunctionparse_scheduler
(config, optimizer)
instant-nsr-pl/systems/utils.py:328
↓ 1 callersMethodpredict_dataloader
(self)
instant-nsr-pl/datasets/dtu.py:200
↓ 1 callersMethodprepare
(self)
instant-nsr-pl/systems/base.py:22
↓ 1 callersMethodprepare_data
(self)
instant-nsr-pl/datasets/dtu.py:178
↓ 1 callersFunctionprepare_masked_img
(img_path, out_path, rembg=True, normal=False, real=False)
instant-nsr-pl/tools.py:89
↓ 1 callersMethodpreprocess
(self, x)
ldm/modules/encoders/modules.py:307
↓ 1 callersMethodpreprocess
(self, x)
ldm/modules/encoders/modules.py:363
↓ 1 callersMethodpreprocess
(self, x)
ldm/modules/encoders/modules.py:409
↓ 1 callersMethodprocess_im
(self, im)
ldm/data/teeth_dreamer.py:83
↓ 1 callersFunctionproject_and_normalize
@param ref_grid: b 3 n @param src_proj: b 4 4 @param length: int @return: b, n, 2
ldm/models/diffusion/sync_dreamer_utils.py:5
↓ 1 callersMethodq_sample
(self, x_start, t, noise=None)
ldm/modules/encoders/modules.py:517
← previousnext →201–300 of 848, ranked by callers