# configs/infer_1photon.yaml
experiment:
  name: "FlatField_3filters_pos0"  # options: SiemenStarLowerLeft, SiemenStarLowerRight
  task: "1Photon" 
  output_base: "./InferenceResults"

data:
  sample_folder: "/home/xie_x1/MLXID/DataProcess/Samples"
  names:
    SiemenStarLowerLeft:
      file_pattern: "SiemenStarLowerLeft/1Photon_CS3_chunk{}.h5" ### 15 keV
      NX: 400
      NY: 400
      file_range: [0, 1]  # [0, 160)
      roi: [140, 230, 120, 210]  # [x_min, x_max, y_min, y_max]
    SiemenStarLowerRight:
      file_pattern: "SiemenStarLowerRight/1Photon_CS3_chunk{}.h5" ### 15 keV
      NX: 400
      NY: 400
      file_range: [0, 320] # [0, 320)
      roi: [235, 345, 110, 220]
    KnifeEdge_3filters_pos0:
      file_pattern: "2603MaxIV_Edge3Filters_pos0_12keV/1Photon_CS3_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
    FlatField_3filters_pos0:
      file_pattern: "2603MaxIV_FlatField3Filters_pos0_12keV/1Photon_CS3_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 48] # [0, 48)
      roi: [0, 101, 0, 101]
  
  energy: 12  # keV
  normalize: false # for inference dataloaders
  batch_size: 8192
  num_workers: 16

model:
  version: "251022"
  base_dir: "/home/xie_x1/MLXID/DeepLearning/Results/"
  experiment_name: "260408_1ph_12keV_v251022_03"
  name: "singlePhoton251022_12keV_Noise0.13keV_E150_aug1.pth"
  noise_keV: 0.13

inference:
  binning_factor: 10
  chunk_size: 16  # 
  num_aug_ops: 8  # TTA: test-time augmentation