配置指南 ========= 配置文件结构 ------------ EMCAD 使用 YAML 格式的配置文件,所有参数集中管理: .. code-block:: yaml # 数据集配置 dataset: name: "Synapse" root_path: "/data/Synapse/train" num_classes: 4 img_size: 224 # 模型配置 model: encoder: "pvt_v2_b2" num_classes: 4 kernel_sizes: [1, 3, 5] expansion_factor: 2 # 训练配置 training: max_epochs: 300 batch_size: 6 base_lr: 0.0001 # 输出配置 output: snapshot_path: "./experiments/" # 日志配置 logging: level: "INFO" 数据集配置 ---------- .. list-table:: :header-rows: 1 :widths: 20 80 * - 参数 - 描述 * - ``name`` - 数据集名称 (Synapse 或 ACDC) * - ``root_path`` - 训练数据根目录 * - ``volume_path`` - 测试数据目录 * - ``num_classes`` - 分割类别数 * - ``img_size`` - 输入图像尺寸 模型配置 -------- .. list-table:: :header-rows: 1 :widths: 20 80 * - 参数 - 描述 * - ``encoder`` - 编码器类型 (pvt_v2_b0/b1/b2, resnet18) * - ``kernel_sizes`` - MSCB 模块卷积核大小列表 * - ``expansion_factor`` - MSCB 模块扩展因子 * - ``dw_parallel`` - 深度卷积并行模式 * - ``add`` - 特征相加模式 (False 为拼接) 训练配置 -------- .. list-table:: :header-rows: 1 :widths: 20 80 * - 参数 - 描述 * - ``max_epochs`` - 最大训练轮数 * - ``batch_size`` - 批次大小 * - ``base_lr`` - 基础学习率 * - ``weight_decay`` - 权重衰减 * - ``w_ce`` - 交叉熵损失权重 * - ``w_dice`` - Dice 损失权重