diff --git a/README.md b/README.md index e23ca2d..1c74769 100644 --- 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 index 6953a9a..858cdd5 100644 Binary files a/assets/blender-lowres.gif and b/assets/blender-lowres.gif differ diff --git a/pretrained/chair-lowres/checkpoint155000.ckpt b/pretrained/chair-lowres/checkpoint155000.ckpt new file mode 100644 index 0000000..50ba30d Binary files /dev/null and b/pretrained/chair-lowres/checkpoint155000.ckpt differ diff --git a/pretrained/chair-lowres/config.yml b/pretrained/chair-lowres/config.yml new file mode 100644 index 0000000..4889674 --- /dev/null +++ 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 new file mode 100644 index 0000000..c244338 Binary files /dev/null and b/pretrained/drums-lowres/checkpoint199999.ckpt differ diff --git a/pretrained/drums-lowres/config.yml b/pretrained/drums-lowres/config.yml new file mode 100644 index 0000000..50cc580 --- /dev/null +++ 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 new file mode 100644 index 0000000..2690cbf Binary files /dev/null and b/pretrained/materials-lowres/checkpoint199999.ckpt differ diff --git a/pretrained/materials-lowres/config.yml b/pretrained/materials-lowres/config.yml new file mode 100644 index 0000000..7902884 --- /dev/null +++ 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