Spaces:
Running
on
Zero
Running
on
Zero
File size: 549 Bytes
e85fecb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
"""
Copied from RT-DETR (https://github.com/lyuwenyu/RT-DETR)
Copyright(c) 2023 lyuwenyu. All Rights Reserved.
"""
from ._misc import convert_to_tv_tensor
from .dataloader import *
from .dataset import *
from .transforms import *
# def set_epoch(self, epoch) -> None:
# self.epoch = epoch
# def _set_epoch_func(datasets):
# """Add `set_epoch` for datasets
# """
# from ..core import register
# for ds in datasets:
# register(ds)(set_epoch)
# _set_epoch_func([CIFAR10, VOCDetection, CocoDetection])
|