SYMBOL INDEX (233 symbols across 14 files)

FILE: cam_utils.py
  function dot (line 6) | def dot(x, y):
  function length (line 13) | def length(x, eps=1e-20):
  function safe_normalize (line 20) | def safe_normalize(x, eps=1e-20):
  function look_at (line 24) | def look_at(campos, target, opengl=True):
  function orbit_camera (line 45) | def orbit_camera(elevation, azimuth, radius=1, is_degree=True, target=No...
  class OrbitCamera (line 65) | class OrbitCamera:
    method __init__ (line 66) | def __init__(self, W, H, r=2, fovy=60, near=0.01, far=100):
    method fovx (line 78) | def fovx(self):
    method campos (line 82) | def campos(self):
    method pose (line 87) | def pose(self):
    method view (line 101) | def view(self):
    method perspective (line 106) | def perspective(self):
    method intrinsics (line 126) | def intrinsics(self):
    method mvp (line 131) | def mvp(self):
    method orbit (line 134) | def orbit(self, dx, dy):
    method scale (line 141) | def scale(self, delta):
    method pan (line 144) | def pan(self, dx, dy, dz=0):

FILE: dataset_4d.py
  class SparseDataset (line 16) | class SparseDataset:
    method __init__ (line 17) | def __init__(self, opt, size,device='cuda', type='train', H=256, W=256):
    method collate_ref (line 32) | def collate_ref(self,index):
    method collate_zero123 (line 55) | def collate_zero123(self,index):
    method collate (line 85) | def collate(self, index):
    method dataloader (line 97) | def dataloader(self):
    method dataloader_d (line 101) | def dataloader_d(self):

FILE: deform.py
  class Deformation (line 23) | class Deformation(nn.Module):
    method __init__ (line 24) | def __init__(self, D=8, W=256, input_ch=27, input_ch_time=9, skips=[],...
    method create_net (line 47) | def create_net(self):
    method query_time (line 66) | def query_time(self, rays_pts_emb, scales_emb, rotations_emb, time_emb):
    method forward (line 79) | def forward(self, rays_pts_emb, scales_emb=None, rotations_emb=None, o...
    method forward_static (line 85) | def forward_static(self, rays_pts_emb):
    method forward_dynamic (line 89) | def forward_dynamic(self,rays_pts_emb, scales_emb, rotations_emb, opac...
    method get_mlp_parameters (line 112) | def get_mlp_parameters(self):
    method get_grid_parameters (line 118) | def get_grid_parameters(self):
  class deform_network (line 121) | class deform_network(nn.Module):
    method __init__ (line 122) | def __init__(self) :
    method forward (line 144) | def forward(self, point, scales=None, rotations=None, opacity=None, ti...
    method forward_static (line 151) | def forward_static(self, points):
    method forward_dynamic (line 154) | def forward_dynamic(self, point, scales=None, rotations=None, opacity=...
    method get_mlp_parameters (line 164) | def get_mlp_parameters(self):
    method get_grid_parameters (line 166) | def get_grid_parameters(self):
  function initialize_weights (line 169) | def initialize_weights(m):
  function get_normalized_directions (line 179) | def get_normalized_directions(directions):
  function normalize_aabb (line 188) | def normalize_aabb(pts, aabb):
  function grid_sample_wrapper (line 190) | def grid_sample_wrapper(grid: torch.Tensor, coords: torch.Tensor, align_...
  function init_grid_param (line 217) | def init_grid_param(
  function interpolate_ms_features (line 242) | def interpolate_ms_features(pts: torch.Tensor,
  class HexPlaneField (line 278) | class HexPlaneField(nn.Module):
    method __init__ (line 279) | def __init__(
    method set_aabb (line 320) | def set_aabb(self,xyz_max, xyz_min):
    method get_density (line 328) | def get_density(self, pts: torch.Tensor, timestamps: Optional[torch.Te...
    method forward (line 345) | def forward(self,
  function compute_plane_tv (line 353) | def compute_plane_tv(t):
  function compute_plane_smoothness (line 362) | def compute_plane_smoothness(t):
  class Regularizer (line 371) | class Regularizer():
    method __init__ (line 372) | def __init__(self, reg_type, initialization):
    method step (line 378) | def step(self, global_step):
    method report (line 381) | def report(self, d):
    method regularize (line 385) | def regularize(self, *args, **kwargs) -> torch.Tensor:
    method _regularize (line 391) | def _regularize(self, *args, **kwargs) -> torch.Tensor:
    method __str__ (line 394) | def __str__(self):
  class PlaneTV (line 398) | class PlaneTV(Regularizer):
    method __init__ (line 399) | def __init__(self, initial_value, what: str = 'field'):
    method step (line 407) | def step(self, global_step):
    method _regularize (line 410) | def _regularize(self, model, **kwargs):
  class TimeSmoothness (line 433) | class TimeSmoothness(Regularizer):
    method __init__ (line 434) | def __init__(self, initial_value, what: str = 'field'):
    method _regularize (line 442) | def _regularize(self, model, **kwargs) -> torch.Tensor:
  class L1ProposalNetwork (line 463) | class L1ProposalNetwork(Regularizer):
    method __init__ (line 464) | def __init__(self, initial_value):
    method _regularize (line 467) | def _regularize(self, model, **kwargs) -> torch.Tensor:
  class DepthTV (line 476) | class DepthTV(Regularizer):
    method __init__ (line 477) | def __init__(self, initial_value):
    method _regularize (line 480) | def _regularize(self, model, model_out, **kwargs) -> torch.Tensor:
  class L1TimePlanes (line 488) | class L1TimePlanes(Regularizer):
    method __init__ (line 489) | def __init__(self, initial_value, what='field'):
    method _regularize (line 496) | def _regularize(self, model, **kwargs) -> torch.Tensor:

FILE: gs_renderer_4d.py
  function inverse_sigmoid (line 19) | def inverse_sigmoid(x):
  function get_expon_lr_func (line 25) | def get_expon_lr_func(
  function strip_lowerdiag (line 50) | def strip_lowerdiag(L):
  function strip_symmetric (line 61) | def strip_symmetric(sym):
  function gaussian_3d_coeff (line 64) | def gaussian_3d_coeff(xyzs, covs):
  function build_rotation (line 85) | def build_rotation(r):
  function build_scaling_rotation (line 108) | def build_scaling_rotation(s, r):
  class BasicPointCloud (line 119) | class BasicPointCloud(NamedTuple):
  class GaussianModel (line 125) | class GaussianModel:
    method setup_functions (line 127) | def setup_functions(self):
    method initialize (line 144) | def initialize(self, initial_values, raw=False):
    method __init__ (line 155) | def __init__(self, sh_degree : int,args = None):
    method capture (line 174) | def capture(self):
    method restore (line 192) | def restore(self, model_args, training_args):
    method get_scaling (line 213) | def get_scaling(self):
    method get_rotation (line 217) | def get_rotation(self):
    method get_xyz (line 221) | def get_xyz(self):
    method get_features (line 225) | def get_features(self):
    method get_opacity (line 231) | def get_opacity(self):
    method get_covariance (line 238) | def get_covariance(self, scaling_modifier = 1):
    method oneupSHdegree (line 241) | def oneupSHdegree(self):
    method create_from_pcd (line 245) | def create_from_pcd(self, pcd : BasicPointCloud, spatial_lr_scale : fl...
    method training_setup (line 276) | def training_setup(self, training_args):
    method update_learning_rate (line 308) | def update_learning_rate(self, iteration):
    method construct_list_of_attributes (line 324) | def construct_list_of_attributes(self):
    method save_ply (line 338) | def save_ply(self, path):
    method compute_deformation (line 357) | def compute_deformation(self,time):
    method load_model (line 362) | def load_model(self, path):
    method save_deformation (line 378) | def save_deformation(self, path):
    method load_ply (line 383) | def load_ply(self, path):
    method replace_tensor_to_optimizer (line 430) | def replace_tensor_to_optimizer(self, tensor, name):
    method _prune_optimizer (line 445) | def _prune_optimizer(self, mask):
    method prune_points (line 465) | def prune_points(self, mask):
    method cat_tensors_to_optimizer (line 481) | def cat_tensors_to_optimizer(self, tensors_dict):
    method densification_postfix (line 504) | def densification_postfix(self, new_xyz, new_features_dc, new_features...
    method densify_and_split (line 529) | def densify_and_split(self, grads, grad_threshold, scene_extent, N=2):
    method densify_and_clone (line 555) | def densify_and_clone(self, grads, grad_threshold, scene_extent):
    method densify_and_prune (line 571) | def densify_and_prune(self, max_grad_percent, min_opacity, extent, max...
    method prune (line 600) | def prune(self, min_opacity=0.01, extent=1, max_screen_size=1):
    method standard_constaint (line 619) | def standard_constaint(self):
    method add_densification_stats (line 633) | def add_densification_stats(self, viewspace_point_tensor, update_filter):
    method update_deformation_table (line 639) | def update_deformation_table(self,threshold):
    method print_deformation_weight_grad (line 642) | def print_deformation_weight_grad(self):
    method _plane_regulation (line 652) | def _plane_regulation(self):
    method _time_regulation (line 664) | def _time_regulation(self):
    method _l1_regulation (line 676) | def _l1_regulation(self):
    method compute_regulation (line 690) | def compute_regulation(self, time_smoothness_weight, l1_time_planes_we...
  function getProjectionMatrix (line 693) | def getProjectionMatrix(znear, zfar, fovX, fovY):
  class MiniCam (line 709) | class MiniCam:
    method __init__ (line 710) | def __init__(self, c2w, width, height, fovy, fovx, znear, zfar,time=0 ):
  class Renderer (line 738) | class Renderer:
    method __init__ (line 739) | def __init__(self, sh_degree=3, white_background=True, radius=1):
    method initialize (line 753) | def initialize(self, input=None, num_pts=5000, radius=0.5,initial_valu...
    method render (line 785) | def render(

FILE: guidance/zero123_4d_utils.py
  class Zero123 (line 21) | class Zero123(nn.Module):
    method __init__ (line 22) | def __init__(self, device, fp16=True, t_range=[0.02, 0.98]):
    method get_img_embeds (line 62) | def get_img_embeds(self, x, input_imgs=None):
    method refine (line 84) | def refine(self, pred_rgb, polar, azimuth, radius,
    method train_step (line 129) | def train_step(self, pred_rgb, polar, azimuth, radius, step_ratio=None...
    method update_step (line 186) | def update_step(self, epoch: int, global_step: int, on_load_weights: b...
    method get_steps (line 193) | def get_steps(self,percent,epoch, global_step):
    method decode_latents (line 203) | def decode_latents(self, latents):
    method encode_imgs (line 211) | def encode_imgs(self, imgs, mode=False):

FILE: guidance/zero123pp/pipeline.py
  function to_rgb_image (line 37) | def to_rgb_image(maybe_rgba: Image.Image):
  class MyAttnProcessor2_0 (line 49) | class MyAttnProcessor2_0:
    method __init__ (line 54) | def __init__(self):
    method __call__ (line 58) | def __call__(
  class ReferenceOnlyAttnProc (line 152) | class ReferenceOnlyAttnProc(torch.nn.Module):
    method __init__ (line 153) | def __init__(
    method __call__ (line 164) | def __call__(
  class RefOnlyNoisedUNet (line 191) | class RefOnlyNoisedUNet(torch.nn.Module):
    method __init__ (line 192) | def __init__(self, unet: UNet2DConditionModel, train_sched: DDPMSchedu...
    method __getattr__ (line 212) | def __getattr__(self, name: str):
    method forward_cond (line 218) | def forward_cond(self, noisy_cond_lat, timestep, encoder_hidden_states...
    method forward (line 230) | def forward(
  function scale_latents (line 275) | def scale_latents(latents):
  function unscale_latents (line 280) | def unscale_latents(latents):
  function scale_image (line 285) | def scale_image(image):
  function unscale_image (line 290) | def unscale_image(image):
  class DepthControlUNet (line 295) | class DepthControlUNet(torch.nn.Module):
    method __init__ (line 296) | def __init__(self, unet: RefOnlyNoisedUNet, controlnet: Optional[diffu...
    method __getattr__ (line 309) | def __getattr__(self, name: str):
    method forward (line 315) | def forward(self, sample, timestep, encoder_hidden_states, class_label...
  class ModuleListDict (line 336) | class ModuleListDict(torch.nn.Module):
    method __init__ (line 337) | def __init__(self, procs: dict) -> None:
    method __getitem__ (line 342) | def __getitem__(self, key):
  class SuperNet (line 346) | class SuperNet(torch.nn.Module):
    method __init__ (line 347) | def __init__(self, state_dict: Dict[str, torch.Tensor]):
  class Zero123PlusPipeline (line 386) | class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
    method __init__ (line 405) | def __init__(
    method prepare (line 431) | def prepare(self):
    method add_controlnet (line 436) | def add_controlnet(self, controlnet: Optional[diffusers.ControlNetMode...
    method encode_condition_image (line 441) | def encode_condition_image(self, image: torch.Tensor):
    method __call__ (line 446) | def __call__(

FILE: main.py
  function save_image_to_local (line 19) | def save_image_to_local(image_tensor, file_path):
  class GUI (line 26) | class GUI:
    method __init__ (line 27) | def __init__(self, opt):
    method __del__ (line 110) | def __del__(self):
    method seed_everything (line 114) | def seed_everything(self):
    method prepare_image (line 129) | def prepare_image(self,idx):
    method prepare_train (line 160) | def prepare_train(self):
    method train_step (line 216) | def train_step(self):
    method set_fix_cam (line 414) | def set_fix_cam(self):
    method test_step (line 497) | def test_step(self):
    method load_input (line 567) | def load_input(self, file):
    method save_renderings (line 573) | def save_renderings(self, elev=0, azim=0, radius=2, name='front'):
    method save_model (line 606) | def save_model(self, mode='geo', texture_size=1024):
    method register_dpg (line 622) | def register_dpg(self):
    method render (line 955) | def render(self):
    method train (line 965) | def train(self, iters=500):

FILE: scripts/app.py
  class SAMAPI (line 12) | class SAMAPI:
    method get_instance (line 17) | def get_instance(sam_checkpoint=None):
    method segment_api (line 40) | def segment_api(rgb, mask=None, bbox=None, sam_checkpoint=None):
  function image_examples (line 74) | def image_examples(samples, ncols, return_key=None, example_text="Exampl...
  function segment_img (line 96) | def segment_img(img: Image):
  function segment_6imgs (line 105) | def segment_6imgs(zero123pp_imgs):
  function expand2square (line 128) | def expand2square(pil_img, background_color):
  function check_dependencies (line 143) | def check_dependencies():
  function load_zero123plus_pipeline (line 177) | def load_zero123plus_pipeline():

FILE: scripts/gen_mv.py
  function segment_img (line 17) | def segment_img(img: Image):
  function segment_6imgs (line 26) | def segment_6imgs(zero123pp_imgs):
  function process_img (line 51) | def process_img(path,destination,pipeline, is_first):

FILE: sh_utils.py
  function eval_sh (line 57) | def eval_sh(deg, sh, dirs):
  function RGB2SH (line 114) | def RGB2SH(rgb):
  function SH2RGB (line 117) | def SH2RGB(sh):

FILE: simple-knn/ext.cpp
  function PYBIND11_MODULE (line 15) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: simple-knn/simple_knn.h
  function class (line 15) | class SimpleKNN

FILE: visualize.py
  function save_image_to_local (line 22) | def save_image_to_local(image_tensor, file_path):
  class GUI (line 28) | class GUI:
    method __init__ (line 29) | def __init__(self, opt):
    method __del__ (line 95) | def __del__(self):
    method seed_everything (line 99) | def seed_everything(self):
    method prepare_train (line 120) | def prepare_train(self):
    method train_step (line 164) | def train_step(self):
    method save_renderings (line 206) | def save_renderings(self, elev=0, azim=0, radius=2, name='front', inte...
    method set_fix_cam2 (line 270) | def set_fix_cam2(self):
    method test_step (line 353) | def test_step(self):
    method load_input (line 423) | def load_input(self, file):
    method save_model (line 479) | def save_model(self, mode='geo', texture_size=1024):
    method register_dpg (line 495) | def register_dpg(self):
    method render (line 828) | def render(self):
    method train (line 838) | def train(self, iters=500):

FILE: zero123.py
  class CLIPCameraProjection (line 41) | class CLIPCameraProjection(ModelMixin, ConfigMixin):
    method __init__ (line 53) | def __init__(self, embedding_dim: int = 768, additional_embeddings: in...
    method forward (line 63) | def forward(
  class Zero123Pipeline (line 81) | class Zero123Pipeline(DiffusionPipeline):
    method __init__ (line 109) | def __init__(
    method enable_sequential_cpu_offload (line 180) | def enable_sequential_cpu_offload(self, gpu_id=0):
    method _execution_device (line 204) | def _execution_device(self):
    method _encode_image (line 221) | def _encode_image(
    method run_safety_checker (line 299) | def run_safety_checker(self, image, device, dtype):
    method decode_latents (line 318) | def decode_latents(self, latents):
    method prepare_extra_step_kwargs (line 332) | def prepare_extra_step_kwargs(self, generator, eta):
    method check_inputs (line 353) | def check_inputs(self, image, height, width, callback_steps):
    method prepare_latents (line 371) | def prepare_latents(
    method _get_latent_model_input (line 405) | def _get_latent_model_input(
    method __call__ (line 449) | def __call__(