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
