File size: 1,006 Bytes
2cd560a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# ------------------------------------------------------------------------
# Copyright (c) 2022 megvii-research. All Rights Reserved.
# ------------------------------------------------------------------------
# Modified from Deformable DETR (https://github.com/fundamentalvision/Deformable-DETR)
# Copyright (c) 2020 SenseTime. All Rights Reserved.
# ------------------------------------------------------------------------
# Modified from DETR (https://github.com/facebookresearch/detr)
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
# ------------------------------------------------------------------------
from .dance import build as build_e2e_dance
from .joint import build as build_e2e_joint
def build_dataset(image_set, args):
if args.dataset_file == 'e2e_joint':
return build_e2e_joint(image_set, args)
if args.dataset_file == 'e2e_dance':
return build_e2e_dance(image_set, args)
raise ValueError(f'dataset {args.dataset_file} not supported')
|