runner:
  general:
    seed: 0
    device: cuda
    cuda_visible_devices: "0,1,2,3,4,5,6,7"
    
  experiment:
    name: train_ab_global_only_p++_wp
    root_dir: "experiments"
    epoch: 922 # -1 stands for last epoch

  test:
    dataset_list:
      - OmniObject3d_test

  blender_script_path: "/media/hofee/data/project/python/nbv_reconstruction/blender/data_renderer.py"
  output_dir: "/media/hofee/data/data/p++_wp"
  pipeline: nbv_reconstruction_pipeline
  voxel_size: 0.003
  min_new_area: 1.0
dataset:
  # OmniObject3d_train:
  #   root_dir: "C:\\Document\\Datasets\\inference_test1"
  #   model_dir: "C:\\Document\\Datasets\\scaled_object_meshes"
  #   source: seq_reconstruction_dataset_preprocessed
  #   split_file: "C:\\Document\\Datasets\\data_list\\sample.txt"
  #   type: test
  #   filter_degree: 75
  #   ratio: 1
  #   batch_size: 1
  #   num_workers: 12
  #   pts_num: 8192
  #   load_from_preprocess: True

  OmniObject3d_test:
    root_dir: "/media/hofee/data/data/new_testset_output"
    model_dir: "/media/hofee/data/data/scaled_object_meshes"
    source: seq_reconstruction_dataset_preprocessed
    # split_file: "C:\\Document\\Datasets\\data_list\\OmniObject3d_test.txt"
    type: test
    filter_degree: 75
    eval_list:
      - pose_diff
      - coverage_rate_increase
    ratio: 0.1
    batch_size: 1
    num_workers: 12
    pts_num: 8192
    load_from_preprocess: True

pipeline:
  nbv_reconstruction_pipeline:
    modules:
      pts_encoder: pointnet++_encoder
      seq_encoder: transformer_seq_encoder
      pose_encoder: pose_encoder
      view_finder: gf_view_finder
    eps: 1e-5
    global_scanned_feat: True

module:
  pointnet++_encoder:
    in_dim: 3
    params_name: light

  pointnet_encoder:
    in_dim: 3
    out_dim: 1024
    global_feat: True
    feature_transform: False
  transformer_seq_encoder:
    embed_dim: 256
    num_heads: 4
    ffn_dim: 256
    num_layers: 3
    output_dim: 1024

  gf_view_finder:
    t_feat_dim: 128
    pose_feat_dim: 256
    main_feat_dim: 2048
    regression_head: Rx_Ry_and_T
    pose_mode: rot_matrix
    per_point_feature: False
    sample_mode: ode
    sampling_steps: 500
    sde_mode: ve
  pose_encoder:
    pose_dim: 9
    out_dim: 256
  pts_num_encoder:
    out_dim: 64