update strong pointnet++

This commit is contained in:
2025-03-13 14:48:15 +08:00
parent 5a03659112
commit b20fa8bb75
6 changed files with 81 additions and 8 deletions

View File

@@ -6,16 +6,16 @@ runner:
cuda_visible_devices: "0,1,2,3,4,5,6,7"
experiment:
name: train_ab_global_only_dense
name: train_ab_global_only_p++_wp
root_dir: "experiments"
epoch: 441 # -1 stands for last epoch
epoch: 922 # -1 stands for last epoch
test:
dataset_list:
- OmniObject3d_test
blender_script_path: "/media/hofee/data/project/python/nbv_reconstruction/blender/data_renderer.py"
output_dir: "/media/hofee/data/data/p++_dense"
output_dir: "/media/hofee/data/data/p++_wp"
pipeline: nbv_reconstruction_pipeline
voxel_size: 0.003
min_new_area: 1.0
@@ -62,7 +62,7 @@ pipeline:
module:
pointnet++_encoder:
in_dim: 3
params_name: dense
params_name: light
pointnet_encoder:
in_dim: 3

View File

@@ -0,0 +1,14 @@
runner:
general:
seed: 0
device: cuda
cuda_visible_devices: "0,1,2,3,4,5,6,7"
experiment:
name: simulation_debug
root_dir: "experiments"
simulation:
robot:
displaytable: