preproc_video.yaml

base_dir: '/media/perukas/Data/Datasets/2d-datasets/CelebV-HQ/downloaded_celebvhq/processed'
output_dir: '/media/perukas/Data/Datasets/2d-datasets/CelebV-HQ-FLAME/'

mica_estimator_kwargs:
  chkp_path: "/media/perukas/Home/_dev/_phd/prjs/MICA/data/pretrained/mica.tar"

video_dataset_kwargs:
  filepath: '/media/perukas/Data/Datasets/2d-datasets/CelebV-HQ/downloaded_celebvhq/processed/-2KG4lLGEl0_1.mp4'
  preload: true
  scale_factor: 2


flame_pose_expression_optimization_kwargs:
  optim_iters: 1000
  log_result_only: true
  cam_init_z_trans: 0.5

  face_parsing_kwargs:
    use_fan: false
    mp_face_mesh_detector_kwargs:
      max_num_faces: 1
      min_detection_conf: 0.5
    face_segmentor_kwargs:
      threshold: 0.8
      chkp: /media/perukas/Home/_dev/_phd/libraries/face_parsing/ibug/face_parsing/rtnet/weights/rtnet50-fcn-11.torch
      nclasses: 11
  
  flame_model_cfg:
    flame_model_path: /media/perukas/Home/_dev/_phd/3dmms/FLAME/data/FLAME2020/generic_model.pkl
    batch_size: 1
    use_face_contour: true
    shape_params: 300
    expression_params: 100
    use_3D_translation: false
    static_landmark_embedding_path: /media/perukas/Home/_dev/_phd/3dmms/FLAME/data/flame_static_embedding.pkl
    dynamic_landmark_embedding_path: /media/perukas/Home/_dev/_phd/3dmms/FLAME/data/flame_dynamic_embedding.npy
    mediapipe_landmark_embedding_path: /media/perukas/Home/_dev/_phd/3dmms/FLAME/data/mediapipe_landmark_embedding/mediapipe_landmark_embedding.npz
    flame_masks: /media/perukas/Home/_dev/_phd/3dmms/FLAME/data/FLAME_masks/FLAME_masks.pkl
  
  loss_kwargs:
    w_mp: 0.3
    w_seg: 0.6
    w_reg: 0.8
    wing_loss_kwargs: 
      omega: 10.0
      eps: 2.0
    adaptive_wing_loss_kwargs: null
      # omega: 24
      # theta: 0.5
      # eps: 1.0
      # alpha: 2.1
  
  optim_kwargs:
    lr: 1.0e-2
    betas: [0.9, 0.999]
  
  sched_kwargs:
    milestones: [200, 800, 1500]
    gamma: 0.1
  
  logger_kwargs:
    filepath: /media/perukas/Data/Datasets/2d-datasets/CelebV-HQ/downloaded_celebvhq/processed/log.txt
    address: "127.0.0.1"
    port: 8097
    experiment_id: "Single video face parsing"
    log_iters: 500

matting_kwargs:
  script_path: /media/perukas/Home/_dev/_phd/libraries/RobustVideoMatting/inference.py
  chkp_path: /media/perukas/Home/_dev/_phd/libraries/RobustVideoMatting/weights/rvm_mobilenetv3.pth