51 lines
1.5 KiB
YAML
51 lines
1.5 KiB
YAML
# 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
|