global: upd inference

configs/local/inference_config.yaml

@@ -6,71 +6,67 @@ runner:
     cuda_visible_devices: "0,1,2,3,4,5,6,7"
     
   experiment:
-    name: w_gf_wo_lf_full
+    name: overfit_ab_global_only
     root_dir: "experiments"
-    epoch: 1 # -1 stands for last epoch
+    epoch: -1 # -1 stands for last epoch
 
   test:
     dataset_list:
       - OmniObject3d_train
 
-  blender_script_path: "/media/hofee/data/project/python/nbv_reconstruction/blender/data_renderer.py"
-  output_dir: "/media/hofee/data/project/python/nbv_reconstruction/nbv_reconstruction/test/inference_global_full_on_testset"
-  pipeline: nbv_reconstruction_global_pts_pipeline
+  blender_script_path: "/data/hofee/project/nbv_rec/blender/data_renderer.py"
+  output_dir: "/data/hofee/data/inference_global_full_on_testset"
+  pipeline: nbv_reconstruction_pipeline
+  voxel_size: 0.003
   
 dataset:
   OmniObject3d_train:
-    root_dir: "/media/hofee/repository/nbv_reconstruction_data_512"
-    model_dir: "/media/hofee/data/data/scaled_object_meshes"
-    source: seq_nbv_reconstruction_dataset
-    split_file: "/media/hofee/data/project/python/nbv_reconstruction/nbv_reconstruction/test/test_set_list.txt"
+    root_dir: "/data/hofee/data/new_full_data"
+    model_dir: "/data/hofee/data/scaled_object_meshes"
+    source: seq_reconstruction_dataset
+    split_file: "/data/hofee/data/sample.txt"
     type: test
     filter_degree: 75
     ratio: 1
     batch_size: 1
     num_workers: 12
-    pts_num: 4096
-    load_from_preprocess: False
+    pts_num: 8192
+    load_from_preprocess: True
+
+  OmniObject3d_test:
+    root_dir: "/data/hofee/data/new_full_data"
+    model_dir: "/data/hofee/data/scaled_object_meshes"
+    source: seq_reconstruction_dataset
+    split_file: "/data/hofee/data/sample.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_local_pts_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: False
-
-  nbv_reconstruction_global_pts_pipeline:
-    modules:
-      pts_encoder: pointnet_encoder
-      pose_seq_encoder: transformer_pose_seq_encoder
-      pose_encoder: pose_encoder
-      view_finder: gf_view_finder
-    eps: 1e-5
     global_scanned_feat: True
 
-
-    
 module:
-
   pointnet_encoder:
     in_dim: 3
     out_dim: 1024
     global_feat: True
     feature_transform: False
-
   transformer_seq_encoder:
-    pts_embed_dim: 1024
-    pose_embed_dim: 256
-    num_heads: 4
-    ffn_dim: 256
-    num_layers: 3
-    output_dim: 2048
-
-  transformer_pose_seq_encoder:
-    pose_embed_dim: 256
+    embed_dim: 256
     num_heads: 4
     ffn_dim: 256
     num_layers: 3
@@ -86,7 +82,8 @@ module:
     sample_mode: ode
     sampling_steps: 500
     sde_mode: ve
-
   pose_encoder:
     pose_dim: 9
     out_dim: 256
+  pts_num_encoder:
+    out_dim: 64