segmenters/base_segmenter.py

import numpy as np

class BaseSegmenter:
    """Base class for segmentation models """
    def get_object_mask(self, image: np.ndarray) -> np.ndarray:
        """
        Args:
            image
        Returns:
            bool mask of shape, where True = foreground object.
        """
        raise NotImplementedError