from monai.transforms import KeepLargestConnectedComponent post_pred = Compose([ Activations(sigmoid=True), AsDiscrete(threshold=0.5), KeepLargestConnectedComponent(applied_labels=[1]), ])