Add pretrained models: chair, drums, materials

Signed-off-by: Krishna Murthy <[email protected]>
krrish94 · Apr 17, 2020 · a3d63f9 · a3d63f9
1 parent 317b14d
commit a3d63f9
Show file tree

Hide file tree

Showing 8 changed files with 207 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -141,8 +141,11 @@ The organization of code **WILL** change around a lot, because I'm actively expe
 **Pretrained models**: Pretrained models for the following scenes are available in the `pretrained` directory (all of them are currently lowres). I will continue adding models herein.
 ```
 # Synthetic (Blender) scenes
+chair
+drums
 hotdog
 lego
+materials
 ship
 
 # Real (LLFF) scenes

diff --git a/assets/blender-lowres.gif b/assets/blender-lowres.gif
diff --git a/pretrained/chair-lowres/checkpoint155000.ckpt b/pretrained/chair-lowres/checkpoint155000.ckpt
diff --git a/pretrained/chair-lowres/config.yml b/pretrained/chair-lowres/config.yml
@@ -0,0 +1,68 @@
+dataset:
+  basedir: cache/nerf_synthetic/chair
+  far: 6
+  half_res: true
+  near: 2
+  no_ndc: true
+  testskip: 1
+  type: blender
+experiment:
+  id: chair
+  logdir: logs
+  print_every: 100
+  randomseed: 42
+  save_every: 5000
+  train_iters: 200000
+  validate_every: 5000
+models:
+  coarse:
+    hidden_size: 256
+    include_input_dir: true
+    include_input_xyz: true
+    log_sampling_dir: true
+    log_sampling_xyz: true
+    num_encoding_fn_dir: 4
+    num_encoding_fn_xyz: 10
+    num_layers: 8
+    skip_connect_every: 4
+    type: FlexibleNeRFModel
+    use_viewdirs: true
+  fine:
+    hidden_size: 256
+    include_input_dir: true
+    include_input_xyz: true
+    log_sampling_dir: true
+    log_sampling_xyz: true
+    num_encoding_fn_dir: 4
+    num_encoding_fn_xyz: 10
+    num_layers: 8
+    skip_connect_every: 4
+    type: FlexibleNeRFModel
+    use_viewdirs: true
+nerf:
+  encode_direction_fn: positional_encoding
+  encode_position_fn: positional_encoding
+  train:
+    chunksize: 16384
+    lindisp: false
+    num_coarse: 64
+    num_fine: 64
+    num_random_rays: 1024
+    perturb: true
+    radiance_field_noise_std: 0.2
+    white_background: true
+  use_viewdirs: true
+  validation:
+    chunksize: 16384
+    lindisp: false
+    num_coarse: 64
+    num_fine: 64
+    perturb: false
+    radiance_field_noise_std: 0.0
+    white_background: true
+optimizer:
+  lr: 0.005
+  type: Adam
+scheduler:
+  lr_decay: 250
+  lr_decay_factor: 0.1
diff --git a/pretrained/drums-lowres/checkpoint199999.ckpt b/pretrained/drums-lowres/checkpoint199999.ckpt
diff --git a/pretrained/drums-lowres/config.yml b/pretrained/drums-lowres/config.yml
@@ -0,0 +1,68 @@
+dataset:
+  basedir: cache/nerf_synthetic/drums
+  far: 6
+  half_res: true
+  near: 2
+  no_ndc: true
+  testskip: 1
+  type: blender
+experiment:
+  id: drums
+  logdir: logs
+  print_every: 100
+  randomseed: 42
+  save_every: 5000
+  train_iters: 200000
+  validate_every: 5000
+models:
+  coarse:
+    hidden_size: 256
+    include_input_dir: true
+    include_input_xyz: true
+    log_sampling_dir: true
+    log_sampling_xyz: true
+    num_encoding_fn_dir: 4
+    num_encoding_fn_xyz: 10
+    num_layers: 8
+    skip_connect_every: 4
+    type: FlexibleNeRFModel
+    use_viewdirs: true
+  fine:
+    hidden_size: 256
+    include_input_dir: true
+    include_input_xyz: true
+    log_sampling_dir: true
+    log_sampling_xyz: true
+    num_encoding_fn_dir: 4
+    num_encoding_fn_xyz: 10
+    num_layers: 8
+    skip_connect_every: 4
+    type: FlexibleNeRFModel
+    use_viewdirs: true
+nerf:
+  encode_direction_fn: positional_encoding
+  encode_position_fn: positional_encoding
+  train:
+    chunksize: 16384
+    lindisp: false
+    num_coarse: 64
+    num_fine: 64
+    num_random_rays: 1024
+    perturb: true
+    radiance_field_noise_std: 0.2
+    white_background: true
+  use_viewdirs: true
+  validation:
+    chunksize: 16384
+    lindisp: false
+    num_coarse: 64
+    num_fine: 64
+    perturb: false
+    radiance_field_noise_std: 0.0
+    white_background: true
+optimizer:
+  lr: 0.005
+  type: Adam
+scheduler:
+  lr_decay: 250
+  lr_decay_factor: 0.1
diff --git a/pretrained/materials-lowres/checkpoint199999.ckpt b/pretrained/materials-lowres/checkpoint199999.ckpt
diff --git a/pretrained/materials-lowres/config.yml b/pretrained/materials-lowres/config.yml
@@ -0,0 +1,68 @@
+dataset:
+  basedir: cache/nerf_synthetic/materials
+  far: 6
+  half_res: true
+  near: 2
+  no_ndc: true
+  testskip: 1
+  type: blender
+experiment:
+  id: materials
+  logdir: logs
+  print_every: 100
+  randomseed: 42
+  save_every: 5000
+  train_iters: 200000
+  validate_every: 5000
+models:
+  coarse:
+    hidden_size: 256
+    include_input_dir: true
+    include_input_xyz: true
+    log_sampling_dir: true
+    log_sampling_xyz: true
+    num_encoding_fn_dir: 4
+    num_encoding_fn_xyz: 10
+    num_layers: 8
+    skip_connect_every: 4
+    type: FlexibleNeRFModel
+    use_viewdirs: true
+  fine:
+    hidden_size: 256
+    include_input_dir: true
+    include_input_xyz: true
+    log_sampling_dir: true
+    log_sampling_xyz: true
+    num_encoding_fn_dir: 4
+    num_encoding_fn_xyz: 10
+    num_layers: 8
+    skip_connect_every: 4
+    type: FlexibleNeRFModel
+    use_viewdirs: true
+nerf:
+  encode_direction_fn: positional_encoding
+  encode_position_fn: positional_encoding
+  train:
+    chunksize: 16384
+    lindisp: false
+    num_coarse: 64
+    num_fine: 64
+    num_random_rays: 1024
+    perturb: true
+    radiance_field_noise_std: 0.2
+    white_background: true
+  use_viewdirs: true
+  validation:
+    chunksize: 16384
+    lindisp: false
+    num_coarse: 64
+    num_fine: 64
+    perturb: false
+    radiance_field_noise_std: 0.0
+    white_background: true
+optimizer:
+  lr: 0.005
+  type: Adam
+scheduler:
+  lr_decay: 250
+  lr_decay_factor: 0.1