2025-04-20 10:26:09 +08:00

52 lines
1.6 KiB
YAML

# 主动重建系统配置
# 基本设置
device: cuda # 使用的设备: cuda 或 cpu
output_dir: ./outputs/nbv_reconstruction # 输出目录
seed: 42 # 随机数种子
# 数据设置
data:
dataset_type: synthetic # 数据集类型: synthetic 或 real
synthetic_dir: ./data/synthetic/ # 合成数据目录
real_dir: ./data/real/ # 真实数据目录
# NeRF模型设置
nerf:
pos_enc_dim: 10 # 位置编码维度
dir_enc_dim: 4 # 方向编码维度
hidden_dim: 256 # 隐藏层维度(兼容旧配置)
# 网络结构设置
netdepth_coarse: 8 # coarse网络深度
netwidth_coarse: 256 # coarse网络宽度
netdepth_fine: 8 # fine网络深度
netwidth_fine: 256 # fine网络宽度
skips: [4] # 跳跃连接层
use_viewdirs: true # 是否使用视角方向信息
# 相机设置
camera:
width: 800 # 图像宽度
height: 800 # 图像高度
focal: 1000.0 # 焦距
near: 2.0 # 近平面距离
far: 6.0 # 远平面距离
# 采样设置
sampling:
coarse_samples: 64 # 粗采样点数
fine_samples: 128 # 精细采样点数
perturb: True # 是否添加噪声
# 重建设置
reconstruction:
max_iterations: 3 # 最大迭代次数
initial_view_count: 15 # 初始视图数量
epochs_per_iteration: 2000 # 每次迭代的训练轮数
mesh_resolution: 256 # 网格提取分辨率
# 视图选择策略设置
view_selection:
n_circles: 5 # 半球上的环数
n_poses_per_circle: 30 # 每个环上的位姿数
distance_threshold: 0.1 # 视图距离阈值