You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
1.7 KiB
100 lines
1.7 KiB
配置指南
|
|
=========
|
|
|
|
配置文件结构
|
|
------------
|
|
|
|
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 损失权重
|