mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
62f70ebedd
* support coco stuff-10k/164k * update docs * fix docs * update docs * fix import lints * Update docs/dataset_prepare.md * Update docs/dataset_prepare.md * Update tools/convert_datasets/coco_stuff164k.py * Update tools/convert_datasets/coco_stuff10k.py * Update tools/convert_datasets/coco_stuff10k.py * Update tools/convert_datasets/coco_stuff10k.py * Update tools/convert_datasets/coco_stuff10k.py * Update coco_stuff.py fix the description of the dataset * Update dataset_prepare.md fix the doc tree of coco stuff 10k * Update coco_stuff10k.py fix img_dir * Update coco_stuff.py fix descriptions * Update coco_stuff164k.py fix out_dir * Update coco_stuff10k.py fix save file name * Update coco_stuff.py fix seg_map_suffix * Update dataset_prepare.md fix -p * Update dataset_prepare.md fix doc tree * modify coco stuff convertor * Remove redundant code * fix 164k convert bug * remove redundant comment * add deeplabv3 configs and more iterations * replace shutil.move with shtil.copyfile * Update deeplabv3_r50-d8_512x512_4x4_80k_coco_stuff10k.py fix wrong config * Update deeplabv3_r101-d8_512x512_4x4_80k_coco_stuff164k.py fix wrong config * fix wrong configs * fix wrong configs * fix wrong path for coco stuff 10k * fix convert bugs * fix seg_filename bug * when nproc=0, use track progress * rename configs: coco_stuff --> coco-stuff * add coco-stuff 10k and 164k to README.md * update configs * add deeplabv3 benchmark * add pspnet benchmark * remove redundant comma Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
25 lines
1.0 KiB
Python
25 lines
1.0 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, RepeatDataset
|
|
from .drive import DRIVEDataset
|
|
from .hrf import HRFDataset
|
|
from .night_driving import NightDrivingDataset
|
|
from .pascal_context import PascalContextDataset, PascalContextDataset59
|
|
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'
|
|
]
|