after first overfit test

This commit is contained in:
2024-09-18 06:49:59 +00:00
parent d80d0ea79d
commit 0280dc7292
6 changed files with 193 additions and 42 deletions

View File

@@ -14,8 +14,8 @@ runner:
voxel_threshold: 0.01
overlap_threshold: 0.5
filter_degree: 75
to_specified_dir: True # if True, output_dir is used, otherwise, root_dir is used
save_points: False
to_specified_dir: False # if True, output_dir is used, otherwise, root_dir is used
save_points: True
save_best_combined_points: True
save_mesh: True
overwrite: False

View File

@@ -1,18 +1,18 @@
runner:
general:
seed: 0
seed: 1
device: cuda
cuda_visible_devices: "0,1,2,3,4,5,6,7"
parallel: False
experiment:
name: test_overfit
name: new_test_overfit_2
root_dir: "experiments"
use_checkpoint: False
epoch: -1 # -1 stands for last epoch
max_epochs: 5000
save_checkpoint_interval: 1
save_checkpoint_interval: 3
test_first: False
train:
@@ -32,22 +32,29 @@ runner:
dataset:
OmniObject3d_train:
root_dir: "../data/sample_for_training/scenes"
model_dir: "../data/scaled_object_meshes"
source: nbv_reconstruction_dataset
split_file: "../data/sample_for_training/OmniObject3d_train.txt"
ratio: 1.0
batch_size: 1
type: train
cache: True
ratio: 1
batch_size: 128
num_workers: 12
pts_num: 4096
OmniObject3d_test:
root_dir: "../data/sample_for_training/scenes"
model_dir: "../data/scaled_object_meshes"
source: nbv_reconstruction_dataset
split_file: "../data/sample_for_training/OmniObject3d_train.txt"
type: test
cache: True
filter_degree: 75
eval_list:
- pose_diff
ratio: 0.1
batch_size: 1
num_workers: 1
num_workers: 12
pts_num: 4096
pipeline:
@@ -92,4 +99,6 @@ loss_function:
gf_loss:
evaluation_method:
pose_diff:
pose_diff:
coverage_rate_increase:
renderer_path: "../blender/data_renderer.py"