add split dataset
This commit is contained in:
22
configs/split_dataset_config.yaml
Normal file
22
configs/split_dataset_config.yaml
Normal 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
|
@@ -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"
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user