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.
55 lines
1.8 KiB
55 lines
1.8 KiB
# One-Prompt Medical Image Segmentation 配置文件
|
|
# 项目: One-Prompt to Segment All Medical Images (CVPR 2024)
|
|
|
|
project:
|
|
name: "one-prompt-segmentation"
|
|
version: "1.0.0"
|
|
description: "一提示分割所有医学图像"
|
|
paper: "https://arxiv.org/abs/2305.10300"
|
|
|
|
# 数据配置
|
|
data:
|
|
dataset: "polyp" # 数据集类型: polyp, isic, refuge
|
|
data_path: "/root/wangtao/paper_reapppearence/data/TestDataset"
|
|
train_ratio: 0.8
|
|
batch_size: 1
|
|
num_workers: 4
|
|
|
|
# 模型配置
|
|
model:
|
|
net: "oneprompt" # 网络类型
|
|
baseline: "unet" # 基线模型: unet, resnet
|
|
mod: "one_adpt" # 模块类型
|
|
image_size: 256 # 输入图像大小
|
|
out_size: 256 # 输出大小
|
|
patch_size: 16 # Patch大小 (需要等于 2^num_pool)
|
|
dim: 256 # 嵌入维度
|
|
depth: 1 # Transformer深度
|
|
heads: 16 # 注意力头数
|
|
mlp_dim: 1024 # MLP维度
|
|
|
|
# 训练配置
|
|
training:
|
|
epochs: 100 # 训练轮数
|
|
learning_rate: 0.0001 # 学习率
|
|
optimizer: "adam" # 优化器
|
|
weight_decay: 0.0 # 权重衰减
|
|
scheduler:
|
|
name: "step" # 学习率调度器
|
|
step_size: 10 # 步长
|
|
gamma: 0.5 # 衰减因子
|
|
early_stopping_patience: 20 # 早停耐心值
|
|
gradient_clip: 1.0 # 梯度裁剪
|
|
|
|
# 验证配置
|
|
validation:
|
|
val_freq: 5 # 验证频率
|
|
vis_freq: 50 # 可视化频率
|
|
|
|
# 日志配置
|
|
logging:
|
|
log_dir: "logs"
|
|
tensorboard: true
|
|
save_best: true
|
|
checkpoint_freq: 10
|