SYMBOL INDEX (171 symbols across 17 files)

FILE: dataset.py
  class MultiResolutionDataset (line 12) | class MultiResolutionDataset(Dataset):
    method __init__ (line 13) | def __init__(self, path, transform, resolution=256, nerf_resolution=64):
    method __len__ (line 33) | def __len__(self):
    method __getitem__ (line 36) | def __getitem__(self, index):

FILE: distributed.py
  function get_rank (line 9) | def get_rank():
  function synchronize (line 19) | def synchronize():
  function get_world_size (line 34) | def get_world_size():
  function reduce_sum (line 44) | def reduce_sum(tensor):
  function gather_grad (line 57) | def gather_grad(params):
  function all_gather (line 69) | def all_gather(data):
  function reduce_loss_dict (line 104) | def reduce_loss_dict(loss_dict):

FILE: download_models.py
  function download_pretrained_models (line 28) | def download_pretrained_models():
  function download_file (line 80) | def download_file(session, file_spec, use_alt_url=False, chunk_size=128,...

FILE: generate_shapes_and_images.py
  function generate (line 27) | def generate(opt, g_ema, surface_g_ema, device, mean_latent, surface_mea...

FILE: losses.py
  function viewpoints_loss (line 7) | def viewpoints_loss(viewpoint_pred, viewpoint_target):
  function eikonal_loss (line 13) | def eikonal_loss(eikonal_term, sdf=None, beta=100):
  function d_logistic_loss (line 27) | def d_logistic_loss(real_pred, fake_pred):
  function d_r1_loss (line 34) | def d_r1_loss(real_pred, real_img):
  function g_nonsaturating_loss (line 43) | def g_nonsaturating_loss(fake_pred):
  function g_path_regularize (line 49) | def g_path_regularize(fake_img, latents, mean_path_length, decay=0.01):

FILE: model.py
  class PixelNorm (line 24) | class PixelNorm(nn.Module):
    method __init__ (line 25) | def __init__(self):
    method forward (line 28) | def forward(self, input):
  class MappingLinear (line 32) | class MappingLinear(nn.Module):
    method __init__ (line 33) | def __init__(self, in_dim, out_dim, bias=True, activation=None, is_las...
    method forward (line 49) | def forward(self, input):
    method __repr__ (line 58) | def __repr__(self):
  function make_kernel (line 64) | def make_kernel(k):
  class Upsample (line 75) | class Upsample(nn.Module):
    method __init__ (line 76) | def __init__(self, kernel, factor=2):
    method forward (line 90) | def forward(self, input):
  class Downsample (line 96) | class Downsample(nn.Module):
    method __init__ (line 97) | def __init__(self, kernel, factor=2):
    method forward (line 111) | def forward(self, input):
  class Blur (line 117) | class Blur(nn.Module):
    method __init__ (line 118) | def __init__(self, kernel, pad, upsample_factor=1):
    method forward (line 130) | def forward(self, input):
  class EqualConv2d (line 136) | class EqualConv2d(nn.Module):
    method __init__ (line 137) | def __init__(
    method forward (line 156) | def forward(self, input):
    method __repr__ (line 167) | def __repr__(self):
  class EqualLinear (line 174) | class EqualLinear(nn.Module):
    method __init__ (line 175) | def __init__(self, in_dim, out_dim, bias=True, bias_init=0, lr_mul=1,
    method forward (line 192) | def forward(self, input):
    method __repr__ (line 203) | def __repr__(self):
  class ModulatedConv2d (line 209) | class ModulatedConv2d(nn.Module):
    method __init__ (line 210) | def __init__(self, in_channel, out_channel, kernel_size, style_dim, de...
    method __repr__ (line 249) | def __repr__(self):
    method forward (line 255) | def forward(self, input, style):
  class NoiseInjection (line 299) | class NoiseInjection(nn.Module):
    method __init__ (line 300) | def __init__(self, project=False):
    method create_pytorch_mesh (line 308) | def create_pytorch_mesh(self, trimesh):
    method load_mc_mesh (line 323) | def load_mc_mesh(self, filename, resolution=128, im_res=64):
    method project_noise (line 349) | def project_noise(self, noise, transform, mesh_path=None):
    method forward (line 380) | def forward(self, image, noise=None, transform=None, mesh_path=None):
  class StyledConv (line 390) | class StyledConv(nn.Module):
    method __init__ (line 391) | def __init__(self, in_channel, out_channel, kernel_size, style_dim,
    method forward (line 408) | def forward(self, input, style, noise=None, transform=None, mesh_path=...
  class ToRGB (line 416) | class ToRGB(nn.Module):
    method __init__ (line 417) | def __init__(self, in_channel, style_dim, upsample=True, blur_kernel=[...
    method forward (line 428) | def forward(self, input, style, skip=None):
  class ConvLayer (line 441) | class ConvLayer(nn.Sequential):
    method __init__ (line 442) | def __init__(self, in_channel, out_channel, kernel_size, downsample=Fa...
  class Decoder (line 478) | class Decoder(nn.Module):
    method __init__ (line 479) | def __init__(self, model_opt, blur_kernel=[1, 3, 3, 1]):
    method mean_latent (line 559) | def mean_latent(self, renderer_latent):
    method get_latent (line 564) | def get_latent(self, input):
    method styles_and_noise_forward (line 567) | def styles_and_noise_forward(self, styles, noise, inject_index=None, t...
    method forward (line 610) | def forward(self, features, styles, rgbd_in=None, transform=None,
  class Generator (line 641) | class Generator(nn.Module):
    method __init__ (line 642) | def __init__(self, model_opt, renderer_opt, blur_kernel=[1, 3, 3, 1], ...
    method make_noise (line 673) | def make_noise(self):
    method mean_latent (line 684) | def mean_latent(self, n_latent, device):
    method get_latent (line 695) | def get_latent(self, input):
    method styles_and_noise_forward (line 698) | def styles_and_noise_forward(self, styles, inject_index=None, truncati...
    method init_forward (line 715) | def init_forward(self, styles, cam_poses, focals, near=0.88, far=1.12):
    method forward (line 722) | def forward(self, styles, cam_poses, focals, near=0.88, far=1.12, retu...
  class VolumeRenderDiscConv2d (line 763) | class VolumeRenderDiscConv2d(nn.Module):
    method __init__ (line 764) | def __init__(self, in_channels, out_channels, kernel_size, stride=1,
    method forward (line 778) | def forward(self, input):
  class AddCoords (line 791) | class AddCoords(nn.Module):
    method __init__ (line 792) | def __init__(self):
    method forward (line 795) | def forward(self, input_tensor):
  class CoordConv2d (line 817) | class CoordConv2d(nn.Module):
    method __init__ (line 818) | def __init__(self, in_channels, out_channels, kernel_size, stride=1,
    method forward (line 826) | def forward(self, input_tensor):
  class CoordConvLayer (line 838) | class CoordConvLayer(nn.Module):
    method __init__ (line 839) | def __init__(self, in_channel, out_channel, kernel_size, bias=True, ac...
    method forward (line 856) | def forward(self, input):
  class VolumeRenderResBlock (line 864) | class VolumeRenderResBlock(nn.Module):
    method __init__ (line 865) | def __init__(self, in_channel, out_channel):
    method forward (line 877) | def forward(self, input):
  class VolumeRenderDiscriminator (line 893) | class VolumeRenderDiscriminator(nn.Module):
    method __init__ (line 894) | def __init__(self, opt):
    method forward (line 926) | def forward(self, input):
  class ResBlock (line 940) | class ResBlock(nn.Module):
    method __init__ (line 941) | def __init__(self, in_channel, out_channel, blur_kernel=[1, 3, 3, 1], ...
    method forward (line 949) | def forward(self, input):
  class Discriminator (line 957) | class Discriminator(nn.Module):
    method __init__ (line 958) | def __init__(self, opt, blur_kernel=[1, 3, 3, 1]):
    method forward (line 1001) | def forward(self, input):

FILE: op/fused_act.py
  class FusedLeakyReLUFunctionBackward (line 20) | class FusedLeakyReLUFunctionBackward(Function):
    method forward (line 22) | def forward(ctx, grad_output, out, bias, negative_slope, scale):
    method backward (line 47) | def backward(ctx, gradgrad_input, gradgrad_bias):
  class FusedLeakyReLUFunction (line 56) | class FusedLeakyReLUFunction(Function):
    method forward (line 58) | def forward(ctx, input, bias, negative_slope, scale):
    method backward (line 74) | def backward(ctx, grad_output):
  class FusedLeakyReLU (line 87) | class FusedLeakyReLU(nn.Module):
    method __init__ (line 88) | def __init__(self, channel, bias=True, negative_slope=0.2, scale=2 ** ...
    method forward (line 100) | def forward(self, input):
  function fused_leaky_relu (line 104) | def fused_leaky_relu(input, bias=None, negative_slope=0.2, scale=2 ** 0.5):

FILE: op/fused_bias_act.cpp
  function fused_bias_act (line 11) | torch::Tensor fused_bias_act(const torch::Tensor& input, const torch::Te...
  function PYBIND11_MODULE (line 19) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: op/upfirdn2d.cpp
  function upfirdn2d (line 12) | torch::Tensor upfirdn2d(const torch::Tensor& input, const torch::Tensor&...
  function PYBIND11_MODULE (line 21) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: op/upfirdn2d.py
  class UpFirDn2dBackward (line 20) | class UpFirDn2dBackward(Function):
    method forward (line 22) | def forward(
    method backward (line 64) | def backward(ctx, gradgrad_input):
  class UpFirDn2d (line 89) | class UpFirDn2d(Function):
    method forward (line 91) | def forward(ctx, input, kernel, up, down, pad):
    method backward (line 128) | def backward(ctx, grad_output):
  function upfirdn2d (line 146) | def upfirdn2d(input, kernel, up=1, down=1, pad=(0, 0)):
  function upfirdn2d_native (line 160) | def upfirdn2d_native(

FILE: options.py
  class BaseOptions (line 5) | class BaseOptions():
    method __init__ (line 6) | def __init__(self):
    method initialize (line 10) | def initialize(self):
    method parse (line 94) | def parse(self):

FILE: prepare_data.py
  function resize_and_convert (line 14) | def resize_and_convert(img, size, resample):
  function resize_multiple (line 24) | def resize_multiple(
  function resize_worker (line 34) | def resize_worker(img_file, sizes, resample):
  function prepare (line 43) | def prepare(

FILE: render_video.py
  function render_video (line 26) | def render_video(opt, g_ema, surface_g_ema, device, mean_latent, surface...

FILE: train_full_pipeline.py
  function train (line 28) | def train(opt, experiment_opt, loader, generator, discriminator, g_optim...

FILE: train_volume_renderer.py
  function train (line 28) | def train(opt, experiment_opt, loader, generator, discriminator, g_optim...

FILE: utils.py
  function data_sampler (line 27) | def data_sampler(dataset, shuffle, distributed):
  function sample_data (line 38) | def sample_data(loader):
  function requires_grad (line 45) | def requires_grad(model, flag=True):
  function accumulate (line 50) | def accumulate(model1, model2, decay=0.999):
  function make_noise (line 59) | def make_noise(batch, latent_dim, n_noise, device):
  function mixing_noise (line 68) | def mixing_noise(batch, latent_dim, prob, device):
  function generate_camera_params (line 75) | def generate_camera_params(resolution, device, batch=1, locations=None, ...
  function align_volume (line 141) | def align_volume(volume, near=0.88, far=1.12):
  function extract_mesh_with_marching_cubes (line 164) | def extract_mesh_with_marching_cubes(sdf):
  function xyz2mesh (line 183) | def xyz2mesh(xyz):
  function add_textures (line 201) | def add_textures(meshes:Meshes, vertex_colors=None) -> Meshes:
  function create_cameras (line 215) | def create_cameras(
  function create_mesh_renderer (line 229) | def create_mesh_renderer(
  class MeshRendererWithDepth (line 263) | class MeshRendererWithDepth(nn.Module):
    method __init__ (line 264) | def __init__(self, rasterizer, shader):
    method forward (line 269) | def forward(self, meshes_world, **kwargs) -> torch.Tensor:
  function create_depth_mesh_renderer (line 275) | def create_depth_mesh_renderer(

FILE: volume_renderer.py
  class LinearLayer (line 12) | class LinearLayer(nn.Module):
    method __init__ (line 13) | def __init__(self, in_dim, out_dim, bias=True, bias_init=0, std_init=1...
    method forward (line 27) | def forward(self, input):
  class FiLMSiren (line 33) | class FiLMSiren(nn.Module):
    method __init__ (line 34) | def __init__(self, in_channel, out_channel, style_dim, is_first=False):
    method forward (line 50) | def forward(self, input, style):
  class SirenGenerator (line 62) | class SirenGenerator(nn.Module):
    method __init__ (line 63) | def __init__(self, D=8, W=256, style_dim=256, input_ch=3, input_ch_vie...
    method forward (line 82) | def forward(self, x, styles):
  class VolumeFeatureRenderer (line 103) | class VolumeFeatureRenderer(nn.Module):
    method __init__ (line 104) | def __init__(self, opt, style_dim=256, out_im_res=64, mode='train'):
    method get_rays (line 158) | def get_rays(self, focal, c2w):
    method get_eikonal_term (line 175) | def get_eikonal_term(self, pts, sdf):
    method sdf_activation (line 182) | def sdf_activation(self, input):
    method volume_integration (line 187) | def volume_integration(self, raw, z_vals, rays_d, pts, return_eikonal=...
    method run_network (line 254) | def run_network(self, inputs, viewdirs, styles=None):
    method render_rays (line 261) | def render_rays(self, ray_batch, styles=None, return_eikonal=False):
    method render (line 305) | def render(self, focal, c2w, near, far, styles, c2w_staticcam=None, re...
    method mlp_init_pass (line 319) | def mlp_init_pass(self, cam_poses, focal, near, far, styles=None):
    method forward (line 348) | def forward(self, cam_poses, focal, near, far, styles=None, return_eik...