sapiens/pose/src/datasets/__init__.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

from .keypoints308_3po_dataset import Keypoints308_3PODataset
from .keypoints308_goliath_dataset import Keypoints308GoliathDataset
from .keypoints308_goliath_eval_dataset import Keypoints308GoliathEvalDataset
from .keypoints308_shutterstock_dataset import Keypoints308ShutterstockDataset
from .keypoints308_shutterstock_eval_dataset import Keypoints308ShutterstockEvalDataset
from .transforms import *
from .utils import parse_pose_metainfo

__all__ = [
    "Keypoints308ShutterstockDataset",
    "Keypoints308GoliathDataset",
    "Keypoints308_3PODataset",
    "Keypoints308ShutterstockEvalDataset",
    "Keypoints308GoliathEvalDataset",
    "parse_pose_metainfo",
]