Summary: Adds the readme file.

Reviewed By: nikhilaravi

Differential Revision: D25684459

fbshipit-source-id: f1aaa621a2a67c98d5fcfe33fe9bbfea8f95b537
This commit is contained in:
David Novotny
2021-02-02 05:42:59 -08:00
committed by Facebook GitHub Bot
parent 2628fb56f2
commit 51de308b80
7 changed files with 87 additions and 5 deletions

View File

@@ -1,6 +1,6 @@
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
import os
from typing import Tuple, Optional, List
from typing import List, Optional, Tuple
import numpy as np
import requests
@@ -9,6 +9,7 @@ from PIL import Image
from pytorch3d.renderer import PerspectiveCameras
from torch.utils.data import Dataset
DEFAULT_DATA_ROOT = os.path.join(
os.path.dirname(os.path.realpath(__file__)), "..", "data"
)

View File

@@ -3,7 +3,7 @@ import math
from typing import Tuple
import torch
from pytorch3d.renderer import look_at_view_transform, PerspectiveCameras
from pytorch3d.renderer import PerspectiveCameras, look_at_view_transform
def generate_eval_video_cameras(

View File

@@ -1,5 +1,5 @@
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
from typing import Tuple, List, Optional
from typing import List, Optional, Tuple
import torch
from pytorch3d.renderer import ImplicitRenderer, ray_bundle_to_ray_points
@@ -11,7 +11,7 @@ from visdom import Visdom
from .implicit_function import NeuralRadianceField
from .raymarcher import EmissionAbsorptionNeRFRaymarcher
from .raysampler import NeRFRaysampler, ProbabilisticRaysampler
from .utils import sample_images_at_mc_locs, calc_psnr, calc_mse
from .utils import calc_mse, calc_psnr, sample_images_at_mc_locs
class RadianceFieldRenderer(torch.nn.Module):

View File

@@ -3,7 +3,7 @@ import math
from typing import List
import torch
from pytorch3d.renderer import RayBundle, NDCGridRaysampler, MonteCarloRaysampler
from pytorch3d.renderer import MonteCarloRaysampler, NDCGridRaysampler, RayBundle
from pytorch3d.renderer.cameras import CamerasBase
from .utils import sample_pdf