config.yaml

# ── sequence-specific settings ───────────────────────────────────────────────
sequences:
  pants_C:
    image_dir: data/capture_pants_C_input
    output_root: data/capture_pants_C_output
    garments: [shirt, pants]
    start_frame: 1520
    end_frame: 1580
    camera_params_file: data/camera_params_pants_C.json
    recog_threshold: 0.98

  skirt_J:
    image_dir: data/capture_skirt_J_input
    output_root: data/capture_skirt_J_output
    garments: [skirt]
    start_frame: 2040
    end_frame: 2100
    camera_params_file: data/camera_params_skirt_J.json
    recog_threshold: 0.98

  shirt_S:
    image_dir: data/capture_shirt_S_input
    output_root: data/capture_shirt_S_output
    garments: [shirt]
    start_frame: 4140
    end_frame: 4200
    camera_params_file: data/camera_params_shirt_S.json

  suit_Z:
    image_dir: data/capture_suit_Z_input
    output_root: data/capture_suit_Z_output
    garments: [suit]
    start_frame: 1610
    end_frame: 1670
    camera_params_file: data/camera_params_suit_Z.json
    recog_threshold: 0.98

# ── pipeline defaults ─────────────────────────────────────────────────────────
# Shared settings applied to all sequences unless overridden above.

pipeline:
  # Camera IDs to use (one character per camera, up to 16).
  cameras: ABCDEFGHIJKLMNOP

  # CUDA device index. Use -1 for CPU.
  cuda: 0

  # Root output directory. All stage outputs are auto-derived from this:
  #   {output_root}/stage1_detection/    ← marker detection dumps + vis/
  #   {output_root}/stage2_recognition/  ← recognition dumps + vis/
  #   {output_root}/stage3_uv/           ← UV localization dumps + vis/
  #   {output_root}/stage4_mesh2d/       ← 2D mesh recovery vis/
  #   {output_root}/coords_2d/           ← triangulation inputs (per garment)
  #   {output_root}/coords_3d/           ← triangulation outputs (per garment)
  #   {output_root}/dense/               ← 3D reconstruction outputs (per garment)
  output_root: output

  detection_model_path: models/marker_detection.pth

  # Marker recognition acceptance threshold (gen_garment_labels). Override per
  # sequence below if a capture needs a looser/tighter cutoff.
  recog_threshold: 0.99999


# Per-garment model and data paths.
garments:
  suit:
    name: suit
    recognition_model_path: models/suit_recognition.pth
    uv_model_path: models/uv_localization.pth
    crop_info_path: data/suit/suit_crop_info.hdf5
    mesh_crop_info_path: data/suit/suit_mesh_crop_info.hdf5
    mesh_npz_path: data/suit/suit_pixel.npz
    marker_info_path: data/suit/suit_marker_info.hdf5
    mesh_neighbor_file: data/suit/suit_mesh_neighbor.npy
    inner_scale: 3
    outer_scale: 8
    markerW: 24
    markerH: 24

  shirt:
    name: shirt
    recognition_model_path: models/shirt_recognition.pth
    uv_model_path: models/uv_localization.pth
    crop_info_path: data/shirt/shirt_crop_info.hdf5
    mesh_crop_info_path: data/shirt/shirt_mesh_crop_info.hdf5
    mesh_npz_path: data/shirt/pixel_shirt.npz
    marker_info_path: data/shirt/shirt_marker_info.hdf5
    mesh_neighbor_file: data/shirt/shirt_mesh_neighbor.npy
    inner_scale: 3
    outer_scale: 8
    markerW: 24
    markerH: 24

  skirt:
    name: skirt
    recognition_model_path: models/skirt_recognition.pth
    uv_model_path: models/uv_localization.pth
    crop_info_path: data/skirt/skirt_crop_info.hdf5
    mesh_crop_info_path: data/skirt/skirt_mesh_crop_info.hdf5
    mesh_npz_path: data/skirt/pixel_skirt.npz
    marker_info_path: data/skirt/skirt_marker_info.hdf5
    mesh_neighbor_file: data/skirt/skirt_mesh_neighbor.npy
    inner_scale: 3
    outer_scale: 8
    markerW: 24
    markerH: 24

  pants:
    name: pants
    recognition_model_path: models/pants_recognition.pth
    uv_model_path: models/uv_localization.pth
    crop_info_path: data/pants/pants_crop_info.hdf5
    mesh_crop_info_path: data/pants/pants_mesh_crop_info.hdf5
    mesh_npz_path: data/pants/pixel_pants.npz
    marker_info_path: data/pants/pants_marker_info.hdf5
    mesh_neighbor_file: data/pants/pants_mesh_neighbor.npy
    inner_scale: 4
    outer_scale: 4
    markerW: 24
    markerH: 24


# Triangulation settings.
triangulation:
  thres2: 0.5
  thres3: 1.0


# Per-sequence 3D reconstruction settings.
# Each sequence maps garment names to their multigrid config.
# Each garment runs a multigrid chain: coarse levels first, then fine.
# Public level "1" uses the existing coarse mesh and mapping files.
# Upscaling between levels is automatic: {garment_data_dir}/{garment}_coarse{N}_to_coarse{M}.npz
# n_smooth: number of temporal-smoothing passes after the base (non-smooth) pass per level.
recover_3d:
  suit_Z:
    suit:
      garment: suit
      garment_data_dir: data/suit
      mesh_crop_info_path: data/suit/suit_mesh_crop_info.hdf5
      smooth: true
      levels:
        "1":
          rest_shape_file: data/suit/suit_coarse.obj
          init_file: data/suit/suit_init.npy
          color_file: data/suit/suit_coarse_colors.npy
          n_iterations: 10
          n_frames: 500
          min_n_frames: 100
          part_size: 15
          min_threshold: 0.2
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 3.0e-1
          mass_scale: 1.0e-7
          n_smooth: 5
        "3":
          rest_shape_file: data/suit/suit_coarse3.obj
          init_from_level: "1"
          init_interp_file: data/suit/suit_coarse3_to_coarse.npz
          color_file: data/suit/suit_coarse3_colors.npy
          n_iterations: 20
          n_frames: 200
          min_n_frames: 50
          part_size: 50
          min_threshold: 0.1
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 3.0e-1
          mass_scale: 1.0e-7
          n_smooth: 3
        "6":
          rest_shape_file: data/suit/suit_coarse6.obj
          init_from_level: "3"
          init_interp_file: data/suit/suit_coarse6_to_coarse3.npz
          color_file: data/suit/suit_coarse6_colors.npy
          n_iterations: 30
          n_frames: 20
          min_n_frames: 5
          part_size: 200
          min_threshold: 0.05
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          n_smooth: 1

  pants_C:
    shirt:
      garment: shirt
      garment_data_dir: data/shirt
      mesh_crop_info_path: data/shirt/shirt_mesh_crop_info.hdf5
      smooth: true
      levels:
        "1":
          rest_shape_file: data/shirt/shirt_coarse.obj
          init_file: data/shirt/shirt_init.npy
          color_file: data/shirt/shirt_coarse_colors.npy
          n_iterations: 10
          n_frames: 500
          min_n_frames: 100
          part_size: 15
          min_threshold: 0.2
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          n_smooth: 5
        "3":
          rest_shape_file: data/shirt/shirt_coarse3.obj
          init_from_level: "1"
          init_interp_file: data/shirt/shirt_coarse3_to_coarse.npz
          color_file: data/shirt/shirt_coarse3_colors.npy
          n_iterations: 20
          n_frames: 100
          min_n_frames: 30
          part_size: 100
          min_threshold: 0.1
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          n_smooth: 3
        "6":
          rest_shape_file: data/shirt/shirt_coarse6.obj
          init_from_level: "3"
          init_interp_file: data/shirt/shirt_coarse6_to_coarse3.npz
          color_file: data/shirt/shirt_coarse6_colors.npy
          n_iterations: 30
          n_frames: 20
          min_n_frames: 5
          part_size: 400
          min_threshold: 0.05
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          n_smooth: 1
        "12":
          rest_shape_file: data/shirt/pixel_shirt.npz
          init_from_level: "6"
          init_interp_file: data/shirt/shirt_coarse12_to_coarse6.npz
          color_file: data/shirt/shirt_coarse12_colors.npy
          n_iterations: 50
          n_frames: 1
          min_n_frames: 1
          part_size: 1200
          min_threshold: 0.02
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e0
          particle_self_contact_radius: 1.0e-4
          particle_self_contact_margin: 2.0e-4
          n_smooth: 0
    pants:
      garment: pants
      garment_data_dir: data/pants
      mesh_crop_info_path: data/pants/pants_mesh_crop_info.hdf5
      smooth: true
      levels:
        "1":
          rest_shape_file: data/pants/pants_coarse.obj
          init_file: data/pants/pants_init.npy
          color_file: data/pants/pants_coarse_colors.npy
          n_iterations: 10
          n_frames: 500
          min_n_frames: 100
          part_size: 15
          min_threshold: 0.2
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          n_smooth: 5
        "3":
          rest_shape_file: data/pants/pants_coarse3.obj
          init_from_level: "1"
          init_interp_file: data/pants/pants_coarse3_to_coarse.npz
          color_file: data/pants/pants_coarse3_colors.npy
          n_iterations: 20
          n_frames: 100
          min_n_frames: 30
          part_size: 100
          min_threshold: 0.1
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          n_smooth: 3
        "6":
          rest_shape_file: data/pants/pants_coarse6.obj
          init_from_level: "3"
          init_interp_file: data/pants/pants_coarse6_to_coarse3.npz
          color_file: data/pants/pants_coarse6_colors.npy
          n_iterations: 30
          n_frames: 20
          min_n_frames: 5
          part_size: 200
          min_threshold: 0.05
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          n_smooth: 1
        "12":
          rest_shape_file: data/pants/pixel_pants.npz
          init_from_level: "6"
          init_interp_file: data/pants/pants_coarse12_to_coarse6.npz
          color_file: data/pants/pants_coarse12_colors.npy
          n_iterations: 50
          n_frames: 1
          min_n_frames: 1
          part_size: 600
          min_threshold: 0.02
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e0
          particle_self_contact_radius: 1.0e-4
          particle_self_contact_margin: 2.0e-4
          n_smooth: 0

  skirt_J:
    skirt:
      garment: skirt
      garment_data_dir: data/skirt
      mesh_crop_info_path: data/skirt/skirt_mesh_crop_info.hdf5
      smooth: true
      levels:
        "1":
          rest_shape_file: data/skirt/skirt_coarse.obj
          init_file: data/skirt/skirt_init_02040.npy
          color_file: data/skirt/skirt_coarse_colors.npy
          n_iterations: 10
          n_frames: 500
          min_n_frames: 100
          part_size: 15
          min_threshold: 0.2
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 3.0e-2
          mass_scale: 1.0e-7
          n_smooth: 5
        "3":
          rest_shape_file: data/skirt/skirt_coarse3.obj
          init_from_level: "1"
          init_interp_files:
            - data/skirt/skirt_legacy_coarse_to_coarse.npz
            - data/skirt/skirt_coarse3_to_legacy_coarse.npz
          color_file: data/skirt/skirt_coarse3_colors.npy
          n_iterations: 20
          n_frames: 100
          min_n_frames: 30
          part_size: 100
          min_threshold: 0.1
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          n_smooth: 3
        "6":
          rest_shape_file: data/skirt/skirt_coarse6.npz
          init_from_level: "3"
          init_interp_file: data/skirt/skirt_coarse6_to_coarse3.npz
          color_file: data/skirt/skirt_coarse6_colors.npy
          n_iterations: 30
          n_frames: 20
          min_n_frames: 5
          part_size: 200
          min_threshold: 0.05
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          n_smooth: 1

  shirt_S:
    shirt:
      garment: shirt
      garment_data_dir: data/shirt
      mesh_crop_info_path: data/shirt/shirt_mesh_crop_info.hdf5
      smooth: true
      levels:
        "1":
          rest_shape_file: data/shirt/shirt_coarse.obj
          init_file: data/shirt/shirt_init2.npy
          color_file: data/shirt/shirt_coarse_colors.npy
          n_iterations: 10
          n_frames: 500
          min_n_frames: 100
          part_size: 15
          min_threshold: 0.2
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 3.0e-2
          mass_scale: 1.0e-7
          n_smooth: 5
        "3":
          rest_shape_file: data/shirt/shirt_coarse3.obj
          init_from_level: "1"
          init_interp_file: data/shirt/shirt_coarse3_to_coarse.npz
          color_file: data/shirt/shirt_coarse3_colors.npy
          n_iterations: 20
          n_frames: 100
          min_n_frames: 30
          part_size: 100
          min_threshold: 0.1
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          n_smooth: 3
        "6":
          rest_shape_file: data/shirt/shirt_coarse6.obj
          init_from_level: "3"
          init_interp_file: data/shirt/shirt_coarse6_to_coarse3.npz
          color_file: data/shirt/shirt_coarse6_colors.npy
          n_iterations: 30
          n_frames: 20
          min_n_frames: 5
          part_size: 400
          min_threshold: 0.05
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          n_smooth: 1
        "12":
          rest_shape_file: data/shirt/pixel_shirt.npz
          init_from_level: "6"
          init_interp_file: data/shirt/shirt_coarse12_to_coarse6.npz
          color_file: data/shirt/shirt_coarse12_colors.npy
          n_iterations: 50
          n_frames: 10
          min_n_frames: 3
          part_size: 1200
          min_threshold: 0.05
          density: 1.0e6
          tri_ke: 1.0e2
          tri_ka: 1.0e1
          edge_ke: 1.0e-1
          mass_scale: 1.0e-7
          particle_self_contact_radius: 3.0e-4
          particle_self_contact_margin: 6.0e-4
          n_smooth: 0