Skip to content
5 changes: 5 additions & 0 deletions monai/networks/nets/milmodel.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,11 @@
class MILModel(nn.Module):
"""
Multiple Instance Learning (MIL) model, with a backbone classification model.
Currently, it only works for 2D images, typical use case is for classification of the

@wyli wyli Nov 26, 2021

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

looks like this model will be 2D only, the name MILModel is too generic and confusing cc @myron

digital pathology whole slide images. The expected shape of input data is `[B, N, C, H, W]`,
where `B` is the batch_size of PyTorch Dataloader and `N` is the number of the instances
extracted from every original image in the batch. A tutorial example is available at:
https://github.com/Project-MONAI/tutorials/tree/master/pathology/multiple_instance_learning.

Args:
num_classes: number of output classes.
Expand Down