MLVLab/Human_Object_Interaction

1# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved2import torch.utils.data3import torchvision4 5from hotr.data.datasets.coco import build as build_coco6from hotr.data.datasets.vcoco import build as build_vcoco7from hotr.data.datasets.hico import build as build_hico8 9def get_coco_api_from_dataset(dataset):10    for _ in range(10): # what is this for?11        if isinstance(dataset, torch.utils.data.Subset):12            dataset = dataset.dataset13    if isinstance(dataset, torchvision.datasets.CocoDetection):14        return dataset.coco15 16 17def build_dataset(image_set, args):18    if args.dataset_file == 'coco':19        return build_coco(image_set, args)20    elif args.dataset_file == 'vcoco':21        return build_vcoco(image_set, args)22    elif args.dataset_file == 'hico-det':23        return build_hico(image_set, args)24    raise ValueError(f'dataset {args.dataset_file} not supported')