diff --git a/docs/source/index.rst b/docs/source/index.rst index 93b29e3..b812fde 100644 --- a/docs/source/index.rst +++ b/docs/source/index.rst @@ -1,5 +1,5 @@ stroke_segmentor documentation -=================== +============================== .. include:: ../README_preprocessed.md :parser: myst_parser.sphinx_ diff --git a/stroke_segmentor/__init__.py b/stroke_segmentor/__init__.py index bb35ec8..1ad64a5 100644 --- a/stroke_segmentor/__init__.py +++ b/stroke_segmentor/__init__.py @@ -1,3 +1,5 @@ from loguru import logger logger.disable("stroke_segmentor") + +from stroke_segmentor.inferer import Inferer # noqa: F401 diff --git a/stroke_segmentor/inferer.py b/stroke_segmentor/inferer.py index 0838ee6..7dbc9fc 100644 --- a/stroke_segmentor/inferer.py +++ b/stroke_segmentor/inferer.py @@ -15,14 +15,6 @@ class Inferer: Handles inference for stroke segmentation using DWI and ADC images. This class wraps a model handler to perform inference on given images and can optionally save the resulting segmentation mask to disk. - - Basic usage: - ``` - .. code-block:: python - - inferer = Inferer() - prediction = inferer.infer(adc_path, dwi_path, segmentation_path) - ``` """ def __init__(