修改nnUNet的部分组件(参考作者提供的doc)

本文档介绍了如何修改nnUNet的部分组件,包括蓝图参数、推断参数和网络结构。修改蓝图参数涉及损失函数、训练策略等,可通过创建新的trainer class实现。推断参数由数据指纹决定,可通过自定义ExperimentPlanner进行调整。对于网络结构的修改,需要考虑下采样次数、深监督等因素,并确保显存兼容。作者提供了示例代码和实验计划者以供参考。
摘要由CSDN通过智能技术生成

修改nnUNet的部分组件(参考作者提供的doc)

下文主要是作者在nnUNet.documentation.extending_nnunet的翻译
https://github.com/MIC-DKFZ/nnUNet/tree/master/documentation/extending_nnunet.md

  1. 首先确保安装nnunet的方法为从github拖最新代码到本地,并使用pip install -e .进行安装,这样可确保本地有一份源码的拷贝
  2. 根据参数不同(推断,蓝图,经验参数/inferred, blueprint or empirical parameters)需要修改不同的地方
  • 修改蓝图参数/blueprint parameters:
    这部分是修改损失函数,训练策略,学习率,优化器,部分结构参数,数据增强等,以上所有部分均是nnU-Net trainer class的一部分。2D,3D fullresolution,lowresolution的默认trainer class是nnUNetTrainerV2,3d cascade fullresolution的默认trainer class是nnUNetTrainerV2CascadeFullRes。不同的trainer class相互继承。
    修改部分功能组件,只需要新建一个trainer class,仿照其他trainer class进行改写,在训练时使用这个trainer class即可。这一步需要将新的trainer class防止在nnunet.training.network_training下的子文件夹中,并且不要和其他文件重名。
    作者在这些子文件夹中给出了几种修改部分组件的例程,
  • 1
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 7
    评论
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值