From 61f6b94575c6fe2730d1af51d4c593968d7b0a44 Mon Sep 17 00:00:00 2001 From: "xiangyu.xie" Date: Fri, 20 Mar 2026 16:24:15 +0100 Subject: [PATCH] Add infer yaml --- Configs/infer_1photon.yaml | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 Configs/infer_1photon.yaml diff --git a/Configs/infer_1photon.yaml b/Configs/infer_1photon.yaml new file mode 100644 index 0000000..a04bc03 --- /dev/null +++ b/Configs/infer_1photon.yaml @@ -0,0 +1,38 @@ +# configs/infer_1photon.yaml +experiment: + name: "SiemenStarLowerLeft" # 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" + 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" + NX: 400 + NY: 400 + file_range: [0, 320] # [0, 320) + roi: [235, 345, 110, 220] + + energy: 15 # 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: "260320_1ph_15keV_v251022_01" + name: "singlePhoton251022_15keV_Noise0.13keV_E15_aug1_normalized.pth" + noise_keV: 0.13 + +inference: + binning_factor: 10 + chunk_size: 16 # + num_aug_ops: 8 # TTA: test-time augmentation