task: 'if_nerf'
gpus: [0]

train_dataset_module: 'lib.datasets.light_stage.monocular_dataset'
train_dataset_path: 'lib/datasets/light_stage/monocular_dataset.py'
test_dataset_module: 'lib.datasets.light_stage.monocular_dataset'
test_dataset_path: 'lib/datasets/light_stage/monocular_dataset.py'

network_module: 'lib.networks.latent_xyzc'
network_path: 'lib/networks/latent_xyzc.py'
renderer_module: 'lib.networks.renderer.if_clight_renderer'
renderer_path: 'lib/networks/renderer/if_clight_renderer.py'

trainer_module: 'lib.train.trainers.if_nerf_clight'
trainer_path: 'lib/train/trainers/if_nerf_clight.py'

evaluator_module: 'lib.evaluators.if_nerf'
evaluator_path: 'lib/evaluators/if_nerf.py'

visualizer_module: 'lib.visualizers.if_nerf'
visualizer_path: 'lib/visualizers/if_nerf.py'

train_dataset:
    data_root: 'data/people_snapshot/female-3-casual'
    human: 'female-3-casual'
    ann_file: 'data/people_snapshot/female-3-casual/params.npy'
    split: 'train'

test_dataset:
    data_root: 'data/people_snapshot/female-3-casual'
    human: 'female-3-casual'
    ann_file: 'data/people_snapshot/female-3-casual/params.npy'
    split: 'test'

train:
    batch_size: 1
    collator: ''
    lr: 5e-4
    weight_decay: 0
    epoch: 400
    scheduler:
        type: 'exponential'
        gamma: 0.1
        decay_epochs: 1000
    num_workers: 16

test:
    batch_size: 1
    collator: ''

ep_iter: 500
save_ep: 100
eval_ep: 1000

# rendering options
i_embed: 0
xyz_res: 10
view_res: 4
raw_noise_std: 0

N_samples: 64
N_importance: 128
N_rand: 1024

perturb: 1
white_bkgd: False

num_render_views: 50

# data options
H: 1080
W: 1080
ratio: 1.
num_train_frame: 230

voxel_size: [0.005, 0.005, 0.005]  # dhw

# record options
log_interval: 1


novel_view_cfg:
    train_dataset_module: 'lib.datasets.light_stage.monocular_demo_dataset'
    train_dataset_path: 'lib/datasets/light_stage/monocular_demo_dataset.py'
    test_dataset_module: 'lib.datasets.light_stage.monocular_demo_dataset'
    test_dataset_path: 'lib/datasets/light_stage/monocular_demo_dataset.py'

    renderer_module: 'lib.networks.renderer.if_clight_renderer_msk'
    renderer_path: 'lib/networks/renderer/if_clight_renderer_msk.py'

    visualizer_module: 'lib.visualizers.if_nerf_demo'
    visualizer_path: 'lib/visualizers/if_nerf_demo.py'

    ratio: 0.5

    test:
        sampler: ''

novel_pose_cfg:
    train_dataset_module: 'lib.datasets.light_stage.monocular_dataset'
    train_dataset_path: 'lib/datasets/light_stage/monocular_dataset.py'
    test_dataset_module: 'lib.datasets.light_stage.monocular_dataset'
    test_dataset_path: 'lib/datasets/light_stage/monocular_dataset.py'

    renderer_module: 'lib.networks.renderer.if_clight_renderer_msk'
    renderer_path: 'lib/networks/renderer/if_clight_renderer_msk.py'

    visualizer_module: 'lib.visualizers.if_nerf_perform'
    visualizer_path: 'lib/visualizers/if_nerf_perform.py'

    ratio: 0.5

    test:
        sampler: ''

mesh_cfg:
    train_dataset_module: 'lib.datasets.light_stage.monocular_mesh_dataset'
    train_dataset_path: 'lib/datasets/light_stage/monocular_mesh_dataset.py'
    test_dataset_module: 'lib.datasets.light_stage.monocular_mesh_dataset'
    test_dataset_path: 'lib/datasets/light_stage/monocular_mesh_dataset.py'

    network_module: 'lib.networks.latent_xyzc'
    network_path: 'lib/networks/latent_xyzc.py'
    renderer_module: 'lib.networks.renderer.if_mesh_renderer'
    renderer_path: 'lib/networks/renderer/if_mesh_renderer.py'

    visualizer_module: 'lib.visualizers.if_nerf_mesh'
    visualizer_path: 'lib/visualizers/if_nerf_mesh.py'

    mesh_th: 5

    test:
        sampler: 'FrameSampler'
        frame_sampler_interval: 1