File size: 1,709 Bytes
ec9a6bc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
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