/
__init__.py
34 lines (30 loc) · 887 Bytes
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
"""Top-level package for torchio."""
__author__ = """Fernando Perez-Garcia"""
__email__ = 'fernando.perezgarcia.17@ucl.ac.uk'
__version__ = '0.17.33'
import os
from . import utils
from .torchio import *
from .transforms import *
from .data import (
io,
sampler,
inference,
ImagesDataset,
SubjectsDataset,
Image,
ScalarImage,
LabelMap,
Queue,
Subject,
)
from . import datasets
from . import reference
CITATION = """If you use TorchIO for your research, please cite the following paper:
Pérez-García et al., TorchIO: a Python library for efficient loading,
preprocessing, augmentation and patch-based sampling of medical images
in deep learning. Link: https://arxiv.org/abs/2003.04696
"""
# Thanks for citing torchio. Without citations, researchers will not use TorchIO
if 'TORCHIO_HIDE_CITATION_PROMPT' not in os.environ:
print(CITATION)