config.yaml

# 超声提示多标签分类模型配置文件
# TransMIL + Query2Label Hybrid Model

data:
  # 【需要修改】数据根目录（包含 Report_XXX 文件夹的目录）
  data_root: "/path/to/your/ReportData_ROI/"
  # 【需要修改】多标签注释 CSV 文件路径
  annotation_csv: "/path/to/your/ReportData_ROI/thyroid_multilabel_annotations.csv"
  # 【需要修改】验证集 JSON 文件路径
  val_json: "/path/to/your/ReportData_ROI/classification_val_set_single.json"
  # 【需要修改】测试集 JSON 文件路径
  test_json: "/path/to/your/ReportData_ROI/classification_test_set_single.json"
  img_size: 224
  max_images_per_case: 20
  num_workers: 8

model:
  num_class: 17  # 17类标签（已删除"切除术后"）
  hidden_dim: 512
  nheads: 8
  num_decoder_layers: 2
  pretrained_resnet: True
  use_ppeg: False

training:
  batch_size: 4  
  epochs: 50
  lr: 0.0001
  weight_decay: 0.0001
  optimizer: "AdamW"

  # Asymmetric Loss 参数（处理多标签不平衡）
  gamma_neg: 4
  gamma_pos: 1
  clip: 0.05

  # 内存优化策略
  use_amp: true  # 混合精度训练
  gradient_accumulation_steps: 4  # 有效 batch_size = 4 * 4 = 16
  gradient_checkpointing: true

  # 学习率调度器
  scheduler: "cosine"
  warmup_epochs: 5

  # 模型保存
  save_dir: "checkpoints/"
  save_freq: 5