xfr/CBT_project
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
CBT_project/config/constant.py

38 lines
865 B
Python
Raw Normal View History

first commit
2026-04-10 05:25:27 +00:00
"""
LABEL_MAP: 一般脊椎與label對應的順序，若是自己標記的需要額外修改定義。
ALLOWED_DIAMETERS: 螺絲直徑範圍。
ALLOWED_LENGTHS: 螺絲長度範圍。
OVERLAP_THRESH: 圓柱體與骨頭接觸的比例（0-1），若低於這個閾值則會重跑。
spacing: 影像的spacing。
"""
LABEL_MAP = {
1: "C1", 2: "C2", 3: "C3", 4: "C4", 5: "C5", 6: "C6", 7: "C7",
8: "T1", 9: "T2", 10: "T3", 11: "T4", 12: "T5", 13: "T6", 14: "T7",
15: "T8", 16: "T9", 17: "T10", 18: "T11", 19: "T12",
20: "L1", 21: "L2", 22: "L3", 23: "L4", 24: "L5"
}
1. correct the bounding box and cortical mask 2. make the plot isometric 3. now it should work after tuning the parameters
2026-04-16 16:03:10 +00:00
ALLOWED_DIAMETERS = [
refactor(core): improve cylinder parameter snapping and optimization logic Refactor the cylinder parameter selection and optimization process to improve accuracy and robustness. - Implement `snap_to_discrete_values_xfr` using a KDTree for efficient mapping of continuous diameter and length values to a predefined set of discrete points. - Update `objective_function_xfr` to use the new snapping mechanism and introduce a weighted loss component for diameter and length. - Adjust PSO optimization bounds and search ranges in `run_pso_torch_xfr` to better align with image dimensions and anatomical constraints. - Refine scoring logic in `cl_score_torch_xfr` with updated penalty weights for overlaps and out-of-bone voxels. - Update `config/constant.py` with new allowed diameter and length ranges. - Improve `imaging/preprocessing.py` by making `PROGRESS_FILE` a parameter to allow per-output-directory progress tracking. - Update `xfr_debug.py` with improved error handling and directory paths for batch processing.
2026-08-26 14:36:07 +00:00
# 3.5,
# 4.0,
1. correct the bounding box and cortical mask 2. make the plot isometric 3. now it should work after tuning the parameters
2026-04-16 16:03:10 +00:00
4.5,
5.0,
refactor(core): improve cylinder parameter snapping and optimization logic Refactor the cylinder parameter selection and optimization process to improve accuracy and robustness. - Implement `snap_to_discrete_values_xfr` using a KDTree for efficient mapping of continuous diameter and length values to a predefined set of discrete points. - Update `objective_function_xfr` to use the new snapping mechanism and introduce a weighted loss component for diameter and length. - Adjust PSO optimization bounds and search ranges in `run_pso_torch_xfr` to better align with image dimensions and anatomical constraints. - Refine scoring logic in `cl_score_torch_xfr` with updated penalty weights for overlaps and out-of-bone voxels. - Update `config/constant.py` with new allowed diameter and length ranges. - Improve `imaging/preprocessing.py` by making `PROGRESS_FILE` a parameter to allow per-output-directory progress tracking. - Update `xfr_debug.py` with improved error handling and directory paths for batch processing.
2026-08-26 14:36:07 +00:00
5.5,
6.0,
6.5,
# 7.0,
# 7.5,
1. correct the bounding box and cortical mask 2. make the plot isometric 3. now it should work after tuning the parameters
2026-04-16 16:03:10 +00:00
]
ALLOWED_LENGTHS = [
refactor(imaging): improve orientation detection and segmentation robustness Refactor the preprocessing and segmentation pipeline to handle AP orientation variations and improve anatomical boundary detection. Key changes include: - Implement automated AP orientation detection in `process_single_image` to handle prone scans by flipping CT and labels when necessary. - Enhance `segment_spinous_process` using a gap-based approach to identify the spinal canal, providing more stable thresholds for spinous process and vertebral body segmentation. - Improve optimization search space by using the vertebral body (VBODY) projection for x/z bounding box calculation instead of the whole bone. - Refactor `render_bone_figure` to unify 2D/3D visualization and support detailed anatomical coloring (VBODY, spinous process). - Update `cl_score_torch_xfr` with more robust penalty handling for out-of-bone and null-voxel regions. - Add `retry_robust` utility to handle transient NFS file system errors. - Update `xfr_preprocess.py` to include anatomical segmentation coloring in rotated level visualizations.
2026-09-07 10:46:06 +00:00
# 25,
refactor(core): improve cylinder parameter snapping and optimization logic Refactor the cylinder parameter selection and optimization process to improve accuracy and robustness. - Implement `snap_to_discrete_values_xfr` using a KDTree for efficient mapping of continuous diameter and length values to a predefined set of discrete points. - Update `objective_function_xfr` to use the new snapping mechanism and introduce a weighted loss component for diameter and length. - Adjust PSO optimization bounds and search ranges in `run_pso_torch_xfr` to better align with image dimensions and anatomical constraints. - Refine scoring logic in `cl_score_torch_xfr` with updated penalty weights for overlaps and out-of-bone voxels. - Update `config/constant.py` with new allowed diameter and length ranges. - Improve `imaging/preprocessing.py` by making `PROGRESS_FILE` a parameter to allow per-output-directory progress tracking. - Update `xfr_debug.py` with improved error handling and directory paths for batch processing.
2026-08-26 14:36:07 +00:00
30,
1. correct the bounding box and cortical mask 2. make the plot isometric 3. now it should work after tuning the parameters
2026-04-16 16:03:10 +00:00
35,
40,
45,
refactor(core): improve cylinder parameter snapping and optimization logic Refactor the cylinder parameter selection and optimization process to improve accuracy and robustness. - Implement `snap_to_discrete_values_xfr` using a KDTree for efficient mapping of continuous diameter and length values to a predefined set of discrete points. - Update `objective_function_xfr` to use the new snapping mechanism and introduce a weighted loss component for diameter and length. - Adjust PSO optimization bounds and search ranges in `run_pso_torch_xfr` to better align with image dimensions and anatomical constraints. - Refine scoring logic in `cl_score_torch_xfr` with updated penalty weights for overlaps and out-of-bone voxels. - Update `config/constant.py` with new allowed diameter and length ranges. - Improve `imaging/preprocessing.py` by making `PROGRESS_FILE` a parameter to allow per-output-directory progress tracking. - Update `xfr_debug.py` with improved error handling and directory paths for batch processing.
2026-08-26 14:36:07 +00:00
# 50,
# 60,
# 70,
# 80,
1. correct the bounding box and cortical mask 2. make the plot isometric 3. now it should work after tuning the parameters
2026-04-16 16:03:10 +00:00
]
first commit
2026-04-10 05:25:27 +00:00
OVERLAP_THRESH = 0.50
DEFAULT_SPACING = [0.5, 0.5, 0.5]
Reference in a new issue Copy permalink