跳转到主要内容

PyTorch实现的HighRes3DNet

项目描述

highresnet

https://img.shields.io/pypi/v/highresnet.svg https://img.shields.io/travis/fepegar/highresnet.svg https://zenodo.org/badge/DOI/10.5281/zenodo.3349989.svg Documentation Status Updates
$ NII_FILE=`download_oasis`
$ deepgif $NII_FILE
3D Slicer screenshot

来自Li et al. 2017, *On the Compactness, Efficiency, and Representation of 3D Convolutional Networks: Brain Parcellation as a Pretext Task* 的PyTorch实现的HighRes3DNet。

有关如何将权重从NiftyNet迁移的所有信息,请参阅我在MICCAI Educational Challenge 2019的提交

用法

命令行界面

(deepgif) $ deepgif t1_mri.nii.gz
Using cache found in /home/fernando/.cache/torch/hub/fepegar_highresnet_master
100%|███████████████████████████████████████████| 36/36 [01:13<00:00,  2.05s/it]

PyTorch Hub

如果您正在使用 pytorch>=1.1.0,您可以直接从PyTorch Hub导入模型。

>>> import torch
>>> repo = 'fepegar/highresnet'
>>> model_name = 'highres3dnet'
>>> print(torch.hub.help(repo, model_name))
    "HighRes3DNet by Li et al. 2017 for T1-MRI brain parcellation"
    "pretrained (bool): load parameters from pretrained model"
>>> model = torch.hub.load(repo, model_name, pretrained=True)
>>>

安装

2. 安装PyTorch和highresnet

在conda环境中

pip install pytorch highresnet

现在您可以

>>> from highresnet import HighRes3DNet
>>> model = HighRes3DNet(in_channels=1, out_channels=160)
>>>

致谢

此包是用Cookiecutteraudreyr/cookiecutter-pypackage项目模板创建的。

历史

0.7.1 (2019-11-05)

  • PyPI上的首次发布。

项目详情


下载文件

下载您平台对应的文件。如果您不确定选择哪个,请了解更多关于安装包的信息。

源分布

highresnet-0.10.2.tar.gz (19.6 kB 查看哈希值)

上传时间 源代码

构建分布

highresnet-0.10.2-py2.py3-none-any.whl (17.0 kB 查看哈希值)

上传时间 Python 2 Python 3

由以下支持