GHA/config/train_meshhead_N031.yaml

gpu_id: 0
load_meshhead_checkpoint: 'checkpoints/meshhead_NeRSemble031/meshhead_latest'
lr_net: 1e-3
lr_lmk: 1e-4
batch_size: 1

dataset:
  # dataroot: '../data/face_data'
  dataroot: 'D:/GaussianSplatting/data/Shuangqing/huawei0425/face0425_single'
  camera_ids: ['01', '02', '04', '05']
  original_resolution: 2048
  resolution: 256
  num_sample_view: 4

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

recorder:
  name: 'meshhead_huawei0425'
  logdir: 'log/meshhead_huawei0425'
  checkpoint_path: 'checkpoints'
  result_path: 'results'
  save_freq: 1000
  show_freq: 100