文件名称:PyTorch-UNet:UNet模型的PyTorch实现-https
文件大小:20KB
文件格式:ZIP
更新时间:2024-06-04 23:03:21
pytorch medical-imaging image-segmentation u-net unet-pytorch
U-Net-PyTorch实施
模型(一种流行的图像分割网络)的实现。 这是非常稳定和可配置的,我已经在多个数据集中使用了它,并将其作为几个项目的组成部分。
更新:还支持基于的3-D卷的分段
更新:所有批次归一化操作已被实例归一化所取代(以解决医学图像中的小批量),并且ReLU激活已被LeakyReLU取代,因为它在最近的工作中得到了更大的采用。
安装
您可以将此软件包安装在本地python环境中,并将其作为模块导入项目中。
将此存储库克隆到您选择的文件夹中。
cd
【文件预览】:
PyTorch-UNet-master
----LICENSE.txt(1KB)
----CHANGELOG.rst(128B)
----src()
--------unet()
----requirements.txt(655B)
----AUTHORS.rst(78B)
----setup.cfg(3KB)
----setup.py(568B)
----README.md(3KB)
----docs()
--------authors.rst(41B)
--------conf.py(9KB)
--------index.rst(2KB)
--------Makefile(7KB)
--------license.rst(67B)
--------changelog.rst(43B)
--------_static()
----tests()
--------test_skeleton.py(288B)
--------conftest.py(225B)
----.gitignore(510B)
----.coveragerc(587B)