add split dataset

This commit is contained in:
hofee
2024-09-02 18:21:38 +08:00
parent f58360c0c0
commit 2fcfcd1966
5 changed files with 111 additions and 9 deletions

View File

@@ -0,0 +1,22 @@
runner:
general:
seed: 0
device: cpu
cuda_visible_devices: "0,1,2,3,4,5,6,7"
experiment:
name: debug
root_dir: "experiments"
split:
root_dir: "C:\\Document\\Local Project\\nbv_rec\\data\\sample"
type: "unseen_instance" # "unseen_category"
datasets:
OmniObject3d_train:
path: "C:\\Document\\Local Project\\nbv_rec\\data\\OmniObject3d_train.txt"
ratio: 0.5
OmniObject3d_test:
path: "C:\\Document\\Local Project\\nbv_rec\\data\\OmniObject3d_test.txt"
ratio: 0.5

View File

@@ -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"