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 callersFunctionqvec2rotmat
(qvec)
instant-nsr-pl/datasets/colmap_utils.py:271
↓ 1 callersMethodqvec2rotmat
(self)
instant-nsr-pl/datasets/colmap_utils.py:48
↓ 1 callersFunctionread_cameras_text
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasText(const std::string& path) void Reconstruction::ReadCamerasTe
instant-nsr-pl/datasets/colmap_utils.py:81
↓ 1 callersFunctionread_images_text
see: src/base/reconstruction.cc void Reconstruction::ReadImagesText(const std::string& path) void Reconstruction::WriteImagesText
instant-nsr-pl/datasets/colmap_utils.py:136
↓ 1 callersFunctionread_pickle
(pkl_path)
instant-nsr-pl/tools.py:55
↓ 1 callersFunctionread_points3D_text
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3D
instant-nsr-pl/datasets/colmap_utils.py:202
↓ 1 callersFunctionrecon
(path, out_path, normal=False, rembg=False, real=False, cpu=8)
instant-nsr-pl/run.py:5
↓ 1 callersMethodregister_schedule
(self, beta_schedule="linear", timesteps=1000, linear_start=1e-4, linear_end=2e-2, c
ldm/modules/encoders/modules.py:490
↓ 1 callersFunctionremove_sun_light
()
normal_render.py:239
↓ 1 callersFunctionremove_sun_light
()
color_render.py:250
↓ 1 callersFunctionrender
(azimuths, elevations, distances, output_dir, object_uid, num_images)
normal_render.py:318
↓ 1 callersFunctionrun_colmap
(basedir, match_type)
instant-nsr-pl/scripts/imgs2poses.py:12
↓ 1 callersMethodsample
(self)
ldm/modules/distributions/distributions.py:17
↓ 1 callersMethodsave_code_snapshot
(self)
instant-nsr-pl/utils/callbacks.py:64
↓ 1 callersMethodsave_config_snapshot
(self)
instant-nsr-pl/utils/callbacks.py:85
↓ 1 callersMethodsave_image
(self, x_sample, batch, output_dir)
ldm/models/diffusion/zero123.py:422
↓ 1 callersMethodsave_image
(self, x_sample, batch, output_dir)
ldm/models/diffusion/teeth_dreamer.py:570
↓ 1 callersMethodsave_image
(self, x_sample, batch, output_dir)
ldm/models/diffusion/sync_dreamer.py:844
↓ 1 callersFunctionsave_images
(object_file, output_dir, num_images)
normal_render.py:380
↓ 1 callersFunctionsave_images
(object_file, output_dir, num_images)
color_render.py:395
↓ 1 callersFunctionsave_pickle
(data, pkl_path)
color_render.py:41
↓ 1 callersFunctionscene_meshes
()
normal_render.py:274
↓ 1 callersFunctionscene_meshes
()
color_render.py:285
↓ 1 callersMethodschedule
(self, n, **kwargs)
ldm/lr_scheduler.py:17
↓ 1 callersMethodschedule
(self, n, **kwargs)
ldm/lr_scheduler.py:59
↓ 1 callersFunctionset_sun_light
(loc, rot)
normal_render.py:245
↓ 1 callersFunctionset_sun_light
(loc, rot)
color_render.py:256
↓ 1 callersFunctionshow_mask
(mask, ax, random_color=False)
seg_foreground.py:15
↓ 1 callersFunctionshow_mask
(mask, ax, random_color=False)
seg_teeth.py:15
↓ 1 callersFunctionsphere_init_tcnn_network
(n_input_dims, n_output_dims, config, network)
instant-nsr-pl/models/network_utils.py:142
↓ 1 callersMethodstep
(self)
instant-nsr-pl/systems/utils.py:125
↓ 1 callersMethodtest_dataloader
(self)
instant-nsr-pl/datasets/dtu.py:197
↓ 1 callersMethodtrain_dataloader
(self)
instant-nsr-pl/datasets/dtu.py:191
↓ 1 callersFunctiontrainer_args
(opt)
TeethDreamer.py:49
↓ 1 callersFunctiontrans_normal
(normal, RT_w2c_src, RT_w2c_tgt)
instant-nsr-pl/datasets/blender.py:42
↓ 1 callersFunctiontrans_normal
(normal, RT_w2c_src, RT_w2c_tgt)
ldm/base_utils.py:29
↓ 1 callersFunctiontransform_points_pose
(pts, pose)
ldm/base_utils.py:164
↓ 1 callersMethodupdate_step
(self, epoch, global_step)
instant-nsr-pl/models/base.py:18
↓ 1 callersMethodupdate_step
(self, epoch, global_step)
instant-nsr-pl/models/network_utils.py:32
↓ 1 callersMethodval_dataloader
(self)
instant-nsr-pl/datasets/dtu.py:194
↓ 1 callersFunctionworldNormal2camNormal
(rot_w2c, normal_map_world)
instant-nsr-pl/datasets/blender.py:26
↓ 1 callersFunctionworldNormal2camNormal
(rot_w2c, normal_map_world)
ldm/base_utils.py:13
FunctionIR_101
Constructs a ir-101 model.
ldm/thirdp/psp/model_irse.py:59
FunctionIR_152
Constructs a ir-152 model.
ldm/thirdp/psp/model_irse.py:65
FunctionIR_50
Constructs a ir-50 model.
ldm/thirdp/psp/model_irse.py:53
FunctionIR_SE_101
Constructs a ir_se-101 model.
ldm/thirdp/psp/model_irse.py:77
FunctionIR_SE_152
Constructs a ir_se-152 model.
ldm/thirdp/psp/model_irse.py:83
FunctionIR_SE_50
Constructs a ir_se-50 model.
ldm/thirdp/psp/model_irse.py:71
Method__call__
(self, output, target, reduction='mean')
instant-nsr-pl/systems/criterions.py:101
Method__call__
(self, n, **kwargs)
ldm/lr_scheduler.py:32
Method__call__
(self, n, **kwargs)
ldm/lr_scheduler.py:77
Method__getitem__
(self, index)
instant-nsr-pl/datasets/dtu.py:147
Method__getitem__
(self, index)
instant-nsr-pl/datasets/blender.py:172
Method__getitem__
(self, index)
instant-nsr-pl/datasets/colmap.py:278
Method__getitem__
(self, index)
ldm/data/teeth_dreamer.py:152
Method__getitem__
(self, index)
ldm/data/teeth_dreamer.py:215
Method__getitem__
(self, index)
ldm/data/teeth_dreamer.py:273
Method__init__
(self, resume, logdir, ckptdir, cfgdir, config)
TeethDreamer.py:56
Method__init__
(self, log_keys=[])
instant-nsr-pl/utils/loggers.py:10
Method__init__
(self, save_root, version=None, use_version=True)
instant-nsr-pl/utils/callbacks.py:53
Method__init__
(self, config, save_root, version=None, use_version=True)
instant-nsr-pl/utils/callbacks.py:80
Method__init__
(self, optimizer, schedulers)
instant-nsr-pl/systems/utils.py:36
Method__init__
(self, optimizer, schedulers, milestones, last_epoch=-1, verbose=False)
instant-nsr-pl/systems/utils.py:107
Method__init__
(self, optimizer, factor=1.0 / 3, total_iters=5, last_epoch=-1, verbose=False)
instant-nsr-pl/systems/utils.py:194
Method__init__
(self, config)
instant-nsr-pl/systems/base.py:15
Method__init__
(self)
instant-nsr-pl/systems/criterions.py:41
Method__init__
(self, data_range=(0, 1), kernel_size=(11, 11), sigma=(1.5, 1.5), k1=0.01, k2=0.03, gaussian=True)
instant-nsr-pl/systems/criterions.py:56
Method__init__
(self, config, split)
instant-nsr-pl/datasets/dtu.py:154
Method__init__
(self, config)
instant-nsr-pl/datasets/dtu.py:164
Method__init__
(self, config, split)
instant-nsr-pl/datasets/blender.py:179
Method__init__
(self, config)
instant-nsr-pl/datasets/blender.py:189
Method__init__
(self, config, split)
instant-nsr-pl/datasets/colmap.py:285
Method__init__
(self, config)
instant-nsr-pl/datasets/colmap.py:295
Method__init__
(self, config)
instant-nsr-pl/models/base.py:7
Method__init__
(self, config)
instant-nsr-pl/models/geometry.py:70
Method__init__
(self, config)
instant-nsr-pl/models/texture.py:12
Method__init__
(self, config)
instant-nsr-pl/models/neus.py:16
Method__init__
(self, in_channels, config)
instant-nsr-pl/models/network_utils.py:15
Method__init__
(self, in_channels, config)
instant-nsr-pl/models/network_utils.py:41
Method__init__
(self, encoding, include_xyz=False, xyz_scale=1., xyz_offset=0.)
instant-nsr-pl/models/network_utils.py:69
Method__init__
(self, encoding, network)
instant-nsr-pl/models/network_utils.py:188
Method__init__
(self, warm_up_steps, lr_min, lr_max, lr_start, max_decay_steps, verbosity_interval=0)
ldm/lr_scheduler.py:8
Method__init__
(self, warm_up_steps, f_min, f_max, f_start, cycle_lengths, verbosity_interval=0)
ldm/lr_scheduler.py:41
Method__init__
AdamW that saves EMA versions of the parameters.
ldm/util.py:177
Method__init__
(self, dim, max_seq_len)
ldm/modules/x_transformer.py:26
Method__init__
(self, dim)
ldm/modules/x_transformer.py:40
Method__init__
(self, fn)
ldm/modules/x_transformer.py:129
Method__init__
(self, dim, eps=1e-5)
ldm/modules/x_transformer.py:140
Method__init__
(self, dim, eps=1e-8)
ldm/modules/x_transformer.py:152
Method__init__
(self, dim)
ldm/modules/x_transformer.py:169
Method__init__
(self, dim_in, dim_out)
ldm/modules/x_transformer.py:185
Method__init__
(self, dim, dim_out=None, mult=4, glu=False, dropout=0.)
ldm/modules/x_transformer.py:195
Method__init__
( self, dim, dim_head=DEFAULT_DIM_HEAD, heads=8, c
ldm/modules/x_transformer.py:216
Method__init__
( self, dim, depth, heads=8, causal=False,
ldm/modules/x_transformer.py:371
Method__init__
(self, **kwargs)
ldm/modules/x_transformer.py:542
Method__init__
( self, *, num_tokens, max_seq_len, attn_layers,
ldm/modules/x_transformer.py:549
Method__init__
(self, dim_in, dim_out)
ldm/modules/attention.py:47
Method__init__
(self, dim, dim_out=None, mult=4, glu=False, dropout=0.)
ldm/modules/attention.py:57
Method__init__
(self, dim, heads=4, dim_head=32)
ldm/modules/attention.py:90
Method__init__
(self, in_channels)
ldm/modules/attention.py:109
← previousnext →301–400 of 848, ranked by callers