finish nbv_reconstruction_dataset

This commit is contained in:
hofee
2024-08-22 22:28:20 +08:00
parent 7cd1954a25
commit 954fed1122
5 changed files with 117 additions and 18 deletions

View File

@@ -4,17 +4,17 @@ runner:
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"
dataset_list:
- OmniObject3d
generate:
voxel_threshold: 0.005
overlap_threshold: 0.3
dataset_list:
- OmniObject3d
datasets:
OmniObject3d:

21
configs/train_config.yaml Normal file
View File

@@ -0,0 +1,21 @@
runner:
general:
seed: 0
device: cpu
cuda_visible_devices: "0,1,2,3,4,5,6,7"
experiment:
name: debug
root_dir: "experiments"
train:
dataset_list:
- OmniObject3d
datasets:
OmniObject3d:
root_dir: "C:\\Document\\Local Project\\nbv_rec\\sample_dataset"
label_dir: "C:\\Document\\Local Project\\nbv_rec\\sample_output"