full_gaussian_avatar / GHA /config /train_gaussianhead_N031.yaml
pengc02's picture
all
ec9a6bc
raw
history blame
1.71 kB
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