Spaces:
Running
on
Zero
Running
on
Zero
File size: 642 Bytes
e85fecb |
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 |
"""
Copied from RT-DETR (https://github.com/lyuwenyu/RT-DETR)
Copyright(c) 2023 lyuwenyu. All Rights Reserved.
"""
from typing import Callable, Optional
import torchvision
from ...core import register
@register()
class CIFAR10(torchvision.datasets.CIFAR10):
__inject__ = ["transform", "target_transform"]
def __init__(
self,
root: str,
train: bool = True,
transform: Optional[Callable] = None,
target_transform: Optional[Callable] = None,
download: bool = False,
) -> None:
super().__init__(root, train, transform, target_transform, download)
|