CBT_project/config/constant.py

28 lines
762 B
Python
Raw Normal View History

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"
}
ALLOWED_DIAMETERS = [
3.5,
4.0,
4.5,
5.0,
]
ALLOWED_LENGTHS = [
35,
40,
45,
50,
]
2026-04-10 05:25:27 +00:00
OVERLAP_THRESH = 0.50
DEFAULT_SPACING = [0.5, 0.5, 0.5]