configs/example.yaml

trial_name: "body_head_avatar"
device: cuda
animatablegaussians:
  train:
    dataset: MvRgbDatasetAvatarReX
    data:
      subject_name: 1007_slow10
      data_dir: ./checkpoints/pos_map_ys/body_mix
      frame_range: &id001
      - 0
      - 200
      - 1
      used_cam_ids:
      - 0
      - 1
      - 2
      - 3
      - 4
      - 5
      - 6
      - 8
      - 9
      - 10
      - 11
      - 12
      - 14
      - 15
      load_smpl_pos_map: true
    pretrained_dir: null
    net_ckpt_dir: ./results/huawei0425/avatar2
    prev_ckpt: null
    ckpt_interval:
      epoch: 10
      batch: 50000
    eval_interval: 1000
    eval_training_ids:
    - 190
    - 7
    eval_testing_ids:
    - 354
    - 7
    eval_img_factor: 1.0
    lr_init: 0.0005
    loss_weight:
      l1: 1.0
      lpips: 0.1
      offset: 0.005
    finetune_color: false
    batch_size: 1
    num_workers: 8
    random_bg_color: true
  test:
    output_dir: ./test_results/temp_test
    dataset: MvRgbDatasetAvatarReX
    data:
      data_dir: ./checkpoints/pos_map_ys/body_mix
      frame_range: [0, 800]
      subject_name: huawei0425
    pose_data:
      data_path: ./test_data/AMASS/1007_train_data_slow10.npz
      frame_range: [0, 2000]
    view_setting: degree90
    render_view_idx: 13
    global_orient: true
    img_scale: 2.0
    save_mesh: false
    render_skeleton: false
    save_tex_map: false
    save_ply: true
    fix_hand: true
    fix_hand_id: 23
    n_pca: 20
    sigma_pca: 2.0
    prev_ckpt: ./checkpoints/checkpoints/body_ys
  model:
    with_viewdirs: true
    random_style: false

gha:
  config_path: configs/head.yaml
  offline_rendering_param_fpath: ./checkpoints/render_param/head_zoomin_render_param.npz

cat:
  body_gaussian_root_dir: ./checkpoints/pos_map_ys/body_mix
  ref_head_gaussian_path: ./checkpoints/ref_gaussian/head/000000.ply
  ref_head_param_path: ./checkpoints/ref_gaussian/head/000000_param.npz
  render_cam_fpath: /home/pengc02/pengcheng/projects/gaussian_avatar/avatar_final/AnimatableGaussians/test_results/1007_slow10/checkpoints/AMASS__1007_train_data_slow10_degree90_view/batch_789377/pca_20_sigma_2.00/cameras.json
  body_head_blending_param_path: ./checkpoints/render_param/body_head_blending_param.npz