facepoints / train /data /__init__.py
antonios-makro's picture
Facepoints 1.0.0: code, model, prep data and card
88607d8 verified
Raw History Blame Contribute Delete
399 Bytes
"""Datasets and augmentation: studio set (ava), in-the-wild sets (wild), augmentation (augment)."""
# What the trainer imports; everything else is reached through the submodules.
from facepoints.train.data.augment import Affine, LowRes # noqa: F401
from facepoints.train.data.ava import PointData # noqa: F401
from facepoints.train.data.wild import FLIP68, MILD_BIT, Wild, WildData # noqa: F401