-
PyTorch 实现的 UNet 模型:PyTorch-UNet,访问链接 - https
资源介绍
U-Net-PyTorch实施
模型(一种流行的图像分割网络)的实现。 这是非常稳定和可配置的,我已经在多个数据集中使用了它,并将其作为几个项目的组成部分。
更新:还支持基于的3-D卷的分段
更新:所有批次归一化操作已被实例归一化所取代(以解决医学图像中的小批量),并且ReLU激活已被LeakyReLU取代,因为它在最近的工作中得到了更大的采用。
安装
您可以将此软件包安装在本地python环境中,并将其作为模块导入项目中。
将此存储库克隆到您选择的文件夹中。
cd
git clone https://github.com/kilgore92/PyTorch-UNet.git
安装软件包依赖项,如下所示:
cd
/bin/pip install -r requirements.txt