Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ShanghaiTech-IMPACT/TeethDreamer
/ functions
Functions
848 in github.com/ShanghaiTech-IMPACT/TeethDreamer
⨍
Functions
848
◇
Types & classes
168
↳
Endpoints
1
Method
nll
(self, sample, dims=[1,2,3])
ldm/modules/distributions/distributions.py:53
Function
noise_like
(shape, device, repeat=False)
ldm/modules/diffusionmodules/util.py:264
Function
normal_kl
source: https://github.com/openai/guided-diffusion/blob/27c20a8fab9cb472df5d6bdd6c8d11c8f430b924/guided_diffusion/losses.py#L12 Compute the K
ldm/modules/distributions/distributions.py:65
Function
normalize
(path)
instant-nsr-pl/tools.py:25
Function
normalize_scene
()
normal_render.py:304
Function
normalize_scene
()
color_render.py:315
Method
occ_eval_fn
(x)
instant-nsr-pl/models/nerf.py:49
Method
occ_eval_fn
(x)
instant-nsr-pl/models/neus.py:90
Method
occ_eval_fn_bg
(x)
instant-nsr-pl/models/neus.py:103
Function
on_click
(event)
seg_foreground.py:47
Function
on_click
(event)
seg_teeth.py:47
Method
on_fit_start
(self, trainer, pl_module)
TeethDreamer.py:64
Method
on_fit_start
(self, trainer, pl_module)
instant-nsr-pl/utils/callbacks.py:72
Method
on_fit_start
(self, trainer, pl_module)
instant-nsr-pl/utils/callbacks.py:90
Function
on_move
(event)
seg_foreground.py:40
Function
on_move
(event)
seg_teeth.py:40
Method
on_predict_batch_start
(self, batch, batch_idx, dataloader_idx)
instant-nsr-pl/systems/base.py:69
Method
on_test_batch_start
(self, batch, batch_idx, dataloader_idx)
instant-nsr-pl/systems/base.py:64
Method
on_train_batch_end
(self, trainer, pl_module, outputs, batch, batch_idx)
TeethDreamer.py:137
Method
on_train_batch_end
(self, *args, **kwargs)
ldm/models/autoencoder.py:92
Method
on_train_batch_start
(self, batch, batch_idx, unused=0)
instant-nsr-pl/systems/base.py:54
Method
on_train_epoch_end
(self, trainer, pl_module)
TeethDreamer.py:155
Method
on_train_epoch_start
(self, trainer, pl_module)
TeethDreamer.py:149
Method
on_train_start
(self, trainer, pl_module)
TeethDreamer.py:178
Method
on_validation_batch_end
(self, trainer, pl_module, outputs, batch, batch_idx, dataloader_idx=0)
TeethDreamer.py:141
Method
on_validation_batch_start
(self, batch, batch_idx, dataloader_idx)
instant-nsr-pl/systems/base.py:59
Function
output_points
(fn,pts,colors=None)
ldm/base_utils.py:145
Function
pick_and_pop
(keys, d)
ldm/modules/x_transformer.py:88
Function
pil_rectangle_crop
(im)
ldm/util.py:27
Function
pose_apply
(pose,pts)
ldm/base_utils.py:170
Function
pose_inverse
(pose)
ldm/base_utils.py:116
Method
predict_dataloader
(self)
instant-nsr-pl/datasets/blender.py:225
Method
predict_dataloader
(self)
instant-nsr-pl/datasets/colmap.py:331
Method
prepare
(self)
instant-nsr-pl/systems/nerf.py:23
Method
prepare
(self)
instant-nsr-pl/systems/neus.py:36
Method
prepare_data
(self)
instant-nsr-pl/datasets/blender.py:203
Method
prepare_data
(self)
instant-nsr-pl/datasets/colmap.py:309
Method
preprocess_data
(self, batch, stage)
instant-nsr-pl/systems/nerf.py:33
Method
preprocess_data
(self, batch, stage)
instant-nsr-pl/systems/neus.py:47
Function
project_points
(pts,RT,K)
ldm/base_utils.py:121
Function
prompt
(question)
instant-nsr-pl/utils/misc.py:17
Function
read_model
(path, ext)
instant-nsr-pl/datasets/colmap_utils.py:259
Function
read_pickle
(pkl_path)
normal_render.py:37
Function
read_pickle
(pkl_path)
color_render.py:37
Function
reflect
(x, n)
instant-nsr-pl/models/utils.py:104
Function
register
(name)
instant-nsr-pl/systems/__init__.py:4
Function
register
(name)
instant-nsr-pl/datasets/__init__.py:4
Function
register
(name)
instant-nsr-pl/models/__init__.py:4
Method
regularizations
(self, out)
instant-nsr-pl/models/nerf.py:146
Method
regularizations
(self, out)
instant-nsr-pl/models/texture.py:35
Method
regularizations
(self, out)
instant-nsr-pl/models/texture.py:56
Method
regularizations
(self, out)
instant-nsr-pl/models/neus.py:327
Method
rgb_sigma_fn
(t_starts, t_ends, ray_indices)
instant-nsr-pl/models/nerf.py:73
Function
rotate_ray
(xyz, angle, offset)
instant-nsr-pl/models/ray_utils.py:45
Function
rotmat2qvec
(R)
instant-nsr-pl/datasets/colmap_utils.py:284
Method
sample
(self)
ldm/modules/distributions/distributions.py:6
Method
sample
(self)
ldm/modules/distributions/distributions.py:35
Method
sample
@param input_info: x, elevation @param clip_embed: B,M,768 @param unconditional_scale: @param log_every_t:
ldm/models/diffusion/zero123.py:599
Method
sample
@param input_info: x, elevation @param clip_embed: B,M,768 @param unconditional_scale: @param log_every_t:
ldm/models/diffusion/teeth_dreamer.py:737
Function
save_cond_cams
(object_file, output_dir)
normal_render.py:339
Function
save_cond_cams
(object_file, output_dir)
color_render.py:354
Method
save_cubemap
(self, filename, img, data_range=(0, 1))
instant-nsr-pl/utils/mixins.py:156
Method
save_data
(self, filename, data)
instant-nsr-pl/utils/mixins.py:177
Method
save_dir
(self)
instant-nsr-pl/utils/mixins.py:18
Method
save_file
(self, filename, src_path)
instant-nsr-pl/utils/mixins.py:224
Method
save_grayscale_image
(self, filename, img, data_range=DEFAULT_GRAYSCALE_KWARGS['data_range'], cmap=DEFAULT_GRAYSCALE_KWARGS['cmap']
instant-nsr-pl/utils/mixins.py:119
Method
save_image
(self, filename, img)
instant-nsr-pl/utils/mixins.py:147
Method
save_json
(self, filename, payload)
instant-nsr-pl/utils/mixins.py:227
Function
save_pickle
(data, pkl_path)
normal_render.py:41
Function
save_pickle
(data, pkl_path)
ldm/base_utils.py:36
Method
save_rgb_image
(self, filename, img, data_format=DEFAULT_RGB_KWARGS['data_format'], data_range=DEFAULT_RGB_KWARGS['data_range
instant-nsr-pl/utils/mixins.py:55
Method
save_state_dict
(self, filename, data)
instant-nsr-pl/utils/mixins.py:188
Method
save_uv_image
(self, filename, img, data_format=DEFAULT_UV_KWARGS['data_format'], data_range=DEFAULT_UV_KWARGS['data_range']
instant-nsr-pl/utils/mixins.py:82
Method
savedir
(self)
instant-nsr-pl/utils/callbacks.py:46
Function
scale_module
Scale the parameters of a module and return it.
ldm/modules/diffusionmodules/util.py:183
Method
schedule
(self, n, **kwargs)
ldm/lr_scheduler.py:83
Method
setup
(self, config, split)
instant-nsr-pl/datasets/dtu.py:59
Method
setup
(self, config, split)
instant-nsr-pl/datasets/blender.py:58
Method
setup
(self, config, split)
instant-nsr-pl/datasets/colmap.py:137
Method
setup
(self)
instant-nsr-pl/models/geometry.py:117
Method
setup
(self)
instant-nsr-pl/models/geometry.py:195
Method
setup
(self)
instant-nsr-pl/models/nerf.py:16
Method
setup
(self)
instant-nsr-pl/models/neus.py:48
Method
setup
(self, stage)
ldm/data/teeth_dreamer.py:297
Function
show_box
(box, ax)
seg_foreground.py:34
Function
show_box
(box, ax)
seg_teeth.py:34
Function
show_points
(coords, labels, ax, marker_size=175)
seg_foreground.py:25
Function
show_points
(coords, labels, ax, marker_size=175)
seg_teeth.py:25
Method
sigma_fn
(t_starts, t_ends, ray_indices)
instant-nsr-pl/models/nerf.py:65
Method
sigma_fn
(t_starts, t_ends, ray_indices)
instant-nsr-pl/models/neus.py:145
Function
split
(path)
instant-nsr-pl/tools.py:19
Method
state_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:50
Method
step
(self)
instant-nsr-pl/systems/utils.py:46
Method
step
Performs a single optimization step. Args: closure (callable, optional): A closure that reevaluates the model and
ldm/util.py:204
Function
str2bool
(v)
TeethDreamer.py:26
Function
string_begins_with
(prefix, str)
ldm/modules/x_transformer.py:102
Method
test_dataloader
(self)
instant-nsr-pl/datasets/blender.py:222
Method
test_dataloader
(self)
instant-nsr-pl/datasets/colmap.py:328
Method
test_dataloader
(self)
ldm/data/teeth_dreamer.py:313
Method
test_epoch_end
Gather metrics from all devices, compute mean. Purge repeated results using data index.
instant-nsr-pl/systems/base.py:108
← previous
next →
701–800 of 848, ranked by callers