nnUNet V2修改网络——替换为EGE-UNet

​更换前，要用nnUNet V2跑通所用数据集，证明nnUNet V2、数据集、运行环境等没有问题

阅读nnU-Net V2 的 U-Net结构，初步了解要修改的网络，知己知彼，修改起来才能游刃有余。

EGE-UNet 是 UNet 的一个变体，专为皮肤病变分割而设计。它在 UNet 的基础上引入了 GHPA 模块和 GAB 模块，分别用于多维度特征提取和多尺度信息融合。GHPA 运用多轴分组和 Hadamard 乘积机制，能从不同视角提取病理信息，而 GAB 则通过分组聚合和卷积操作，有效整合不同层次和尺度的特征。这种创新设计使得 EGE-UNet 不仅在 ISIC2017 和 ISIC2018 数据集上取得卓越的分割效果，还大幅降低了参数和计算成本，成为低资源环境下理想的分割模型。

EGE-UNet官方代码仓库： https://github.com/JCruan519/EGE-UNet

一 准备工作

1. 安装dynamic-network-architectures

点击链接，将其clone到本地后，进入文件夹内，pip install -e . 即可（注意-e后有个点）。

二 修改思路

1. 查看 EGE-UNet 网络结构

EGE-Unet 也分为三部分，编码器，解码器，跳跃连接部分。

编码器：论文中提到EGE-UNet的编码器有6层，前三层为普通的卷积层，后三个阶段使用设计的GHPA模块，每层卷积后依次通过 norm层、 pool层、non_linear层；出于轻量化的考虑，各层通道数也不再是一般的32 --> 64 --> 128 -->…，而是设计为{8, 16, 24, 32, 48, 64}。

解码器：也是六层，前三层为GHPA模块，后三层为普通的卷积层，每层卷积后依次通过 norm层、 pool层、linear interpolation层，与编码器是对称的。通道数也和编码器对应层一致。