V-Net 的 PyTorch 实现:用于体积医学图像分割的全卷积神经网络
Vnet 是 Fausto Milletari、Nassir Navab 和 Seyed-Ahmad Ahmadi 论文的 PyTorch 实现。虽然这个实现仍在进行中,但我已经在 249 个 epoch 后,从 LUNA16 数据集中分割出肺部,测试误差率为 0.355%,Dice 系数为 0.9825。官方实现可在 faustomilletari/VNet 仓库中找到。
此版本使用批量归一化和丢弃。CT 中的肺部体积约占扫描体积的 10%,这是一个不太不合理的类别平衡。对于此特定的测试应用程序,我添加了使用 NLLoss 而不是 Dice 系数的选项。
V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation
暂无开放 Issues,或尚未同步最近议题。