CBT_project/config/constant.py

18 lines
710 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]
OVERLAP_THRESH = 0.50
DEFAULT_SPACING = [0.5, 0.5, 0.5]