clean code
This commit is contained in:
parent
2f5d724818
commit
a26a4b94ce
5 changed files with 5 additions and 163 deletions
|
@ -4,8 +4,8 @@ Codebase for paper:
|
||||||
* Support multi-gpu training and inference.
|
* Support multi-gpu training and inference.
|
||||||
|
|
||||||
## Data
|
## Data
|
||||||
* Download our preprocessed data from [tanks_and_temples](), [lf_data]().
|
* Download our preprocessed data from [tanks_and_temples](https://drive.google.com/file/d/11KRfN91W1AxAW6lOFs4EeYDbeoQZCi87/view?usp=sharing), [lf_data](https://drive.google.com/file/d/1gsjDjkbTh4GAR9fFqlIDZ__qR9NYTURQ/view?usp=sharing).
|
||||||
* Put the data in the code directory.
|
* Put the data in the sub-folder data/ of this code directory.
|
||||||
* Data format.
|
* Data format.
|
||||||
* Each scene consists of 3 splits: train/test/validation.
|
* Each scene consists of 3 splits: train/test/validation.
|
||||||
* Intrinsics and poses are stored as flattened 4x4 matrices.
|
* Intrinsics and poses are stored as flattened 4x4 matrices.
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
### INPUT
|
### INPUT
|
||||||
datadir = /home/zhangka2/gernot_experi/lf_data/lf_nerf
|
datadir = ./data/lf_data
|
||||||
scene = africa
|
scene = africa
|
||||||
expname = africa_ddp
|
expname = africa
|
||||||
basedir = ./logs
|
basedir = ./logs
|
||||||
config = None
|
config = None
|
||||||
ckpt_path = None
|
ckpt_path = None
|
||||||
|
@ -18,15 +18,9 @@ lrate_decay_steps = 50000000
|
||||||
### CASCADE
|
### CASCADE
|
||||||
cascade_level = 2
|
cascade_level = 2
|
||||||
cascade_samples = 64,128
|
cascade_samples = 64,128
|
||||||
near_depth = 0.
|
|
||||||
far_depth = 1.
|
|
||||||
|
|
||||||
### TESTING
|
### TESTING
|
||||||
render_only = False
|
chunk_size = 8192
|
||||||
render_test = False
|
|
||||||
render_train = False
|
|
||||||
# chunk_size = 16384
|
|
||||||
chunk_size = 4096
|
|
||||||
|
|
||||||
### RENDERING
|
### RENDERING
|
||||||
det = False
|
det = False
|
||||||
|
@ -34,15 +28,9 @@ max_freq_log2 = 10
|
||||||
max_freq_log2_viewdirs = 4
|
max_freq_log2_viewdirs = 4
|
||||||
netdepth = 8
|
netdepth = 8
|
||||||
netwidth = 256
|
netwidth = 256
|
||||||
raw_noise_std = 1.0
|
|
||||||
N_iters_perturb = 1000
|
|
||||||
inv_uniform = False
|
|
||||||
use_viewdirs = True
|
use_viewdirs = True
|
||||||
white_bkgd = False
|
|
||||||
|
|
||||||
### CONSOLE AND TENSORBOARD
|
### CONSOLE AND TENSORBOARD
|
||||||
i_img = 2000
|
i_img = 2000
|
||||||
i_print = 100
|
i_print = 100
|
||||||
i_testset = 5000000
|
|
||||||
i_video = 5000000
|
|
||||||
i_weights = 5000
|
i_weights = 5000
|
||||||
|
|
|
@ -1,48 +0,0 @@
|
||||||
### INPUT
|
|
||||||
datadir = /home/zhangka2/gernot_experi/lf_data/lf_nerf
|
|
||||||
scene = basket
|
|
||||||
expname = basket_ddp
|
|
||||||
basedir = ./logs
|
|
||||||
config = None
|
|
||||||
ckpt_path = None
|
|
||||||
no_reload = False
|
|
||||||
testskip = 1
|
|
||||||
|
|
||||||
### TRAINING
|
|
||||||
N_iters = 1250001
|
|
||||||
N_rand = 512
|
|
||||||
lrate = 0.0005
|
|
||||||
lrate_decay_factor = 0.1
|
|
||||||
lrate_decay_steps = 50000000
|
|
||||||
|
|
||||||
### CASCADE
|
|
||||||
cascade_level = 2
|
|
||||||
cascade_samples = 64,128
|
|
||||||
near_depth = 0.
|
|
||||||
far_depth = 1.
|
|
||||||
|
|
||||||
### TESTING
|
|
||||||
render_only = False
|
|
||||||
render_test = False
|
|
||||||
render_train = False
|
|
||||||
# chunk_size = 16384
|
|
||||||
chunk_size = 4096
|
|
||||||
|
|
||||||
### RENDERING
|
|
||||||
det = False
|
|
||||||
max_freq_log2 = 10
|
|
||||||
max_freq_log2_viewdirs = 4
|
|
||||||
netdepth = 8
|
|
||||||
netwidth = 256
|
|
||||||
raw_noise_std = 1.0
|
|
||||||
N_iters_perturb = 1000
|
|
||||||
inv_uniform = False
|
|
||||||
use_viewdirs = True
|
|
||||||
white_bkgd = False
|
|
||||||
|
|
||||||
### CONSOLE AND TENSORBOARD
|
|
||||||
i_img = 2000
|
|
||||||
i_print = 100
|
|
||||||
i_testset = 5000000
|
|
||||||
i_video = 5000000
|
|
||||||
i_weights = 5000
|
|
|
@ -1,50 +0,0 @@
|
||||||
### INPUT
|
|
||||||
datadir = /home/zhangka2/gernot_experi/lf_data/lf_nerf
|
|
||||||
scene = ship
|
|
||||||
expname = ship_ddp
|
|
||||||
basedir = ./logs
|
|
||||||
config = None
|
|
||||||
ckpt_path = None
|
|
||||||
no_reload = False
|
|
||||||
testskip = 1
|
|
||||||
|
|
||||||
### TRAINING
|
|
||||||
N_iters = 1250001
|
|
||||||
# N_rand = 512
|
|
||||||
N_rand = 1024
|
|
||||||
lrate = 0.0005
|
|
||||||
lrate_decay_factor = 0.1
|
|
||||||
lrate_decay_steps = 50000000
|
|
||||||
|
|
||||||
### CASCADE
|
|
||||||
cascade_level = 2
|
|
||||||
cascade_samples = 64,128
|
|
||||||
near_depth = 0.
|
|
||||||
far_depth = 1.
|
|
||||||
|
|
||||||
### TESTING
|
|
||||||
render_only = False
|
|
||||||
render_test = False
|
|
||||||
render_train = False
|
|
||||||
# chunk_size = 16384
|
|
||||||
# chunk_size = 4096
|
|
||||||
chunk_size = 8192
|
|
||||||
|
|
||||||
### RENDERING
|
|
||||||
det = False
|
|
||||||
max_freq_log2 = 10
|
|
||||||
max_freq_log2_viewdirs = 4
|
|
||||||
netdepth = 8
|
|
||||||
netwidth = 256
|
|
||||||
raw_noise_std = 1.0
|
|
||||||
N_iters_perturb = 1000
|
|
||||||
inv_uniform = False
|
|
||||||
use_viewdirs = True
|
|
||||||
white_bkgd = False
|
|
||||||
|
|
||||||
### CONSOLE AND TENSORBOARD
|
|
||||||
i_img = 2000
|
|
||||||
i_print = 100
|
|
||||||
i_testset = 5000000
|
|
||||||
i_video = 5000000
|
|
||||||
i_weights = 5000
|
|
|
@ -1,48 +0,0 @@
|
||||||
### INPUT
|
|
||||||
datadir = /home/zhangka2/gernot_experi/lf_data/lf_nerf
|
|
||||||
scene = torch
|
|
||||||
expname = torch_ddp
|
|
||||||
basedir = ./logs
|
|
||||||
config = None
|
|
||||||
ckpt_path = None
|
|
||||||
no_reload = False
|
|
||||||
testskip = 1
|
|
||||||
|
|
||||||
### TRAINING
|
|
||||||
N_iters = 1250001
|
|
||||||
N_rand = 512
|
|
||||||
lrate = 0.0005
|
|
||||||
lrate_decay_factor = 0.1
|
|
||||||
lrate_decay_steps = 50000000
|
|
||||||
|
|
||||||
### CASCADE
|
|
||||||
cascade_level = 2
|
|
||||||
cascade_samples = 64,128
|
|
||||||
near_depth = 0.
|
|
||||||
far_depth = 1.
|
|
||||||
|
|
||||||
### TESTING
|
|
||||||
render_only = False
|
|
||||||
render_test = False
|
|
||||||
render_train = False
|
|
||||||
# chunk_size = 16384
|
|
||||||
chunk_size = 4096
|
|
||||||
|
|
||||||
### RENDERING
|
|
||||||
det = False
|
|
||||||
max_freq_log2 = 10
|
|
||||||
max_freq_log2_viewdirs = 4
|
|
||||||
netdepth = 8
|
|
||||||
netwidth = 256
|
|
||||||
raw_noise_std = 1.0
|
|
||||||
N_iters_perturb = 1000
|
|
||||||
inv_uniform = False
|
|
||||||
use_viewdirs = True
|
|
||||||
white_bkgd = False
|
|
||||||
|
|
||||||
### CONSOLE AND TENSORBOARD
|
|
||||||
i_img = 2000
|
|
||||||
i_print = 100
|
|
||||||
i_testset = 5000000
|
|
||||||
i_video = 5000000
|
|
||||||
i_weights = 5000
|
|
Loading…
Reference in a new issue