File size: 1,092 Bytes
e8b8483 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | """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',
]
|