INNER CODE UNIT · Python
__init__
openfoodfacts/openfoodfacts-ai · logo-ann/benchmarks/embedding_models_benchmark/main.py:41
def __init__(
self,
root_dir: Path,
transform_func: Callable[[Image.Image], Image.Image],
split_set: Optional[Set[str]] = None,
) -> None:
super().__init__()
self.root_dir = root_dir
self.transform_func = transform_func
children = [p for p in root_dir.iterdir() if p.is_dir()]
self.label_names = []
self.image_paths = []
labels_by_name = []
for child in children:
image_paths = list(child.glob("*.png"))
if split_set is not None:
image_paths = [
x for x in image_paths if f"{x.parent.name}/{x.name}" in split_set