Spaces:
Sleeping
Sleeping
File size: 1,497 Bytes
77da9e2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
# ------------------------------------------------------------------------
# LW-DETR
# Copyright (c) 2024 Baidu. All Rights Reserved.
# Licensed under the Apache License, Version 2.0 [see LICENSE for details]
# ------------------------------------------------------------------------
# Modified from Conditional DETR (https://github.com/Atten4Vis/ConditionalDETR)
# Copyright (c) 2021 Microsoft. All Rights Reserved.
# ------------------------------------------------------------------------
# Copied from DETR (https://github.com/facebookresearch/detr)
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
# ------------------------------------------------------------------------
import torch.utils.data
import torchvision
from .coco import build as build_coco
from .o365 import build_o365
from .coco import build_roboflow
def get_coco_api_from_dataset(dataset):
for _ in range(10):
if isinstance(dataset, torch.utils.data.Subset):
dataset = dataset.dataset
if isinstance(dataset, torchvision.datasets.CocoDetection):
return dataset.coco
def build_dataset(image_set, args, resolution):
if args.dataset_file == 'coco':
return build_coco(image_set, args, resolution)
if args.dataset_file == 'o365':
return build_o365(image_set, args, resolution)
if args.dataset_file == 'roboflow':
return build_roboflow(image_set, args, resolution)
raise ValueError(f'dataset {args.dataset_file} not supported')
|