Skip to content

Commit

Permalink
[DLMED] add more clear doc-string
Browse files Browse the repository at this point in the history
Signed-off-by: Nic Ma <[email protected]>
  • Loading branch information
Nic-Ma committed Jan 11, 2021
1 parent 60639e3 commit bb7064e
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions monai/inferers/inferer.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,19 @@ class Inferer(ABC):
"""
A base class for model inference.
Extend this class to support operations during inference, e.g. a sliding window method.
Example code::
device = torch.device("cuda:0")
data = ToTensor()(LoadImage(...)()).to(device)
model = UNet(...).to(device)
inferer = SlidingWindowInferer(...)
model.eval()
with torch.no_grad():
pred = inferer(inputs=data, network=model)
...
"""

@abstractmethod
Expand Down Expand Up @@ -53,6 +66,7 @@ def __call__(
class SimpleInferer(Inferer):
"""
SimpleInferer is the normal inference method that run model forward() directly.
Usage example can be found in the :py:class:`monai.inferers.Inferer` base class.
"""

Expand Down Expand Up @@ -83,6 +97,7 @@ class SlidingWindowInferer(Inferer):
"""
Sliding window method for model inference,
with `sw_batch_size` windows for every model.forward().
Usage example can be found in the :py:class:`monai.inferers.Inferer` base class.
Args:
roi_size: the window size to execute SlidingWindow evaluation.
Expand Down

0 comments on commit bb7064e

Please sign in to comment.