39 lines
1.1 KiB
YAML
39 lines
1.1 KiB
YAML
# 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
|