44 lines
917 B
YAML
44 lines
917 B
YAML
|
|
||
|
|
runner:
|
||
|
|
general:
|
||
|
|
seed: 0
|
||
|
|
device: cuda
|
||
|
|
cuda_visible_devices: "0,1,2,3,4,5,6,7"
|
||
|
|
|
||
|
|
experiment:
|
||
|
|
name: train_ab_global_only
|
||
|
|
root_dir: "experiments"
|
||
|
|
epoch: -1 # -1 stands for last epoch
|
||
|
|
|
||
|
|
pipeline: nbv_reconstruction_pipeline
|
||
|
|
voxel_size: 0.003
|
||
|
|
enable_cluster: False
|
||
|
|
heuristic_method: hemisphere_random
|
||
|
|
|
||
|
|
pipeline:
|
||
|
|
nbv_reconstruction_pipeline:
|
||
|
|
modules:
|
||
|
|
pts_encoder: pointnet_encoder
|
||
|
|
seq_encoder: transformer_seq_encoder
|
||
|
|
pose_encoder: pose_encoder
|
||
|
|
view_finder: gf_view_finder
|
||
|
|
eps: 1e-5
|
||
|
|
global_scanned_feat: True
|
||
|
|
|
||
|
|
heuristic_methods:
|
||
|
|
hemisphere_random:
|
||
|
|
center: [0, 0, 0]
|
||
|
|
radius_fixed: True
|
||
|
|
fixed_radius: 0.6
|
||
|
|
min_radius: 0.4
|
||
|
|
max_radius: 0.8
|
||
|
|
|
||
|
|
hemisphere_circle_trajectory:
|
||
|
|
center: [0, 0, 0]
|
||
|
|
radius_fixed: False
|
||
|
|
fixed_radius: 0.6
|
||
|
|
min_radius: 0.4
|
||
|
|
max_radius: 0.8
|
||
|
|
phi_list: [15, 45, 75]
|
||
|
|
circle_times: 12
|