update inferencer; add load_from_preprocessed_pts

This commit is contained in:
2024-09-20 19:00:08 +08:00
parent a621749cc9
commit 8517255245
7 changed files with 69 additions and 40 deletions

View File

@@ -7,9 +7,9 @@ runner:
parallel: False
experiment:
name: local_eval
name: debug
root_dir: "experiments"
use_checkpoint: True
use_checkpoint: False
epoch: 600 # -1 stands for last epoch
max_epochs: 5000
save_checkpoint_interval: 1
@@ -40,6 +40,7 @@ dataset:
batch_size: 1
num_workers: 12
pts_num: 4096
load_from_preprocess: True
OmniObject3d_test:
root_dir: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/scenes"
@@ -55,6 +56,7 @@ dataset:
batch_size: 1
num_workers: 12
pts_num: 4096
load_from_preprocess: True
pipeline:
nbv_reconstruction_pipeline:

View File

@@ -41,6 +41,7 @@ dataset:
batch_size: 160
num_workers: 16
pts_num: 4096
load_from_preprocess: True
OmniObject3d_test:
root_dir: "../data/sample_for_training/scenes"
@@ -56,6 +57,7 @@ dataset:
batch_size: 1
num_workers: 12
pts_num: 4096
load_from_preprocess: True
pipeline:
nbv_reconstruction_pipeline: