我正在研究生物医学图像分割。为此,我需要实现 SegNet 模型。我在很多地方搜索了 SegNet 实现,但这些都没有为我提供正确的实现。我在没有使用预训练编码器的情况下得到了一些实现。但是从 SegNet 的论文中,我知道 SegNet 使用预训练的编码器,它是在 ImageNet 数据集上训练的 VGG-16 网络的修剪部分。我需要在 Keras 中实现。
注意 keras 中有一个预训练的 VGG-16 网络。但这缺少 SegNet 的原始论文中存在的批量归一化层。
PS 由于计算资源的稀缺,我无法自己重新训练 VGG-16 网络。