mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
2628fb56f2
Summary: Implements the test script of NeRF. Reviewed By: nikhilaravi Differential Revision: D25684450 fbshipit-source-id: 739169d9df706795814912bb9a15e2e65ac92df8
46 lines
999 B
YAML
46 lines
999 B
YAML
seed: 3
|
|
resume: True
|
|
stats_print_interval: 10
|
|
validation_epoch_interval: 30
|
|
checkpoint_epoch_interval: 30
|
|
checkpoint_path: 'checkpoints/lego_pt3d.pth'
|
|
data:
|
|
dataset_name: 'lego'
|
|
image_size: [800, 800] # [height, width]
|
|
precache_rays: True
|
|
test:
|
|
mode: 'evaluation'
|
|
trajectory_type: 'circular'
|
|
up: [0.0, 0.0, 1.0]
|
|
scene_center: [0.0, 0.0, 0.0]
|
|
n_frames: 100
|
|
fps: 20
|
|
trajectory_scale: 0.2
|
|
optimizer:
|
|
max_epochs: 20000
|
|
lr: 0.0005
|
|
lr_scheduler_step_size: 5000
|
|
lr_scheduler_gamma: 0.1
|
|
visualization:
|
|
history_size: 10
|
|
visdom: True
|
|
visdom_server: 'localhost'
|
|
visdom_port: 8097
|
|
visdom_env: 'nerf_pytorch3d'
|
|
raysampler:
|
|
n_pts_per_ray: 64
|
|
n_pts_per_ray_fine: 64
|
|
n_rays_per_image: 1024
|
|
min_depth: 2.0
|
|
max_depth: 6.0
|
|
stratified: True
|
|
stratified_test: False
|
|
chunk_size_test: 6000
|
|
implicit_function:
|
|
n_harmonic_functions_xyz: 10
|
|
n_harmonic_functions_dir: 4
|
|
n_hidden_neurons_xyz: 256
|
|
n_hidden_neurons_dir: 128
|
|
density_noise_std: 0.0
|
|
n_layers_xyz: 8
|