finish generate_sequence.py

2024-08-22 20:27:21 +08:00
parent ff3e89b17c
commit 7cd1954a25
7 changed files with 121 additions and 57 deletions
--- a/configs/generate_config.yaml
+++ b/configs/generate_config.yaml
@@ -1,23 +1,24 @@

-runners:
+runner:
  general:
    seed: 0
    device: cpu
    cuda_visible_devices: "0,1,2,3,4,5,6,7"
    
+  generate:
+    voxel_threshold: 0.005
+    overlap_threshold: 0.3
+    
  experiment:
    name: debug
    root_dir: "experiments"
  
-  generate:
-    - name: OmniObject3d_train
-      component: OmniObject3d
-      data_type: train
+  dataset_list:
+    - OmniObject3d
  
 datasets:
-  general:
-  components:
    OmniObject3d:
-      root_dir: "C:\\Document\\Local Project\\nbv_rec\\output"
+      root_dir: "C:\\Document\\Local Project\\nbv_rec\\sample_dataset"
+      output_dir: "C:\\Document\\Local Project\\nbv_rec\\sample_output"