fix a bug for rendering only

weiyithu · Dec 9, 2021 · 2851119 · 2851119
1 parent 8c8f962
commit 2851119
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/run_nerf.py b/src/run_nerf.py
@@ -494,15 +494,23 @@ def train(args):
                 testsavedir = os.path.join(save_path, 'results', 
                                            'renderonly_{}_{:06d}'.format('test', start))
                 render_poses = poses_tensor[i_test]
+                depth_priors = depth_priors[i_test]
+                depth_confidences = depth_confidences[i_test]
+                image_list = image_list[i_test]
             else:
                 testsavedir = os.path.join(save_path, 'results', 
                                            'renderonly_{}_{:06d}'.format('train', start))
                 render_poses = poses_tensor[i_train]
+                depth_priors = depth_priors[i_train]
+                depth_confidences = depth_confidences[i_train]
+                image_list = image_list[i_train]
 
             os.makedirs(testsavedir, exist_ok=True)
             rgbs, disps, depths = render_path(render_poses, hwf, args.chunk, render_kwargs_test, sc=sc,
-                                              savedir=testsavedir, render_factor=args.render_factor, 
-                                              image_list=image_list)
+                                          depth_priors=torch.from_numpy(depth_priors).to(device),
+                                          depth_confidences=torch.from_numpy(depth_confidences).to(device), 
+                                          savedir=testsavedir, render_factor=args.render_factor, 
+                                          image_list=image_list)
             print('Done rendering', testsavedir)
 
             return