nnUNet V2修改网络——暴力替换网络为UNet++
更换前,要用nnUNet V2跑通所用数据集,证明nnUNet V2、数据集、运行环境等没有问题
阅读nnU-Net V2 的 U-Net结构,初步了解要修改的网络,知己知彼,修改起来才能游刃有余。
U-Net存在两个局限,一是网络的最佳深度因应用场景而异,这取决于任务的难度和可用于训练的标注数据量,导致在部署时需要对不同深度的模型进行独立训练和集成,效率低下;二是跳跃连接的设计较为局限,仅允许融合编码器和解码器子网络中相同尺度的特征图,从而可能遗漏一些有用的特征融合机会。为了解决上述问题,提出了UNet++。
UNet++论文:arxiv
UNet++官方代码:点此查看
本文目录
- 一 准备工作
- 1. 生成nnUNetPlans.json文件
- 二 修改思路
- 三 修改网络