nnUnet代码分析一训练

本文介绍了nnUNet在医学图像分析中的应用,特别是2D分割任务。nnUNet_train2d用于训练,nnUNetTrainerV2作为trainer,使用dice和CE损失函数。数据增强包括旋转和缩放。训练过程展示了在不同GPU上(如k80和2080ti)的速度差异。实验结果显示,随着epoch增加,模型的分割性能逐渐提升,例如在海马体分割任务中达到了高精度。

nnUnet是做分割的一套完整代码,用在医学图像分析中较多,效果还很不错。

先看训练的代码 run_training.py

一般用法:nnUNet_train 2d nnUNetTrainerV2 TaskXXX_MYTASK FOLD --npz

2d代表2d Unet网络,nnUNetTrainerV2代表trainer,Task是任务id,

还有其他的参数。详见代码。

plans_file, output_folder_name, dataset_directory, batch_dice, stage, \

    trainer_class = get_default_configuration(network, task, network_trainer, plans_identifier)

根据网络、任务、trainer,计划生产一个trainer_class.

命令:nnUNet_train 2d  nnUNetTrainerV2 Task004_Hippocampus 1 --npz

输出结果如下:

###############################################
I am running the following nnUNet: 2d
My trainer class is:  <class 'nnunet.training.network_training.nnUNetTrainerV2.nnUNetTrainerV2'>
For that I will be using the following configuration:
num_classes:  2
modalities:  {0: 'MRI'}
use_mask_for_norm OrderedDict([(0, False)])
keep_only_large

评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值