add llff dataset

configs/fern.txt

@@ -1,15 +1,24 @@
-expname = fern_test
-basedir = ./logs
-datadir = ./data/nerf_llff_data/fern
+output_dir = ./output
+obs_imgs_dir = ./data/nerf_llff_data/fern
+ckpt_dir = ./ckpts
+
+ckpt_name = fern
+model_name = fern
 dataset_type = llff
 
 factor = 8
 llffhold = 8
-
-N_rand = 1024
+use_viewdirs = True
 N_samples = 64
 N_importance = 64
-
-use_viewdirs = True
 raw_noise_std = 1e0
 
+obs_img_num = 18
+dil_iter = 1
+kernel_size = 3
+start_pose_num = 19
+batch_size = 2048
+lrate = 0.01
+
+
+

configs/flower.txt

@@ -1,15 +1,24 @@
-expname = flower_test
-basedir = ./logs
-datadir = ./data/nerf_llff_data/flower
+output_dir = ./output
+obs_imgs_dir = ./data/nerf_llff_data/flower
+ckpt_dir = ./ckpts
+
+ckpt_name = flower
+model_name = flower
 dataset_type = llff
 
 factor = 8
 llffhold = 8
-
-N_rand = 1024
+use_viewdirs = True
 N_samples = 64
 N_importance = 64
-
-use_viewdirs = True
 raw_noise_std = 1e0
 
+obs_img_num = 18
+dil_iter = 1
+kernel_size = 3
+start_pose_num = 19
+batch_size = 2048
+lrate = 0.01
+
+
+

configs/hotdog.txt

@@ -1,15 +1,20 @@
-expname = hotdog_test
-basedir = ./logs
-datadir = ./data/nerf_llff_data/hotdog
-dataset_type = llff
-
-factor = 8
-llffhold = 8
-
-N_rand = 1024
-N_samples = 64
-N_importance = 64
+model_name = hotdog
+output_dir = ./output
+data_dir = ./data/nerf_synthetic/hotdog
+dataset_type = blender
+obs_imgs_dir = ./data/nerf_synthetic/hotdog/obs_imgs
+ckpt_dir = ./ckpts
+ckpt_name = hotdog
 
 use_viewdirs = True
-raw_noise_std = 1e0
+white_bkgd = True
+N_samples = 50
+N_importance = 80
+half_res = True
 
+obs_img_num = 15
+dil_iter = 1
+kernel_size = 3
+start_pose_num = 27
+batch_size = 2050
+lrate = 0.01

configs/lego.txt

@@ -8,13 +8,13 @@ ckpt_name = lego
 
 use_viewdirs = True
 white_bkgd = True
-N_samples = 32
-N_importance = 64
+N_samples = 50
+N_importance = 80
 half_res = True
 
-obs_img_num = 30
+obs_img_num = 15
 dil_iter = 1
 kernel_size = 3
-start_pose_num = 48
-batch_size = 500
+start_pose_num = 27
+batch_size = 2050
 lrate = 0.01

configs/orchids.txt

@@ -1,15 +1,24 @@
-expname = orchids_test
-basedir = ./logs
-datadir = ./data/nerf_llff_data/orchids
+output_dir = ./output
+obs_imgs_dir = ./data/nerf_llff_data/orchids
+ckpt_dir = ./ckpts
+
+ckpt_name = orchids
+model_name = orchids
 dataset_type = llff
 
 factor = 8
 llffhold = 8
-
-N_rand = 1024
+use_viewdirs = True
 N_samples = 64
 N_importance = 64
-
-use_viewdirs = True
 raw_noise_std = 1e0
 
+obs_img_num = 18
+dil_iter = 1
+kernel_size = 3
+start_pose_num = 19
+batch_size = 2048
+lrate = 0.01
+
+
+

configs/room.txt

@@ -1,15 +1,24 @@
-expname = room_test
-basedir = ./logs
-datadir = ./data/nerf_llff_data/room
+output_dir = ./output
+obs_imgs_dir = ./data/nerf_llff_data/room
+ckpt_dir = ./ckpts
+
+ckpt_name = room
+model_name = room
 dataset_type = llff
 
 factor = 8
 llffhold = 8
-
-N_rand = 1024
+use_viewdirs = True
 N_samples = 64
 N_importance = 64
-
-use_viewdirs = True
 raw_noise_std = 1e0
 
+obs_img_num = 18
+dil_iter = 1
+kernel_size = 3
+start_pose_num = 19
+batch_size = 2048
+lrate = 0.01
+
+
+

nerf_helpers.py

@@ -47,11 +47,15 @@ def load_nerf(args, device):
         'network_fn': model,
         'use_viewdirs': args.use_viewdirs,
         'white_bkgd': args.white_bkgd,
-        'raw_noise_std': args.raw_noise_std,
-        'ndc' : False,
-        'lindisp' : args.lindisp
+        'raw_noise_std': args.raw_noise_std
     }
 
+    # NDC only good for LLFF-style forward facing data
+    if args.dataset_type != 'llff' or args.no_ndc:
+        print('Not ndc!')
+        render_kwargs['ndc'] = False
+        render_kwargs['lindisp'] = args.lindisp
+
     # Disable updating of the weights
     for param in model.parameters():
         param.requires_grad = False