add split dataset

2024-09-02 18:21:38 +08:00
parent f58360c0c0
commit 2fcfcd1966
5 changed files with 111 additions and 9 deletions
--- a/configs/train_config.yaml
+++ b/configs/train_config.yaml
@@ -11,10 +11,14 @@ runner:
  
  train:
    dataset_list:
-      - OmniObject3d
+      - OmniObject3d_train
  
 datasets:
-    OmniObject3d:
-      root_dir: "/media/hofee/data/data/nbv_rec/sample"
+  OmniObject3d_train:
+    root_dir: "C:\\Document\\Local Project\\nbv_rec\\data\\sample"
+    split_file: "C:\\Document\\Local Project\\nbv_rec\\data\\OmniObject3d_train.txt"
+
+  OmniObject3d_test:
+    root_dir: "C:\\Document\\Local Project\\nbv_rec\\data\\sample"
+    split_file: "C:\\Document\\Local Project\\nbv_rec\\data\\OmniObject3d_test.txt"

-