services/training_service/engine/data/__init__.py

"""
Copied from RT-DETR (https://github.com/lyuwenyu/RT-DETR)
Copyright(c) 2023 lyuwenyu. All Rights Reserved.
"""

from .dataset import *
from .transforms import *
from .dataloader import *

from ._misc import convert_to_tv_tensor




# 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])