From a1406c98d8724fa81cbe2bc7c310f6bcc104aa85 Mon Sep 17 00:00:00 2001 From: Krishna Murthy Date: Wed, 15 Apr 2020 07:07:59 -0400 Subject: [PATCH] Replace eval_nerf call Signed-off-by: Krishna Murthy --- eval_nerf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/eval_nerf.py b/eval_nerf.py index 2024163..29fb27d 100644 --- a/eval_nerf.py +++ b/eval_nerf.py @@ -14,7 +14,7 @@ from load_blender import load_blender_data from load_llff import load_llff_data from nerf_helpers import get_ray_bundle, positional_encoding -from train_utils import eval_nerf +from train_utils import run_one_iter_of_nerf def cast_to_image(tensor): @@ -140,7 +140,7 @@ def encode_direction_fn(x): rgb_coarse = None with torch.no_grad(): ray_origins, ray_directions = get_ray_bundle(hwf[0], hwf[1], hwf[2], pose) - rgb_coarse, _, _, rgb_fine, _, _ = eval_nerf( + rgb_coarse, _, _, rgb_fine, _, _ = run_one_iter_of_nerf( hwf[0], hwf[1], hwf[2],