Add fern, hotdog, ship

Signed-off-by: Krishna Murthy <[email protected]>
krrish94 · Apr 16, 2020 · 4eb3359 · 4eb3359
1 parent e9adb2d
commit 4eb3359
Show file tree

Hide file tree

Showing 6 changed files with 205 additions and 0 deletions.
diff --git a/pretrained/fern-lowres/checkpoint249999.ckpt b/pretrained/fern-lowres/checkpoint249999.ckpt
diff --git a/pretrained/fern-lowres/config.yml b/pretrained/fern-lowres/config.yml
@@ -0,0 +1,68 @@
+dataset:
+  basedir: cache/nerf_llff_data/fern
+  downsample_factor: 8
+  far: 1
+  llffhold: 8
+  near: 0
+  no_ndc: false
+  type: llff
+experiment:
+  id: fern
+  logdir: logs
+  print_every: 100
+  randomseed: 34
+  save_every: 5000
+  train_iters: 250000
+  validate_every: 5000
+models:
+  coarse:
+    hidden_size: 64
+    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: 6
+    num_layers: 4
+    skip_connect_every: 3
+    type: FlexibleNeRFModel
+    use_viewdirs: true
+  fine:
+    hidden_size: 64
+    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: 6
+    num_layers: 4
+    skip_connect_every: 3
+    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: 1.0
+    white_background: false
+  use_viewdirs: true
+  validation:
+    chunksize: 16384
+    lindisp: false
+    num_coarse: 64
+    num_fine: 64
+    perturb: false
+    radiance_field_noise_std: 0.0
+    white_background: false
+optimizer:
+  lr: 0.005
+  type: Adam
+scheduler:
+  lr_decay: 250
+  lr_decay_factor: 0.1
diff --git a/pretrained/hotdog-lowres/checkpoint100000.ckpt b/pretrained/hotdog-lowres/checkpoint100000.ckpt
diff --git a/pretrained/hotdog-lowres/config.yml b/pretrained/hotdog-lowres/config.yml
@@ -0,0 +1,68 @@
+dataset:
+  basedir: cache/nerf_synthetic/hotdog
+  far: 6
+  half_res: true
+  near: 2
+  no_ndc: true
+  testskip: 1
+  type: blender
+experiment:
+  id: hotdog
+  logdir: logs
+  print_every: 100
+  randomseed: 42
+  save_every: 5000
+  train_iters: 200000
+  validate_every: 5000
+models:
+  coarse:
+    hidden_size: 64
+    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: 4
+    skip_connect_every: 4
+    type: FlexibleNeRFModel
+    use_viewdirs: true
+  fine:
+    hidden_size: 64
+    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: 4
+    skip_connect_every: 4
+    type: FlexibleNeRFModel
+    use_viewdirs: true
+nerf:
+  encode_direction_fn: positional_encoding
+  encode_position_fn: positional_encoding
+  train:
+    chunksize: 32768
+    lindisp: false
+    num_coarse: 64
+    num_fine: 128
+    num_random_rays: 1024
+    perturb: true
+    radiance_field_noise_std: 0.2
+    white_background: true
+  use_viewdirs: true
+  validation:
+    chunksize: 32768
+    lindisp: false
+    num_coarse: 64
+    num_fine: 128
+    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/ship-lowres/checkpoint125000.ckpt b/pretrained/ship-lowres/checkpoint125000.ckpt
diff --git a/pretrained/ship-lowres/config.yml b/pretrained/ship-lowres/config.yml
@@ -0,0 +1,69 @@
+dataset:
+  basedir: cache/nerf_synthetic/ship
+  cachedir: cache/shipcache200
+  far: 6
+  half_res: true
+  near: 2
+  no_ndc: true
+  testskip: 1
+  type: blender
+experiment:
+  id: ship
+  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: 128
+    num_random_rays: 4096
+    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