fix bug for training

configs/split_dataset_config.yaml

@@ -10,13 +10,13 @@ runner:
     root_dir: "experiments"
   
   split:
-    root_dir: "C:\\Document\\Local Project\\nbv_rec\\data\\sample"
+    root_dir: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/scenes"
     type: "unseen_instance" # "unseen_category"
     datasets:
       OmniObject3d_train: 
-        path: "C:\\Document\\Local Project\\nbv_rec\\data\\OmniObject3d_train.txt"
-        ratio: 0.5
+        path: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/OmniObject3d_train.txt"
+        ratio: 0.9
 
       OmniObject3d_test: 
-        path: "C:\\Document\\Local Project\\nbv_rec\\data\\OmniObject3d_test.txt"
-        ratio: 0.5
+        path: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/OmniObject3d_test.txt"
+        ratio: 0.1

configs/strategy_generate_config.yaml

@@ -18,12 +18,14 @@ runner:
     save_points: False
     save_best_combined_points: True
     save_mesh: True
+    overwrite: False
     dataset_list:
       - OmniObject3d
   
 datasets:
     OmniObject3d:
-      root_dir: "/media/hofee/data/project/python/nbv_reconstruction/nbv_rec_visualize/data/sample"
+      #"/media/hofee/data/data/temp_output"
+      root_dir: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/scenes" 
       model_dir: "/media/hofee/data/data/scaled_object_meshes"
       #output_dir: "/media/hofee/data/data/label_output"
 

configs/train_config.yaml

@@ -2,15 +2,16 @@
 runner:
   general:
     seed: 0
-    device: cpu
+    device: cuda
     cuda_visible_devices: "0,1,2,3,4,5,6,7"
+    parallel: False
     
   experiment:
-    name: debug
+    name: test_overfit
     root_dir: "experiments"
     use_checkpoint: False
     epoch: -1 # -1 stands for last epoch
-    max_epochs: 5
+    max_epochs: 5000
     save_checkpoint_interval: 1
     test_first: False  
 
@@ -19,34 +20,43 @@ runner:
       type: Adam
       lr: 0.0001
     losses: 
-      - mse_loss
+      - gf_loss
     dataset: OmniObject3d_train
   test:
     frequency: 3 # test frequency
     dataset_list:
-      - OmniObject3d_train
+      - OmniObject3d_test
 
   pipeline: nbv_reconstruction_pipeline
   
-datasets:
+dataset:
   OmniObject3d_train:
-    root_dir: "C:\\Document\\Local Project\\nbv_rec\\data\\sample"
-    split_file: "C:\\Document\\Local Project\\nbv_rec\\data\\OmniObject3d_train.txt"
+    root_dir: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/scenes"
+    source: nbv_reconstruction_dataset
+    split_file: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/OmniObject3d_train.txt"
     ratio: 1.0
     batch_size: 1
     num_workers: 12
-    pts_num: 2048
+    pts_num: 4096
 
   OmniObject3d_test:
-    root_dir: "C:\\Document\\Local Project\\nbv_rec\\data\\sample"
-    split_file: "C:\\Document\\Local Project\\nbv_rec\\data\\OmniObject3d_test.txt"
+    root_dir: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/scenes"
+    source: nbv_reconstruction_dataset
+    split_file: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/OmniObject3d_train.txt"
     eval_list:
       - pose_diff
-    ratio: 1.0
+    ratio: 0.1
     batch_size: 1
     num_workers: 1
-    pts_num: 2048
+    pts_num: 4096
 
+pipeline:
+  nbv_reconstruction_pipeline:
+    pts_encoder: pointnet_encoder
+    seq_encoder: transformer_seq_encoder
+    pose_encoder: pose_encoder
+    view_finder: gf_view_finder
+    
 module:
 
   pointnet_encoder:
@@ -58,13 +68,15 @@ module:
   transformer_seq_encoder:
     pts_embed_dim: 1024
     pose_embed_dim: 256
-    num_heads: 4
-    ffn_dim: 256
-    num_layers: 3
-    max_seq_len: 30
-    output_dim: 2048
+    num_heads: 2 # 4
+    ffn_dim: 128 # 256
+    num_layers: 2 # 3
+    output_dim: 1024 # 2048
 
   gf_view_finder:
+    t_feat_dim: 128
+    pose_feat_dim: 256
+    main_feat_dim: 1024 # 2048
     regression_head: Rx_Ry_and_T
     pose_mode: rot_matrix
     per_point_feature: False
@@ -74,4 +86,10 @@ module:
 
   pose_encoder:
     pose_dim: 9
-    output_dim: 256
+    out_dim: 256
+
+loss_function:
+  gf_loss:
+
+evaluation_method:
+  pose_diff:

configs/view_generate_config.yaml

@@ -9,10 +9,10 @@ runner:
   generate:
     object_dir: /media/hofee/data/data/scaled_object_meshes
     table_model_path: /media/hofee/data/data/others/table.obj
-    output_dir: /media/hofee/data/data/temp_output
+    output_dir: /media/hofee/repository/nbv_reconstruction_data_512
     binocular_vision: true
     plane_size: 10
-    max_views: 256
+    max_views: 512
     min_views: 64
     max_diag: 0.7
     min_diag: 0.1