Hugging Face's logo

Spaces:
pengsida
/
NeuralBody
Configuration error

App Files Community
2
NeuralBody
File size: 3,838 Bytes 
1ba539f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
 
task: 'if_nerf'
gpus: [0]

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

network_module: 'lib.networks.nerf'
network_path: 'lib/networks/nerf.py'
renderer_module: 'lib.networks.renderer.volume_renderer'
renderer_path: 'lib/networks/renderer/volume_renderer.py'

trainer_module: 'lib.train.trainers.nerf.py'
trainer_path: 'lib/train/trainers/nerf.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'

human: 313

train_dataset:
    data_root: 'data/zju_mocap/CoreView_313'
    human: 'CoreView_313'
    ann_file: 'data/zju_mocap/CoreView_313/annots.npy'
    split: 'train'

test_dataset:
    data_root: 'data/zju_mocap/CoreView_313'
    human: 'CoreView_313'
    ann_file: 'data/zju_mocap/CoreView_313/annots.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:
    sampler: 'FrameSampler'
    batch_size: 1
    collator: ''

ep_iter: 500
save_ep: 1000
eval_ep: 1000

# training options
netdepth: 8
netwidth: 256
netdepth_fine: 8
netwidth_fine: 256
netchunk: 65536
chunk: 32768

no_batching: True

# rendering options
use_viewdirs: True
i_embed: 0
xyz_res: 10
view_res: 4
raw_noise_std: 0
lindisp: False

N_samples: 64
N_importance: 128
N_rand: 1024

perturb: 1
white_bkgd: False

num_render_views: 50

# data options
ratio: 0.5
num_train_frame: 1
smpl: 'smpl'
params: 'params'

voxel_size: [0.005, 0.005, 0.005]  # dhw

# record options
log_interval: 1


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

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

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

    test:
        sampler: ''

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

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

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

    test:
        sampler: ''

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

    network_module: 'lib.networks.latent_xyzc'
    network_path: 'lib/networks/latent_xyzc.py'
    renderer_module: 'lib.networks.renderer.volume_mesh_renderer'
    renderer_path: 'lib/networks/renderer/volume_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