PyTorch implementation of HighRes3DNet
Project description
highresnet
$ NII_FILE=`download_oasis` $ deepgif $NII_FILE
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
1. Create a conda environment (recommended)
ENVNAME="gifenv"
conda create -n $ENVNAME python -y
conda activate $ENVNAME2. Install PyTorch and highresnet
Within the conda environment:
pip install pytorch highresnetNow 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.
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
File details
Details for the file highresnet-0.9.0.tar.gz.
File metadata
- Download URL: highresnet-0.9.0.tar.gz
- Upload date:
- Size: 19.5 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
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 0817686da24657bbe4beb848c986e230a0a284c3fe796468fba4b5bfd7d762b6 |
|
| MD5 | 832f97881afef7146c42d28835c4eeaa |
|
| BLAKE2b-256 | c5756d41389da72fa1e509f3126fa91a08b2e4f980af94fca274c54ac9b33ef6 |
