| # ============================================================================= | |
| # ERPT Pipeline Configuration（移植自原版 ERPT） | |
| # 严格遵循 ERPT_native 坐标系约定：右手系 [X右, Y上, Z前] | |
| # ============================================================================= | |
| # --- 数据路径 --- | |
| data: | |
| data_dir: "inputs" | |
| output_dir: "outputs" | |
| depth_dir: null # 可选：外部深度目录 | |
| # --- ERP 参数 --- | |
| erp: | |
| auto_size: true # 自动从图像检测尺寸 | |
| width: 4096 # 参考宽度 | |
| height: 2048 # 参考高度 | |
| # --- Tangent 切片参数（原版配置） --- | |
| tangent: | |
| scheme: "icosahedron" | |
| num_faces: 20 | |
| add_poles: true | |
| face_resolution: 768 # 每个 face 的分辨率（像素） | |
| fov_deg: 90.0 # 普通 face 的基础 FOV（度） | |
| padding_factor: 1.3 # 有效 FOV = 90 * 1.3 = 117° | |
| pole_fov_deg: 160.0 # 极区切片使用更大 FOV | |
| pole_resolution: 768 # 极区分辨率 | |
| pole_extra_rings: 2 # 额外极区密采样环数 | |
| seam_wrap: true | |
| # --- Depth Pro 参数（原版配置） --- | |
| depth_pro: | |
| enabled: true | |
| repo_dir: "third_party/ml-depth-pro" | |
| checkpoint_path: "third_party/ml-depth-pro/checkpoints/depth_pro.pt" | |
| precision: "fp16" # "fp32" | "fp16" | "bf16" | |
| depth_def: "z" # "z" (z-depth) | "ray" (ray-depth) | |
| pass_f_px: true # 传递已知焦距 | |
| # --- 深度融合参数（原版配置） --- | |
| fusion: | |
| blend_mode: "multiband" # "softmin_invdepth" | "multiband" | |
| output_scale: 1.10 # 全局尺度校正 | |
| # 权重模式 | |
| weight_mode: "cosine" | |
| k: 4 # cosine 权重指数 | |
| # 深度竞争 | |
| depth_competition: "softmin_invdepth" | |
| softmin_alpha: 10.0 | |
| # 极区处理 | |
| pole_boost: false | |
| pole_boost_factor: 1.5 | |
| pole_latitude_deg: 75.0 | |
| pole_ramp_deg: 10.0 | |
| pole_ring: | |
| enabled: false | |
| min_latitude_deg: 60.0 | |
| ramp_deg: 10.0 | |
| face_pole_suppress: | |
| enabled: false | |
| min_latitude_deg: 70.0 | |
| ramp_deg: 10.0 | |
| min_scale: 0.4 | |
| # 极区一致性校正 | |
| pole_consistency: | |
| enabled: true | |
| min_latitude_deg: 60.0 | |
| min_overlap_pixels: 4000 | |
| max_abs_log_shift: 0.7 | |
| ref_slice_types: ["face", "pole_ring"] | |
| target_slice_types: ["pole_north", "pole_south"] | |
| # Z-buffer 门限 | |
| project_zbuffer_eps_abs_m: 0.02 | |
| project_zbuffer_eps_rel: 0.02 | |
| # Multiband 金字塔 | |
| multiband: | |
| levels: 6 | |
| highfreq_levels: 2 | |
| eps: 1.0e-6 | |
| # 有效性 | |
| min_weight_sum: 1.0e-6 | |
| # --- Warp 参数（原版配置） --- | |
| warp: | |
| enabled: true | |
| center_frame: 0 | |
| target_frames: "auto" # "auto" 自动识别所有非中心帧；或指定列表如 [1, 2, 3] | |
| # Splatting 方法 | |
| method: "softmax_splatting" # "softmax_splatting" | "zbuffer_splatting" | "zbuffer_point" | |
| alpha: 2.0 # softmax 温度 | |
| # 自适应 Splat 半径 | |
| splat_radius_px: 1.5 # 基础半径 | |
| radius_min_px: 0.6 # 半径下限 | |
| radius_max_px: 2.2 # 中纬度上限 | |
| radius_max_pole_px: 3.4 # 极区上限 | |
| pole_radius_scale: 3.0 # 极区放大因子 | |
| pole_lat_threshold: 60.0 # 极区纬度阈值（度） | |
| depth_radius_scale: false # 深度缩放 | |
| depth_ref_m: 2.0 # 深度参考值 | |
| depth_scale_factor: 1.0 # 深度已在 fusion.output_scale 烘焙 | |
| depth_edge_aware: true # 深度边缘感知 | |
| depth_edge_threshold: 0.3 # 深度梯度阈值 | |
| depth_edge_min_scale: 0.12 # 边缘处最小半径缩放 | |
| # 遮挡门控 | |
| occlusion_gate: | |
| enabled: true | |
| abs_eps_m: 0.05 | |
| rel_eps: 0.05 | |
| # Z-buffer 参数 | |
| zbuffer_eps_abs_m: 0.03 | |
| zbuffer_eps_rel: 0.03 | |
| zbuffer_min_weight: 0.001 | |
| # 空洞填充 | |
| hole_fill_enabled: false | |
| max_hole_px: 16 | |
| # 有效性 | |
| min_weight_sum: 1.0e-4 | |
| min_hit_sum: 1.0e-6 | |
| # 输出控制 | |
| output_flow: true | |
| output_depth: true | |
| # --- 运行参数 --- | |
| run: | |
| device: "cuda" | |
| save_intermediates: true | |