def load_words(self, data: DataSetBase, image, masked): words = data.load_words(image) if masked: mask = self.load_mask(data, image) if mask is not None: words = words[mask] return words
def load_words(self, data: DataSetBase, image: str, masked: bool) -> np.ndarray: words = data.load_words(image) if masked: mask = self.load_mask(data, image) if mask is not None: words = words[mask] return words