MCPcopy Create free account

hub / github.com/FrozenBurning/SceneDreamer / functions

Functions542 in github.com/FrozenBurning/SceneDreamer

Functionrand_camera_pose_birdseye
r"""Generating random camera pose in the upper hemisphere, in the format of origin-direction-up Assuming [Y X Z] coordinate. Y is negative gravity
imaginaire/model_utils/gancraft/camctl.py:445
Functionrand_camera_pose_firstperson
r"""Generating random camera pose in the upper hemisphere, in the format of origin-direction-up
imaginaire/model_utils/gancraft/camctl.py:489
Functionrand_camera_pose_insideout
(voxel)
imaginaire/model_utils/gancraft/camctl.py:645
Functionrand_camera_pose_thridperson
(voxel, border=96)
imaginaire/model_utils/gancraft/camctl.py:519
Functionrand_camera_pose_thridperson2
(voxel, border=48)
imaginaire/model_utils/gancraft/camctl.py:542
Functionrand_camera_pose_thridperson3
r"""Attempting to solve the camera too close to wall problem and the lack of aerial poses.
imaginaire/model_utils/gancraft/camctl.py:572
Functionrand_camera_pose_tour
(voxel)
imaginaire/model_utils/gancraft/camctl.py:606
Functionrand_contrast
(x, **kwargs)
imaginaire/utils/diff_aug.py:60
Functionrand_crop
r"""Produces a new cam_c so that the effect of rendering with the new cam_c and target_res is the same as rendering with the old parameters and th
imaginaire/model_utils/gancraft/mc_utils.py:277
Functionrand_cutout
(x, ratio=0.5, **kwargs)
imaginaire/utils/diff_aug.py:88
Functionrand_saturation
(x, **kwargs)
imaginaire/utils/diff_aug.py:53
Functionrand_translation
(x, ratio=0.125, **kwargs)
imaginaire/utils/diff_aug.py:67
Functionrand_translation_scale
(x, trans_r=0.125, scale_r=0.125, mode='bilinear', padding_mode='reflection',
imaginaire/utils/diff_aug.py:111
Functionrandom_shift
r"""Randomly shift the input tensor. Args: x (4D tensor): The input batch of images. offset (int): The maximum offset ratio that
imaginaire/utils/misc.py:216
Methodread_img
(path)
imaginaire/generators/scenedreamer.py:520
Methodrecalculate_batch_norm_statistics
r"""Update the statistics in the moving average model. Args: data_loader (torch.utils.data.DataLoader): Data loader for
imaginaire/trainers/base.py:496
Functionremove_equalized_lr
(module, weight_name='weight', bias_name='bias')
imaginaire/layers/weight_norm.py:210
Functionremove_weight_norms
(module, weight_name='weight', bias_name='bias')
imaginaire/layers/weight_norm.py:187
Functionreset_batch_norm
r"""Reset batch norm statistics Args: m: Pytorch module
imaginaire/utils/model_average.py:13
Functionrsetattr
Recursively find object and set value
imaginaire/config.py:210
Functionsample_depth_batched
r""" Make best effort to sample points within the same distance for every ray. Exception: When there is not enough voxel. Args: de
imaginaire/model_utils/gancraft/mc_utils.py:82
Functionsave_intermediate_training_results
r"""Save intermediate training results for debugging purpose. Args: visualization_images (tensor): Image where pixel values are in [-1, 1
imaginaire/utils/io.py:26
Functionsave_tensor_image
r"""Convert a 3 dimensional torch tensor to a PIL image with the desired width and height. Args: filename (str): Image filename to be
imaginaire/utils/visualization/common.py:14
Functionscaled_lr
( module, weight_name='weight', bias_name='bias', lr_mul=1., equalized=False, )
imaginaire/layers/weight_norm.py:222
Functionsegmask_smooth
(seg_mask, kernel_size=7)
imaginaire/model_utils/gancraft/mc_utils.py:287
Methodsmooth_interp
r"""Smooth interpolation of segmentation maps. Args: x (4D tensor): Segmentation maps. size(2D list): Target size (H,
imaginaire/discriminators/gancraft.py:217
Functionsparse_trilinear_interp_worldcoord
(in_feature, corner_lut_t, in_worldcoord, ign_zero=False)
imaginaire/model_utils/gancraft/voxlib/sp_trilinear.py:34
Methodstep
r"""Performs a single optimization step. Args: closure (callable, optional): A closure that reevaluates the model
imaginaire/optimizers/fromage.py:20
FunctionsystemGetDriverVersion
r"""Get Driver Version
imaginaire/utils/gpu_affinity.py:12
Functiontensor2flow
r"""Convert flow tensor to color image. Args: tensor (tensor) of If tensor then (NxCxHxW) or (NxTxCxHxW) or (CxHxW). imty
imaginaire/utils/visualization/common.py:158
Methodtest
r"""Compute results images for a batch of input data and save the results in the specified folder. Args: data_loader (tor
imaginaire/trainers/base.py:825
Methodtest
r"""Compute results images for a batch of input data and save the results in the specified folder. Args: data_loader (tor
imaginaire/trainers/gancraft.py:309
Functionto_cpu
r"""Move all tensors inside data to cpu. Args: data (dict, list, or tensor): Input data.
imaginaire/utils/misc.py:82
Functionto_half
r"""Move all floats to half. Args: data (dict, list or tensor): Input data.
imaginaire/utils/misc.py:91
Methodtopk_anneal
r"""Anneal k after each epoch.
imaginaire/losses/gan.py:168
Functiontrans_vec_homo
r"""3-dimensional Homogeneous matrix and regular vector multiplication Convert v to homogeneous vector, perform M-V multiplication, and convert ba
imaginaire/model_utils/gancraft/mc_utils.py:54
Functionvolum_rendering_relu
(sigma, dists, dim=2)
imaginaire/model_utils/gancraft/mc_utils.py:154
Functionweight_demod
r"""Weight demodulation.
imaginaire/layers/weight_norm.py:70
Methodworld2local
(self, v, is_vec=False)
imaginaire/model_utils/pcg_gen.py:176
Functionwrapper
r"""Simple function wrapper for the master function
imaginaire/utils/distributed.py:51
Methodwrite_img
(path, img, rgb_input=False)
imaginaire/generators/scenedreamer.py:512
Methodyaml
Convert object to yaml dict and return.
imaginaire/config.py:34
← previous501–542 of 542, ranked by callers