Skip to main content

PyTorch implementation of HighRes3DNet

Project description

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

PyTorch implementation of HighRes3DNet from Li et al. 2017, *On the Compactness, Efficiency, and Representation of 3D Convolutional Networks: Brain Parcellation as a Pretext Task*.

All the information about how the weights were ported from NiftyNet can be found in my submission to the MICCAI Educational Challenge 2019.

Usage

Command line interface

(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

If you are using pytorch>=1.1.0, you can import the model directly from this repository using 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)
>>>

Installation

2. Install PyTorch and highresnet

Within the conda environment:

pip install pytorch highresnet

Now you can do

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

Credits

This package was created with Cookiecutter and the audreyr/cookiecutter-pypackage project template.

History

0.7.1 (2019-11-05)

  • First release on PyPI.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

highresnet-0.9.2.tar.gz (19.4 kB view details)

Uploaded Source

File details

Details for the file highresnet-0.9.2.tar.gz.

File metadata

  • Download URL: highresnet-0.9.2.tar.gz
  • Upload date:
  • Size: 19.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/1.14.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.0.1 requests-toolbelt/0.9.1 tqdm/4.32.2 CPython/3.6.9

File hashes

Hashes for highresnet-0.9.2.tar.gz
Algorithm Hash digest
SHA256 8db44e7c64e92182aa7a16c77a1c9dfb364f1d58f7691b75152f11c45394467d
MD5 e4e4ba189ea662240f7d456bf4a32b7e
BLAKE2b-256 a8407730a16b2a388ae4895f7752666dde0c3ba674cae55e6f1049dd6c4bef26

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page