Switch viewer and training to patient1 fixed flat CT

This commit is contained in:
admin
2026-06-03 10:48:14 +08:00
parent 972fb2435c
commit 0a6a0ece00
5 changed files with 566 additions and 46 deletions

View File

@@ -10,8 +10,8 @@ from pathlib import Path
PROJECT_ROOT = Path(__file__).resolve().parent
DATA_ROOT = PROJECT_ROOT / "Data"
DEFAULT_MOVING_DICOM_DIR = DATA_ROOT / "患者1-平扫CT"
DEFAULT_FIXED_DICOM_DIR = DATA_ROOT / "患者1-仰头CT"
DEFAULT_FIXED_DICOM_DIR = DATA_ROOT / "患者1-平扫CT"
DEFAULT_MOVING_DICOM_DIR = DATA_ROOT / "患者1-仰头CT"
OUTPUT_ROOT = PROJECT_ROOT / "outputs"
NIFTI_DIR = OUTPUT_ROOT / "nifti"
@@ -21,10 +21,10 @@ INFERENCE_DIR = OUTPUT_ROOT / "inference"
DEFAULT_MOVING_NIFTI = PREPROCESSED_DIR / "patient1_moving_preprocessed.nii.gz"
DEFAULT_FIXED_NIFTI = PREPROCESSED_DIR / "patient1_fixed_preprocessed.nii.gz"
DEFAULT_CHECKPOINT = CHECKPOINT_DIR / "vxm_head_ct.pt"
DEFAULT_CHECKPOINT = CHECKPOINT_DIR / "vxm_head_ct_patient1.pt"
# VoxelMorph 的 3D U-Net 多次下采样,三维尺寸建议均为 16 的倍数。
DEFAULT_TARGET_SHAPE = (160, 192, 224) # NIfTI 轴顺序: X, Y, Z
DEFAULT_TARGET_SHAPE = (256, 256, 352) # NIfTI 轴顺序: X, Y, Z
DEFAULT_TARGET_SPACING = (1.0, 1.0, 1.0) # mm, X/Y/Z
# 颈部软组织/气道观察常用窗口W=400, L=40。