PyTorch-UNet:UNet模型的PyTorch实现-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