common/__init__.py

"""Pooled features, the comparison rule, metrics, named pools, artifact contract."""
from .paths import BACKBONE, BACKBONE_SRC, COCO_ROOT, REPO, UPSTREAM_BACKBONE, device
from .data import (MEAN, STD, coco_split, image_paths, load_image, normalize,
                   person_labels)
from .features import D, RES, backbone_pooled, decide, pool, score
from .metrics import Metrics, f1_at, prf1
from .pools import POOLS, TRAIN2017, VAL5000, Pool, by_name
from .pool import LoadedPool, load_pool
from .artifacts import (REGISTRY, ArtifactSpec, provenance, read_artifact,
                        sha256_of, write_artifact)

__all__ = [
    'BACKBONE', 'BACKBONE_SRC', 'COCO_ROOT', 'REPO', 'UPSTREAM_BACKBONE', 'device',
    'MEAN', 'STD', 'coco_split', 'image_paths', 'load_image', 'normalize',
    'person_labels',
    'D', 'RES', 'backbone_pooled', 'decide', 'pool', 'score',
    'Metrics', 'f1_at', 'prf1',
    'POOLS', 'TRAIN2017', 'VAL5000', 'Pool', 'by_name',
    'LoadedPool', 'load_pool',
    'REGISTRY', 'ArtifactSpec', 'provenance', 'read_artifact', 'sha256_of',
    'write_artifact',
]