Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 379 Bytes
Newer Older
Kai Chen's avatar
set up the codebase skeleton (WIP)  
Kai Chen committed
1 
from .coco import CocoDataset
Kai Chen's avatar
fix flake8 error  
Kai Chen committed
2
3
4 
from .loader import (collate, GroupSampler, DistributedGroupSampler,
                     build_dataloader)
from .utils import DataContainer, to_tensor, random_scale, show_ann
Kai Chen's avatar
refactor DataContainer and datasets  
Kai Chen committed
5
Kai Chen's avatar
fix flake8 error  
Kai Chen committed
6
7
8
9
10 
__all__ = [
    'CocoDataset', 'collate', 'GroupSampler', 'DistributedGroupSampler',
    'build_dataloader', 'DataContainer', 'to_tensor', 'random_scale',
    'show_ann'
]