|
a |
|
b/echonet/__init__.py |
|
|
1 |
""" |
|
|
2 |
The echonet package contains code for loading echocardiogram videos, and |
|
|
3 |
functions for training and testing segmentation and ejection fraction |
|
|
4 |
prediction models. |
|
|
5 |
""" |
|
|
6 |
|
|
|
7 |
import click |
|
|
8 |
|
|
|
9 |
from echonet.__version__ import __version__ |
|
|
10 |
from echonet.config import CONFIG as config |
|
|
11 |
import echonet.datasets as datasets |
|
|
12 |
import echonet.utils as utils |
|
|
13 |
|
|
|
14 |
|
|
|
15 |
@click.group() |
|
|
16 |
def main(): |
|
|
17 |
"""Entry point for command line interface.""" |
|
|
18 |
|
|
|
19 |
|
|
|
20 |
del click |
|
|
21 |
|
|
|
22 |
|
|
|
23 |
main.add_command(utils.segmentation.run) |
|
|
24 |
main.add_command(utils.video.run) |
|
|
25 |
|
|
|
26 |
__all__ = ["__version__", "config", "datasets", "main", "utils"] |