Spaces:
Sleeping
Sleeping
gpu_id: 0 | |
load_meshhead_checkpoint: 'checkpoints/meshhead_huawei0425/meshhead_latest' | |
load_gaussianhead_checkpoint: 'checkpoints/meshhead_huawei0425/gaussianhead_latest' | |
load_supres_checkpoint: 'checkpoints/meshhead_huawei0425/supres_latest' | |
load_delta_poses_checkpoint: 'checkpoints/meshhead_huawei0425/delta_poses_latest' | |
lr_net: 1e-4 | |
lr_pose: 1e-5 | |
batch_size: 1 | |
optimize_pose: True | |
dataset: | |
dataroot: '../data/face_data' | |
# dataroot: 'D:/GaussianSplatting/data/Shuangqing/huawei0425/face0425_single' | |
camera_ids: ['01', '02', '04', '05'] | |
original_resolution: 2048 | |
resolution: 2048 | |
meshheadmodule: | |
geo_mlp: [27, 256, 256, 256, 256, 256, 132] | |
exp_color_mlp: [180, 256, 256, 32] | |
pose_color_mlp: [182, 128, 32] | |
exp_deform_mlp: [79, 256, 256, 256, 256, 256, 3] | |
pose_deform_mlp: [81, 256, 256, 3] | |
pos_freq: 4 | |
model_bbox: [[-1.6, 1.6], [-1.7, 1.8], [-2.5, 1.0]] | |
dist_threshold_near: 0.1 | |
dist_threshold_far: 0.25 | |
deform_scale: 0.3 | |
subdivide: False | |
supresmodule: | |
input_dim: 32 | |
output_dim: 3 | |
network_capacity: 32 | |
gaussianheadmodule: | |
num_add_mouth_points: 3000 | |
exp_color_mlp: [180, 256, 256, 32] | |
pose_color_mlp: [182, 128, 32] | |
exp_deform_mlp: [79, 256, 256, 256, 256, 256, 3] | |
pose_deform_mlp: [81, 256, 256, 3] | |
exp_attributes_mlp: [180, 256, 256, 256, 8] | |
pose_attributes_mlp: [182, 128, 128, 8] | |
exp_coeffs_dim: 64 | |
pos_freq: 4 | |
dist_threshold_near: 0.1 | |
dist_threshold_far: 0.25 | |
deform_scale: 0.3 | |
attributes_scale: 0.2 | |
recorder: | |
name: 'gaussianhead_huawei0425' | |
logdir: 'log/gaussianhead_huawei0425' | |
checkpoint_path: 'checkpoints' | |
result_path: 'results' | |
save_freq: 1000 | |
show_freq: 100 | |