mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
b997a13e28
* add isprs potsdam dataset * add isprs dataset configs * fix lint error * fix potsdam conversion bug * fix error in potsdam class * fix error in potsdam class * add vaihingen dataset * add vaihingen dataset * add vaihingen dataset * fix some description errors. * fix some description errors. * fix some description errors. * upload models & logs of Potsdam * remove vaihingen and add unit test * add chinese readme * add pseudodataset * use mmcv and add class_names * use f-string * add new dataset unittest * add docstring and remove global variables args * fix metafile error in PSPNet * fix pretrained value * Add dataset info * fix typo Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
29 lines
1.2 KiB
Python
29 lines
1.2 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .ade import ADE20KDataset
|
|
from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset
|
|
from .chase_db1 import ChaseDB1Dataset
|
|
from .cityscapes import CityscapesDataset
|
|
from .coco_stuff import COCOStuffDataset
|
|
from .custom import CustomDataset
|
|
from .dark_zurich import DarkZurichDataset
|
|
from .dataset_wrappers import (ConcatDataset, MultiImageMixDataset,
|
|
RepeatDataset)
|
|
from .drive import DRIVEDataset
|
|
from .hrf import HRFDataset
|
|
from .loveda import LoveDADataset
|
|
from .night_driving import NightDrivingDataset
|
|
from .pascal_context import PascalContextDataset, PascalContextDataset59
|
|
from .potsdam import PotsdamDataset
|
|
from .stare import STAREDataset
|
|
from .voc import PascalVOCDataset
|
|
|
|
__all__ = [
|
|
'CustomDataset', 'build_dataloader', 'ConcatDataset', 'RepeatDataset',
|
|
'DATASETS', 'build_dataset', 'PIPELINES', 'CityscapesDataset',
|
|
'PascalVOCDataset', 'ADE20KDataset', 'PascalContextDataset',
|
|
'PascalContextDataset59', 'ChaseDB1Dataset', 'DRIVEDataset', 'HRFDataset',
|
|
'STAREDataset', 'DarkZurichDataset', 'NightDrivingDataset',
|
|
'COCOStuffDataset', 'LoveDADataset', 'MultiImageMixDataset',
|
|
'PotsdamDataset'
|
|
]
|